package cn.iocoder.yudao.module.mes.service.wm.transfer;
|
|
import cn.iocoder.yudao.module.mes.dal.dataobject.wm.materialstock.MesWmMaterialStockDO;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.wm.transfer.MesWmTransferDO;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.wm.transfer.MesWmTransferDetailDO;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.wm.transfer.MesWmTransferLineDO;
|
import cn.iocoder.yudao.module.mes.service.wm.materialstock.MesWmMaterialStockService;
|
import cn.iocoder.yudao.module.mes.service.wm.transaction.MesWmTransactionService;
|
import cn.iocoder.yudao.module.mes.service.wm.transaction.dto.MesWmTransactionSaveReqDTO;
|
import org.junit.jupiter.api.BeforeEach;
|
import org.junit.jupiter.api.Test;
|
import org.junit.jupiter.api.extension.ExtendWith;
|
import org.mockito.ArgumentCaptor;
|
import org.mockito.Mock;
|
import org.mockito.junit.jupiter.MockitoExtension;
|
import org.springframework.test.util.ReflectionTestUtils;
|
|
import java.math.BigDecimal;
|
import java.util.List;
|
|
import static org.junit.jupiter.api.Assertions.assertEquals;
|
import static org.mockito.Mockito.*;
|
|
@ExtendWith(MockitoExtension.class)
|
class MesWmTransferVendorTraceTest {
|
|
private final MesWmTransferServiceImpl service = new MesWmTransferServiceImpl();
|
@Mock private MesWmTransferDetailService detailService;
|
@Mock private MesWmMaterialStockService stockService;
|
@Mock private MesWmTransactionService transactionService;
|
|
@BeforeEach
|
void setUp() {
|
ReflectionTestUtils.setField(service, "transferDetailService", detailService);
|
ReflectionTestUtils.setField(service, "materialStockService", stockService);
|
ReflectionTestUtils.setField(service, "wmTransactionService", transactionService);
|
}
|
|
@Test
|
void pairedMoveKeepsSourceVendor() {
|
MesWmTransferDO transfer = new MesWmTransferDO().setId(7L).setCode("MOVE-7");
|
MesWmTransferLineDO line = new MesWmTransferLineDO().setId(11L).setMaterialStockId(1032L)
|
.setItemId(1996L).setFromWarehouseId(24L).setFromLocationId(9L).setFromAreaId(9L);
|
MesWmTransferDetailDO detail = new MesWmTransferDetailDO().setLineId(11L)
|
.setQuantity(BigDecimal.ONE).setToWarehouseId(25L).setToLocationId(10L).setToAreaId(10L);
|
when(detailService.getTransferDetailListByTransferId(7L)).thenReturn(List.of(detail));
|
when(stockService.getMaterialStock(1032L)).thenReturn(new MesWmMaterialStockDO()
|
.setId(1032L).setVendorId(1L));
|
when(transactionService.createTransaction(any())).thenReturn(100L, 101L);
|
|
ReflectionTestUtils.invokeMethod(service, "createTransferTransactions", transfer, List.of(line));
|
|
ArgumentCaptor<MesWmTransactionSaveReqDTO> captor = ArgumentCaptor.forClass(MesWmTransactionSaveReqDTO.class);
|
verify(transactionService, times(2)).createTransaction(captor.capture());
|
List<MesWmTransactionSaveReqDTO> transactions = captor.getAllValues();
|
assertEquals(1L, transactions.get(0).getVendorId());
|
assertEquals(1L, transactions.get(1).getVendorId());
|
assertEquals(100L, transactions.get(1).getRelatedTransactionId());
|
assertEquals(BigDecimal.ONE.negate(), transactions.get(0).getQuantity());
|
assertEquals(BigDecimal.ONE, transactions.get(1).getQuantity());
|
}
|
}
|
