2026-06-26 20b96473f2520590a0dca6b775b81e3ea06a77a0
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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
package cn.iocoder.yudao.module.mes.service.qc.iqc;
 
import cn.iocoder.yudao.framework.test.core.ut.BaseDbUnitTest;
import cn.iocoder.yudao.module.mes.dal.dataobject.qc.iqc.MesQcIqcDO;
import cn.iocoder.yudao.module.mes.dal.mysql.qc.iqc.MesQcIqcMapper;
import cn.iocoder.yudao.module.mes.enums.MesBizTypeConstants;
import cn.iocoder.yudao.module.mes.enums.qc.MesQcStatusEnum;
import cn.iocoder.yudao.module.mes.service.md.item.MesMdItemService;
import cn.iocoder.yudao.module.mes.service.md.vendor.MesMdVendorService;
import cn.iocoder.yudao.module.mes.service.qc.defectrecord.MesQcDefectRecordService;
import cn.iocoder.yudao.module.mes.service.qc.template.MesQcTemplateItemService;
import cn.iocoder.yudao.module.mes.service.wm.arrivalnotice.MesWmArrivalNoticeService;
import cn.iocoder.yudao.module.mes.service.wm.outsourcereceipt.MesWmOutsourceReceiptService;
import cn.iocoder.yudao.module.mes.service.qc.indicatorresult.MesQcIndicatorResultService;
import cn.iocoder.yudao.module.system.api.user.AdminUserApi;
import org.junit.jupiter.api.Test;
import org.springframework.context.annotation.Import;
import org.springframework.test.context.bean.override.mockito.MockitoBean;
 
import jakarta.annotation.Resource;
 
import java.math.BigDecimal;
 
import static cn.iocoder.yudao.framework.test.core.util.RandomUtils.randomLongId;
import static cn.iocoder.yudao.framework.test.core.util.RandomUtils.randomPojo;
import static org.junit.jupiter.api.Assertions.*;
import static org.mockito.Mockito.*;
 
/**
 * {@link MesQcIqcServiceImpl} 的单元测试
 *
 * @author 芋道源码
 */
@Import(MesQcIqcServiceImpl.class)
public class MesQcIqcServiceImplTest extends BaseDbUnitTest {
 
    @Resource
    private MesQcIqcServiceImpl iqcService;
 
    @Resource
    private MesQcIqcMapper iqcMapper;
 
    @MockitoBean
    private MesWmArrivalNoticeService arrivalNoticeService;
    @MockitoBean
    private MesWmOutsourceReceiptService outsourceReceiptService;
    @MockitoBean
    private MesQcIqcLineService iqcLineService;
    @MockitoBean
    private MesQcDefectRecordService defectRecordService;
    @MockitoBean
    private MesMdVendorService vendorService;
    @MockitoBean
    private MesMdItemService itemService;
    @MockitoBean
    private MesQcTemplateItemService templateItemService;
    @MockitoBean
    private AdminUserApi adminUserApi;
    @MockitoBean
    private MesQcIndicatorResultService indicatorResultService;
 
    @Test
    public void testFinishIqc_writeBack_arrivalNotice() {
        // 准备数据:插入一条草稿状态的 IQC 单,来源为到货通知单
        Long sourceDocId = randomLongId();
        Long sourceLineId = randomLongId();
        MesQcIqcDO iqc = randomPojo(MesQcIqcDO.class, o -> {
            o.setStatus(MesQcStatusEnum.DRAFT.getStatus());
            o.setCheckResult(1); // 非空,满足 finishIqc 的校验
            o.setSourceDocType(MesBizTypeConstants.WM_ARRIVAL_NOTICE);
            o.setSourceDocId(sourceDocId);
            o.setSourceLineId(sourceLineId);
            o.setQualifiedQuantity(BigDecimal.valueOf(100));
            o.setUnqualifiedQuantity(BigDecimal.ZERO);
        });
        iqcMapper.insert(iqc);
 
        // 调用
        iqcService.finishIqc(iqc.getId());
 
        // 断言:验证状态更新为已完成
        MesQcIqcDO updatedIqc = iqcMapper.selectById(iqc.getId());
        assertEquals(MesQcStatusEnum.FINISHED.getStatus(), updatedIqc.getStatus());
        // 断言:验证回写到货通知单(参数使用 any 避免 BigDecimal 精度问题)
        verify(arrivalNoticeService).updateArrivalNoticeWhenIqcFinish(
                eq(sourceDocId), eq(sourceLineId), eq(iqc.getId()), any(BigDecimal.class));
        // 断言:不应该调用外协入库单的回写
        verify(outsourceReceiptService, never()).updateOutsourceReceiptWhenIqcFinish(
                anyLong(), anyLong(), anyLong(), any(), any());
    }
 
    @Test
    public void testFinishIqc_writeBack_outsourceReceipt() {
        // 准备数据:插入一条草稿状态的 IQC 单,来源为外协入库单
        Long sourceDocId = randomLongId();
        Long sourceLineId = randomLongId();
        MesQcIqcDO iqc = randomPojo(MesQcIqcDO.class, o -> {
            o.setStatus(MesQcStatusEnum.DRAFT.getStatus());
            o.setCheckResult(1);
            o.setSourceDocType(MesBizTypeConstants.WM_OUTSOURCE_RECPT);
            o.setSourceDocId(sourceDocId);
            o.setSourceLineId(sourceLineId);
            o.setQualifiedQuantity(BigDecimal.valueOf(80));
            o.setUnqualifiedQuantity(BigDecimal.valueOf(20));
        });
        iqcMapper.insert(iqc);
 
        // 调用
        iqcService.finishIqc(iqc.getId());
 
        // 断言:验证状态更新为已完成
        MesQcIqcDO updatedIqc = iqcMapper.selectById(iqc.getId());
        assertEquals(MesQcStatusEnum.FINISHED.getStatus(), updatedIqc.getStatus());
        // 断言:验证回写外协入库单(参数使用 any 避免 BigDecimal 精度问题)
        verify(outsourceReceiptService).updateOutsourceReceiptWhenIqcFinish(
                eq(sourceDocId), eq(sourceLineId), eq(iqc.getId()), any(BigDecimal.class), any(BigDecimal.class));
        // 断言:不应该调用到货通知单的回写
        verify(arrivalNoticeService, never()).updateArrivalNoticeWhenIqcFinish(
                anyLong(), anyLong(), anyLong(), any());
    }
 
    @Test
    public void testFinishIqc_writeBack_noSourceDoc() {
        // 准备数据:sourceDocType 为空,不需要回写
        MesQcIqcDO iqc = randomPojo(MesQcIqcDO.class, o -> {
            o.setStatus(MesQcStatusEnum.DRAFT.getStatus());
            o.setCheckResult(1);
            o.setSourceDocType(null);
            o.setSourceDocId(null);
            o.setSourceLineId(null);
        });
        iqcMapper.insert(iqc);
 
        // 调用
        iqcService.finishIqc(iqc.getId());
 
        // 断言:验证状态更新为已完成
        MesQcIqcDO updatedIqc = iqcMapper.selectById(iqc.getId());
        assertEquals(MesQcStatusEnum.FINISHED.getStatus(), updatedIqc.getStatus());
        // 断言:不应该调用任何回写
        verify(arrivalNoticeService, never()).updateArrivalNoticeWhenIqcFinish(
                anyLong(), anyLong(), anyLong(), any());
        verify(outsourceReceiptService, never()).updateOutsourceReceiptWhenIqcFinish(
                anyLong(), anyLong(), anyLong(), any(), any());
    }
 
    @Test
    public void testFinishIqc_writeBack_sourceDocTypeDirtyData() {
        // 准备数据:sourceDocType 非空但 sourceLineId 为空,应该抛异常
        MesQcIqcDO iqc = randomPojo(MesQcIqcDO.class, o -> {
            o.setStatus(MesQcStatusEnum.DRAFT.getStatus());
            o.setCheckResult(1);
            o.setSourceDocType(MesBizTypeConstants.WM_ARRIVAL_NOTICE);
            o.setSourceDocId(randomLongId());
            o.setSourceLineId(null); // 脏数据
        });
        iqcMapper.insert(iqc);
 
        // 调用,并断言异常
        assertThrows(IllegalArgumentException.class, () -> iqcService.finishIqc(iqc.getId()));
    }
 
    @Test
    public void testFinishIqc_writeBack_unknownSourceDocType() {
        // 准备数据:sourceDocType 无法识别
        MesQcIqcDO iqc = randomPojo(MesQcIqcDO.class, o -> {
            o.setStatus(MesQcStatusEnum.DRAFT.getStatus());
            o.setCheckResult(1);
            o.setSourceDocType(99999); // 无法识别的类型
            o.setSourceDocId(randomLongId());
            o.setSourceLineId(randomLongId());
        });
        iqcMapper.insert(iqc);
 
        // 调用,并断言异常
        assertThrows(IllegalArgumentException.class, () -> iqcService.finishIqc(iqc.getId()));
    }
 
    @Test
    public void testFinishIqc_checkResultEmpty() {
        // 准备数据:checkResult 为空,应该报错
        MesQcIqcDO iqc = randomPojo(MesQcIqcDO.class, o -> {
            o.setStatus(MesQcStatusEnum.DRAFT.getStatus());
            o.setCheckResult(null); // 检测结论为空
            o.setSourceDocType(null);
        });
        iqcMapper.insert(iqc);
 
        // 调用,应该抛异常(检测结论必填)
        assertThrows(Exception.class, () -> iqcService.finishIqc(iqc.getId()));
    }
 
    @Test
    public void testFinishIqc_notDraftStatus() {
        // 准备数据:已完成状态,不能再次完成
        MesQcIqcDO iqc = randomPojo(MesQcIqcDO.class, o -> {
            o.setStatus(MesQcStatusEnum.FINISHED.getStatus());
            o.setCheckResult(1);
        });
        iqcMapper.insert(iqc);
 
        // 调用,应该抛异常(不是草稿状态)
        assertThrows(Exception.class, () -> iqcService.finishIqc(iqc.getId()));
    }
 
}