优化

lastleg-service/src/test/java/com/baiyue/lastleg/transfer/service/WarehouseTransferServiceTest.java

@@ -206,6 +206,7 @@ class WarehouseTransferServiceTest {
         when(transfers.lockNo("TENANT-A", "WTR-1")).thenReturn(transfer);
         when(transfers.findNo("TENANT-A", "WTR-1")).thenReturn(transfer);
         when(inventoryAllocations.lockTransfer("TENANT-A", 21L)).thenReturn(List.of(allocation));
+        when(inventoryCommands.lockBalance("TENANT-A", 601L)).thenReturn(balance(601L, 101L, "IN-1"));
         when(orders.selectById(31L)).thenReturn(tailOrder);
         when(transferItems.selectList(any())).thenReturn(List.of());
         when(sourceOrders.selectList(any())).thenReturn(List.of());
@@ -237,6 +238,7 @@ class WarehouseTransferServiceTest {
             ArgumentCaptor.forClass(com.baiyue.lastleg.inventory.command.dto.InventoryCommandRequests.StockIn.class);
         verify(inventory).stockIn(eq("TENANT-A"), eq("USER-1"), stockIn.capture());
         assertEquals(5, stockIn.getValue().packages());
+        assertEquals("SKU-TRANSFER", stockIn.getValue().skuCode());
         assertEquals("WTR-1", stockIn.getValue().businessOrderNo());
         assertEquals("WAREHOUSE_TRANSFER_RECEIPT:WTR-1:501", stockIn.getValue().idempotencyKey());
         assertEquals(WarehouseTransferService.RECEIVED, transfer.getStatus());
@@ -350,7 +352,7 @@ class WarehouseTransferServiceTest {
     private static InventoryBalance balance(long id, long inboundOrderId, String inboundCode) {
         InventoryBalance balance = new InventoryBalance();
         balance.setId(id); balance.setCustomerId("TENANT-A"); balance.setInboundOrderId(inboundOrderId);
-        balance.setInboundCode(inboundCode);
+        balance.setInboundCode(inboundCode); balance.setSkuCode("SKU-TRANSFER");
         return balance;
     }