装柜

firstleg/src/test/java/com/baiyue/firstleg/service/ContainerServiceTest.java

@@ -1,6 +1,8 @@
 package com.baiyue.firstleg.service;
 
+import com.baiyue.common.core.api.ApiResponse;
 import com.baiyue.common.core.exception.BizException;
+import com.baiyue.firstleg.client.WarehouseMasterDataClient;
 import com.baiyue.firstleg.dto.ContainerDtos.FullUpdateRequest;
 import com.baiyue.firstleg.dto.ContainerDtos.SaveRequest;
 import com.baiyue.firstleg.dto.ContainerDtos.TransportUpdateRequest;
@@ -8,6 +10,7 @@ import com.baiyue.firstleg.entity.ContainerBookingEntity;
 import com.baiyue.firstleg.repository.ContainerRepository;
 import jakarta.annotation.Resource;
 import org.junit.jupiter.api.DisplayName;
+import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
 import org.mockito.ArgumentCaptor;
 import org.springframework.boot.test.mock.mockito.MockBean;
@@ -19,6 +22,8 @@ import java.time.LocalDateTime;
 import static org.assertj.core.api.Assertions.assertThat;
 import static org.assertj.core.api.Assertions.assertThatThrownBy;
 import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.ArgumentMatchers.anyLong;
+import static org.mockito.ArgumentMatchers.anyString;
 import static org.mockito.Mockito.never;
 import static org.mockito.Mockito.verify;
 import static org.mockito.Mockito.when;
@@ -27,8 +32,15 @@ import static org.mockito.Mockito.when;
 @SpringJUnitConfig(ContainerService.class)
 class ContainerServiceTest {
     @MockBean private ContainerRepository repository;
+    @MockBean private WarehouseMasterDataClient warehouses;
     @Resource private ContainerService service;
 
+    @BeforeEach
+    void setUpWarehouse() {
+        when(warehouses.active(anyString(), anyLong(), anyLong())).thenAnswer(invocation ->
+            activeWarehouse(invocation.getArgument(2, Long.class), "DOMESTIC", "上海中转仓"));
+    }
+
     @Test
     @DisplayName("新增货柜保存最终数据库契约字段并生成平台预订单号")
     void createsContainerBooking() {
@@ -40,6 +52,8 @@ class ContainerServiceTest {
         assertThat(created).isSameAs(saved);
         assertThat(saved.getBookingNo()).startsWith("CB");
         assertThat(saved.getCabinetNo()).isEqualTo("柜-001");
+        assertThat(saved.getWarehouseId()).isEqualTo(101L);
+        assertThat(saved.getWarehouseName()).isEqualTo("上海中转仓");
         assertThat(saved.getDeparturePortCode()).isEqualTo("CNSHA");
         assertThat(saved.getCapacityCbm()).isEqualByComparingTo("68.0000");
         assertThat(saved.getPlanStatus()).isEqualTo("EDITING");
@@ -127,6 +141,49 @@ class ContainerServiceTest {
         verify(repository, never()).insert(any());
     }
 
+    @Test
+    @DisplayName("虚拟柜拒绝选择非国内中转仓")
+    void rejectsNonDomesticWarehouse() {
+        when(warehouses.active(anyString(), anyLong(), anyLong()))
+            .thenReturn(activeWarehouse(101L, "FBA", "美国目的仓"));
+
+        assertThatThrownBy(() -> service.create(9L, "admin", request("柜-001")))
+            .isInstanceOf(BizException.class)
+            .hasMessageContaining("国内中转仓");
+
+        verify(repository, never()).insert(any());
+    }
+
+    @Test
+    @DisplayName("虚拟柜仓库名称以仓库主数据为准")
+    void replacesRequestedWarehouseName() {
+        service.create(9L, "admin", request("柜-001"));
+
+        ArgumentCaptor<ContainerBookingEntity> captor = ArgumentCaptor.forClass(ContainerBookingEntity.class);
+        verify(repository).insert(captor.capture());
+        assertThat(captor.getValue().getWarehouseName()).isEqualTo("上海中转仓");
+    }
+
+    @Test
+    @DisplayName("仓库主数据服务不可用时拒绝创建虚拟柜")
+    void rejectsUnavailableWarehouseService() {
+        when(warehouses.active(anyString(), anyLong(), anyLong()))
+            .thenThrow(new IllegalStateException("warehouse service unavailable"));
+
+        assertThatThrownBy(() -> service.create(9L, "admin", request("柜-001")))
+            .isInstanceOf(BizException.class)
+            .hasMessageContaining("暂时不可用");
+
+        verify(repository, never()).insert(any());
+    }
+
+    private static ApiResponse<WarehouseMasterDataClient.Detail> activeWarehouse(long warehouseId,
+                                                                                   String type,
+                                                                                   String name) {
+        return ApiResponse.success(new WarehouseMasterDataClient.Detail(
+            new WarehouseMasterDataClient.Warehouse(warehouseId, 0L, "CN-SHA-01", name, type, "ACTIVE")));
+    }
+
     private static ContainerBookingEntity booking(String status, int version) {
         ContainerBookingEntity value = new ContainerBookingEntity();
         value.setId(11L);