云
2026-10-29 a1a94ab39a3ac9cc28601e3381b3b4fbc0cda02a
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
package cn.iocoder.yudao.module.mes.service.wm.returnsales;
 
import cn.iocoder.yudao.framework.common.exception.ServiceException;
import cn.iocoder.yudao.module.mes.dal.dataobject.mdm.MesMdmItemDO;
import cn.iocoder.yudao.module.mes.dal.dataobject.wm.returnsales.MesWmReturnSalesDO;
import cn.iocoder.yudao.module.mes.dal.dataobject.wm.returnsales.MesWmReturnSalesDetailDO;
import cn.iocoder.yudao.module.mes.dal.dataobject.wm.returnsales.MesWmReturnSalesLineDO;
import cn.iocoder.yudao.module.mes.dal.mysql.wm.returnsales.MesWmReturnSalesMapper;
import cn.iocoder.yudao.module.mes.enums.wm.MesWmReturnSalesStatusEnum;
import cn.iocoder.yudao.module.mes.service.mdm.MesMdmItemService;
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.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.ArgumentCaptor;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;
 
import java.math.BigDecimal;
import java.util.List;
 
import static org.junit.jupiter.api.Assertions.*;
import static org.mockito.ArgumentMatchers.*;
import static org.mockito.Mockito.*;
 
@ExtendWith(MockitoExtension.class)
class MesWmReturnSalesStockTest {
    @InjectMocks private MesWmReturnSalesServiceImpl service;
    @Mock private MesWmReturnSalesMapper mapper;
    @Mock private MesWmReturnSalesLineService lineService;
    @Mock private MesWmReturnSalesDetailService detailService;
    @Mock private MesMdmItemService itemService;
    @Mock private MesWmTransactionService transactionService;
 
    private void setup(BigDecimal quantity) {
        when(mapper.selectById(1L)).thenReturn(MesWmReturnSalesDO.builder().id(1L).code("TEST")
                .status(MesWmReturnSalesStatusEnum.APPROVED.getStatus()).build());
        when(lineService.getReturnSalesLineListByReturnId(1L)).thenReturn(List.of(
                MesWmReturnSalesLineDO.builder().id(2L).itemId(3L).quantity(BigDecimal.ONE).build()));
        when(detailService.getReturnSalesDetailListByReturnId(1L)).thenReturn(List.of(
                MesWmReturnSalesDetailDO.builder().lineId(2L).itemId(3L).quantity(quantity)
                        .batchId(4L).batchCode("TEST-BATCH").warehouseId(5L).build()));
    }
 
    @Test void rejectsExcessStocking() {
        setup(BigDecimal.TWO);
        when(itemService.validateItemExistsAndEnable(3L)).thenReturn(MesMdmItemDO.builder().code("TEST").name("test").build());
        assertThrows(ServiceException.class, () -> service.stockReturnSales(1L));
        verifyNoInteractions(transactionService);
    }
 
    @Test void preservesBatchOnStockTransaction() {
        setup(BigDecimal.ONE);
        service.stockReturnSales(1L);
        ArgumentCaptor<List<MesWmTransactionSaveReqDTO>> captor = ArgumentCaptor.forClass(List.class);
        verify(transactionService).createTransactionList(captor.capture());
        assertEquals(4L, captor.getValue().get(0).getBatchId());
        assertEquals("TEST-BATCH", captor.getValue().get(0).getBatchCode());
        assertEquals(BigDecimal.ONE, captor.getValue().get(0).getQuantity());
    }
}