优化

firstleg/src/test/java/com/baiyue/firstleg/controller/InternalInboundOrderReferenceControllerTest.java

@@ -3,6 +3,9 @@ package com.baiyue.firstleg.controller;
 import com.baiyue.common.core.exception.BizException;
 import com.baiyue.firstleg.dto.InboundOrderReference;
 import com.baiyue.firstleg.service.InboundOrderCreationService;
+import com.baiyue.firstleg.service.UnifiedInboundQueryService;
+import com.baiyue.firstleg.dto.UnifiedInboundQuery;
+import com.baiyue.common.core.api.PageResult;
 import jakarta.annotation.Resource;
 import org.junit.jupiter.api.Test;
 import org.springframework.boot.test.mock.mockito.MockBean;
@@ -22,6 +25,8 @@ import static org.mockito.Mockito.when;
 class InternalInboundOrderReferenceControllerTest {
     @MockBean
     private InboundOrderCreationService service;
+    @MockBean
+    private UnifiedInboundQueryService unifiedQueryService;
     @Resource
     private InternalInboundOrderController controller;
 
@@ -44,4 +49,16 @@ class InternalInboundOrderReferenceControllerTest {
         assertThatThrownBy(() -> controller.references("bad-token", request))
             .isInstanceOf(BizException.class);
     }
+
+    @Test
+    void unifiedQueryRequiresInternalTokenAndForwardsBatchRequest() {
+        UnifiedInboundQuery request = new UnifiedInboundQuery(
+            10001L, null, null, "WAIT_SPLIT", null, null, null, null, null, 1, 20);
+        when(unifiedQueryService.page(request)).thenReturn(new PageResult<>(List.of(), 1, 20, 0, 0));
+
+        assertThat(controller.unifiedPage("internal-token", request).data().records()).isEmpty();
+        assertThatThrownBy(() -> controller.unifiedPage("bad-token", request))
+            .isInstanceOf(BizException.class);
+        verify(unifiedQueryService).page(request);
+    }
 }

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

@@ -0,0 +1,40 @@
+package com.baiyue.firstleg.service;
+
+import org.junit.jupiter.api.Test;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+class UnifiedInboundQueryServiceTest {
+    @Test
+    void mapsAllCompatibleInternalStatesToProductMainStatuses() {
+        assertThat(UnifiedInboundQueryService.mainStatus("WAIT_MEASURE")).isEqualTo("WAIT_MEASURE");
+        assertThat(UnifiedInboundQueryService.mainStatus("MEASURING")).isEqualTo("WAIT_MEASURE");
+        assertThat(UnifiedInboundQueryService.mainStatus("MEASURE_PENDING")).isEqualTo("WAIT_MEASURE");
+        assertThat(UnifiedInboundQueryService.mainStatus("MEASURE_REJECTED")).isEqualTo("WAIT_MEASURE");
+        assertThat(UnifiedInboundQueryService.mainStatus("WAIT_PRICING")).isEqualTo("WAIT_SPLIT");
+        assertThat(UnifiedInboundQueryService.mainStatus("FEE_CONFIRMING")).isEqualTo("WAIT_SPLIT");
+        assertThat(UnifiedInboundQueryService.mainStatus("INBOUNDING")).isEqualTo("WAIT_SPLIT");
+        assertThat(UnifiedInboundQueryService.mainStatus("INBOUNDED")).isEqualTo("INBOUNDED");
+        assertThat(UnifiedInboundQueryService.mainStatus("LOADING")).isEqualTo("LOADING");
+        assertThat(UnifiedInboundQueryService.mainStatus("LOADED")).isEqualTo("LOADED");
+        assertThat(UnifiedInboundQueryService.mainStatus("DELIVERED")).isEqualTo("DELIVERED");
+        assertThat(UnifiedInboundQueryService.mainStatus("VOIDED")).isNull();
+    }
+
+    @Test
+    void nextActionReflectsExecutableMeasurementAndPreloadSubstates() {
+        assertThat(UnifiedInboundQueryService.nextAction("WAIT_MEASURE", "MEASURING", null)).isNull();
+        assertThat(UnifiedInboundQueryService.nextAction("WAIT_MEASURE", "MEASURE_PENDING", null))
+            .isEqualTo("CONFIRM_MEASUREMENT");
+        assertThat(UnifiedInboundQueryService.nextAction("WAIT_SPLIT", "WAIT_SPLIT", null))
+            .isEqualTo("SMART_SPLIT");
+        assertThat(UnifiedInboundQueryService.nextAction("WAIT_SPLIT", "WAIT_SPLIT", "PENDING"))
+            .isEqualTo("CONFIRM_PRELOAD");
+        assertThat(UnifiedInboundQueryService.nextAction("WAIT_SPLIT", "INBOUNDING", null)).isNull();
+        assertThat(UnifiedInboundQueryService.nextAction("WAIT_SPLIT", "WAIT_SPLIT", "CONFIRMED")).isNull();
+        assertThat(UnifiedInboundQueryService.nextAction("INBOUNDED", "INBOUNDED", null)).isNull();
+        assertThat(UnifiedInboundQueryService.nextAction("LOADED", "LOADED", null))
+            .isEqualTo("COMPLETE_DELIVERY");
+        assertThat(UnifiedInboundQueryService.nextAction("DELIVERED", "DELIVERED", null)).isNull();
+    }
+}