package cn.iocoder.yudao.module.mes.service.wm.stocktaking.task;

import cn.iocoder.yudao.module.mes.dal.dataobject.wm.stocktaking.task.MesWmStockTakingTaskDO;
import cn.iocoder.yudao.module.mes.dal.dataobject.wm.stocktaking.task.MesWmStockTakingTaskResultDO;
import cn.iocoder.yudao.module.mes.dal.mysql.wm.stocktaking.task.MesWmStockTakingTaskMapper;
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 MesWmStockTakingDifferenceTest {

    private final MesWmStockTakingTaskServiceImpl service = new MesWmStockTakingTaskServiceImpl();
    @Mock private MesWmStockTakingTaskMapper taskMapper;
    @Mock private MesWmStockTakingTaskResultService resultService;
    @Mock private MesWmTransactionService transactionService;

    @BeforeEach
    void setUp() {
        ReflectionTestUtils.setField(service, "stockTakingTaskMapper", taskMapper);
        ReflectionTestUtils.setField(service, "stockTakingTaskResultService", resultService);
        ReflectionTestUtils.setField(service, "wmTransactionService", transactionService);
    }

    @Test
    void completedTaskPostsGainAndLossOnce() {
        when(taskMapper.selectById(1L)).thenReturn(new MesWmStockTakingTaskDO()
                .setId(1L).setCode("ST-1").setStatus(4));
        when(taskMapper.claimDifferenceProcessing(1L)).thenReturn(1, 0);
        when(resultService.getStockTakingTaskResultListByTaskId(1L)).thenReturn(List.of(
                result(11L, BigDecimal.ONE, BigDecimal.valueOf(2)),
                result(12L, BigDecimal.valueOf(3), BigDecimal.ONE)));

        service.processStockTakingDifference(1L);
        service.processStockTakingDifference(1L);

        ArgumentCaptor<MesWmTransactionSaveReqDTO> captor = ArgumentCaptor.forClass(MesWmTransactionSaveReqDTO.class);
        verify(transactionService, times(2)).createTransaction(captor.capture());
        List<MesWmTransactionSaveReqDTO> transactions = captor.getAllValues();
        assertEquals(1, transactions.get(0).getType());
        assertEquals(BigDecimal.ONE, transactions.get(0).getQuantity());
        assertEquals(11L, transactions.get(0).getMaterialStockId());
        assertEquals(2, transactions.get(1).getType());
        assertEquals(BigDecimal.valueOf(-2), transactions.get(1).getQuantity());
        assertEquals(12L, transactions.get(1).getMaterialStockId());
    }

    private MesWmStockTakingTaskResultDO result(Long stockId, BigDecimal book, BigDecimal actual) {
        return new MesWmStockTakingTaskResultDO().setTaskId(1L).setLineId(stockId)
                .setMaterialStockId(stockId).setItemId(1996L).setQuantity(book)
                .setTakingQuantity(actual).setWarehouseId(21L).setLocationId(6L).setAreaId(6L);
    }
}