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());
    }
}