package cn.iocoder.yudao.module.mes.service.wm.productproduce;
|
|
import cn.hutool.core.collection.ListUtil;
|
import cn.iocoder.yudao.framework.common.exception.ServiceException;
|
import cn.iocoder.yudao.framework.test.core.ut.BaseDbUnitTest;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.wm.productproduce.MesWmProductProduceDO;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.wm.productproduce.MesWmProductProduceDetailDO;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.wm.productproduce.MesWmProductProduceLineDO;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.wm.warehouse.MesWmWarehouseAreaDO;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.wm.warehouse.MesWmWarehouseDO;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.wm.warehouse.MesWmWarehouseLocationDO;
|
import cn.iocoder.yudao.module.mes.dal.mysql.wm.productproduce.MesWmProductProduceMapper;
|
import cn.iocoder.yudao.module.mes.enums.wm.MesWmProductProduceStatusEnum;
|
import cn.iocoder.yudao.module.mes.enums.wm.MesWmQualityStatusEnum;
|
import cn.iocoder.yudao.module.mes.service.pro.workorder.MesProWorkOrderService;
|
import cn.iocoder.yudao.module.mes.service.wm.batch.MesWmBatchService;
|
import cn.iocoder.yudao.module.mes.service.wm.transaction.MesWmTransactionService;
|
import cn.iocoder.yudao.module.mes.service.wm.warehouse.MesWmWarehouseAreaService;
|
import cn.iocoder.yudao.module.mes.service.wm.warehouse.MesWmWarehouseLocationService;
|
import cn.iocoder.yudao.module.mes.service.wm.warehouse.MesWmWarehouseService;
|
import jakarta.annotation.Resource;
|
import org.junit.jupiter.api.Test;
|
import org.mockito.ArgumentCaptor;
|
import org.springframework.context.annotation.Import;
|
import org.springframework.test.context.bean.override.mockito.MockitoBean;
|
|
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 cn.iocoder.yudao.module.mes.enums.ErrorCodeConstants.*;
|
import static org.junit.jupiter.api.Assertions.*;
|
import static org.mockito.Mockito.*;
|
|
/**
|
* {@link MesWmProductProduceServiceImpl} 的单元测试
|
*
|
* @author 芋道源码
|
*/
|
@Import(MesWmProductProduceServiceImpl.class)
|
public class MesWmProductProduceServiceImplTest extends BaseDbUnitTest {
|
|
@Resource
|
private MesWmProductProduceServiceImpl productProduceService;
|
|
@Resource
|
private MesWmProductProduceMapper productProduceMapper;
|
|
@MockitoBean
|
private MesWmProductProduceLineService productProduceLineService;
|
@MockitoBean
|
private MesWmProductProduceDetailService productProduceDetailService;
|
@MockitoBean
|
private MesProWorkOrderService workOrderService;
|
@MockitoBean
|
private MesWmBatchService batchService;
|
@MockitoBean
|
private MesWmTransactionService wmTransactionService;
|
@MockitoBean
|
private MesWmWarehouseService warehouseService;
|
@MockitoBean
|
private MesWmWarehouseLocationService locationService;
|
@MockitoBean
|
private MesWmWarehouseAreaService areaService;
|
|
@Test
|
public void testSplitPendingAndFinishProduce_withUnqualified() {
|
// 准备数据：插入一个 PREPARE 状态的产出单
|
Long feedbackId = randomLongId();
|
Long sourceLineId = 100L;
|
Long itemId = randomLongId();
|
Long batchId = randomLongId();
|
String batchCode = "BATCH-001";
|
|
MesWmProductProduceDO produce = randomPojo(MesWmProductProduceDO.class, o -> {
|
o.setFeedbackId(feedbackId);
|
o.setStatus(MesWmProductProduceStatusEnum.PREPARE.getStatus());
|
});
|
productProduceMapper.insert(produce);
|
|
// mock: 虚拟线边库
|
Long whId = randomLongId(), locId = randomLongId(), areaId = randomLongId();
|
MesWmWarehouseDO wh = new MesWmWarehouseDO(); wh.setId(whId);
|
MesWmWarehouseLocationDO loc = new MesWmWarehouseLocationDO(); loc.setId(locId);
|
MesWmWarehouseAreaDO area = new MesWmWarehouseAreaDO(); area.setId(areaId);
|
when(warehouseService.getWarehouseByCode(MesWmWarehouseDO.WIP_VIRTUAL_WAREHOUSE)).thenReturn(wh);
|
when(locationService.getWarehouseLocationByCode(MesWmWarehouseLocationDO.WIP_VIRTUAL_LOCATION)).thenReturn(loc);
|
when(areaService.getWarehouseAreaByCode(MesWmWarehouseAreaDO.WIP_VIRTUAL_AREA)).thenReturn(area);
|
|
// mock: 通过 sourceLineId 直接定位 PENDING 产出行
|
MesWmProductProduceLineDO pendingLine = MesWmProductProduceLineDO.builder()
|
.id(sourceLineId).produceId(produce.getId()).feedbackId(feedbackId)
|
.itemId(itemId).quantity(BigDecimal.valueOf(100))
|
.batchId(batchId).batchCode(batchCode)
|
.qualityStatus(MesWmQualityStatusEnum.PENDING.getStatus())
|
.build();
|
when(productProduceLineService.validateProductProduceLineExists(sourceLineId))
|
.thenReturn(pendingLine);
|
|
// mock: finishProductProduce 内部查行和明细（用于创建库存事务）
|
MesWmProductProduceLineDO qualifiedLine = MesWmProductProduceLineDO.builder()
|
.id(100L).produceId(produce.getId()).itemId(itemId)
|
.quantity(BigDecimal.valueOf(80))
|
.qualityStatus(MesWmQualityStatusEnum.PASS.getStatus())
|
.build();
|
MesWmProductProduceLineDO unqualifiedLine = MesWmProductProduceLineDO.builder()
|
.id(101L).produceId(produce.getId()).itemId(itemId)
|
.quantity(BigDecimal.valueOf(20))
|
.qualityStatus(MesWmQualityStatusEnum.FAIL.getStatus())
|
.build();
|
when(productProduceLineService.getProductProduceLineListByProduceId(produce.getId()))
|
.thenReturn(ListUtil.of(qualifiedLine, unqualifiedLine));
|
|
// mock: finishProductProduce 中按行查明细
|
MesWmProductProduceDetailDO qualifiedDetail = MesWmProductProduceDetailDO.builder()
|
.lineId(100L).quantity(BigDecimal.valueOf(80)).build();
|
MesWmProductProduceDetailDO unqualifiedDetail = MesWmProductProduceDetailDO.builder()
|
.lineId(101L).quantity(BigDecimal.valueOf(20)).build();
|
when(productProduceDetailService.getProductProduceDetailListByLineId(100L))
|
.thenReturn(ListUtil.of(qualifiedDetail));
|
when(productProduceDetailService.getProductProduceDetailListByLineId(101L))
|
.thenReturn(ListUtil.of(unqualifiedDetail));
|
|
// 调用
|
productProduceService.splitPendingAndFinishProduce(feedbackId, sourceLineId,
|
BigDecimal.valueOf(80), BigDecimal.valueOf(20));
|
|
// 断言 1：不合格品行 - 新建了一行
|
ArgumentCaptor<MesWmProductProduceLineDO> lineCaptor = ArgumentCaptor.forClass(MesWmProductProduceLineDO.class);
|
verify(productProduceLineService).createProductProduceLine(lineCaptor.capture());
|
MesWmProductProduceLineDO createdLine = lineCaptor.getValue();
|
assertEquals(MesWmQualityStatusEnum.FAIL.getStatus(), createdLine.getQualityStatus());
|
assertEquals(0, BigDecimal.valueOf(20).compareTo(createdLine.getQuantity()));
|
assertEquals(itemId, createdLine.getItemId());
|
|
// 断言 2：原待检行 - 更新为合格品
|
ArgumentCaptor<MesWmProductProduceLineDO> updateCaptor = ArgumentCaptor.forClass(MesWmProductProduceLineDO.class);
|
verify(productProduceLineService).updateProductProduceLine(updateCaptor.capture());
|
MesWmProductProduceLineDO updatedLine = updateCaptor.getValue();
|
assertEquals(MesWmQualityStatusEnum.PASS.getStatus(), updatedLine.getQualityStatus());
|
assertEquals(0, BigDecimal.valueOf(80).compareTo(updatedLine.getQuantity()));
|
|
// 断言 3：生成了 2 条明细（不合格品明细 + 合格品明细）
|
verify(productProduceDetailService, times(2)).createProductProduceDetail(any());
|
|
// 断言 4：产出单状态更新为已完成
|
MesWmProductProduceDO updatedProduce = productProduceMapper.selectById(produce.getId());
|
assertEquals(MesWmProductProduceStatusEnum.FINISHED.getStatus(), updatedProduce.getStatus());
|
|
// 断言 5：创建了库存事务
|
verify(wmTransactionService, times(2)).createTransaction(any());
|
}
|
|
@Test
|
public void testSplitPendingAndFinishProduce_allQualified() {
|
// 准备数据：全部合格品场景
|
Long feedbackId = randomLongId();
|
Long sourceLineId = 200L;
|
Long itemId = randomLongId();
|
|
MesWmProductProduceDO produce = randomPojo(MesWmProductProduceDO.class, o -> {
|
o.setFeedbackId(feedbackId);
|
o.setStatus(MesWmProductProduceStatusEnum.PREPARE.getStatus());
|
});
|
productProduceMapper.insert(produce);
|
|
// mock: 虚拟线边库
|
Long whId = randomLongId(), locId = randomLongId(), areaId = randomLongId();
|
MesWmWarehouseDO wh = new MesWmWarehouseDO(); wh.setId(whId);
|
MesWmWarehouseLocationDO loc = new MesWmWarehouseLocationDO(); loc.setId(locId);
|
MesWmWarehouseAreaDO area = new MesWmWarehouseAreaDO(); area.setId(areaId);
|
when(warehouseService.getWarehouseByCode(any())).thenReturn(wh);
|
when(locationService.getWarehouseLocationByCode(any())).thenReturn(loc);
|
when(areaService.getWarehouseAreaByCode(any())).thenReturn(area);
|
|
// mock: 通过 sourceLineId 直接定位 PENDING 产出行
|
MesWmProductProduceLineDO pendingLine = MesWmProductProduceLineDO.builder()
|
.id(sourceLineId).produceId(produce.getId()).feedbackId(feedbackId)
|
.itemId(itemId).quantity(BigDecimal.valueOf(50))
|
.qualityStatus(MesWmQualityStatusEnum.PENDING.getStatus())
|
.build();
|
when(productProduceLineService.validateProductProduceLineExists(sourceLineId))
|
.thenReturn(pendingLine);
|
|
// mock: finishProductProduce 查行和明细
|
MesWmProductProduceLineDO qualifiedLine = MesWmProductProduceLineDO.builder()
|
.id(200L).produceId(produce.getId()).itemId(itemId)
|
.quantity(BigDecimal.valueOf(50))
|
.qualityStatus(MesWmQualityStatusEnum.PASS.getStatus())
|
.build();
|
when(productProduceLineService.getProductProduceLineListByProduceId(produce.getId()))
|
.thenReturn(ListUtil.of(qualifiedLine));
|
MesWmProductProduceDetailDO detail = MesWmProductProduceDetailDO.builder()
|
.lineId(200L).quantity(BigDecimal.valueOf(50)).build();
|
when(productProduceDetailService.getProductProduceDetailListByLineId(200L))
|
.thenReturn(ListUtil.of(detail));
|
|
// 调用：不合格品数量 = 0
|
productProduceService.splitPendingAndFinishProduce(feedbackId, sourceLineId,
|
BigDecimal.valueOf(50), BigDecimal.ZERO);
|
|
// 断言 1：没有新建行（不合格品数量为 0）
|
verify(productProduceLineService, never()).createProductProduceLine(any());
|
|
// 断言 2：更新了原待检行为合格
|
ArgumentCaptor<MesWmProductProduceLineDO> updateCaptor = ArgumentCaptor.forClass(MesWmProductProduceLineDO.class);
|
verify(productProduceLineService).updateProductProduceLine(updateCaptor.capture());
|
assertEquals(MesWmQualityStatusEnum.PASS.getStatus(), updateCaptor.getValue().getQualityStatus());
|
|
// 断言 3：只生成了 1 条明细（全部合格品）
|
verify(productProduceDetailService, times(1)).createProductProduceDetail(any());
|
|
// 断言 4：产出单状态更新为已完成
|
MesWmProductProduceDO updatedProduce = productProduceMapper.selectById(produce.getId());
|
assertEquals(MesWmProductProduceStatusEnum.FINISHED.getStatus(), updatedProduce.getStatus());
|
}
|
|
@Test
|
public void testSplitPendingAndFinishProduce_produceNotExists() {
|
// 调用不存在的 feedbackId，应该抛异常
|
Long feedbackId = randomLongId();
|
Long sourceLineId = randomLongId();
|
ServiceException ex = assertThrows(ServiceException.class, () ->
|
productProduceService.splitPendingAndFinishProduce(feedbackId, sourceLineId,
|
BigDecimal.TEN, BigDecimal.ZERO));
|
assertEquals(WM_PRODUCT_PRODUCE_NOT_EXISTS.getCode(), ex.getCode());
|
}
|
|
@Test
|
public void testSplitPendingAndFinishProduce_lineNotExists() {
|
// 准备数据：产出单存在，但 sourceLineId 对应的行不存在
|
Long feedbackId = randomLongId();
|
Long sourceLineId = randomLongId();
|
MesWmProductProduceDO produce = randomPojo(MesWmProductProduceDO.class, o -> {
|
o.setFeedbackId(feedbackId);
|
o.setStatus(MesWmProductProduceStatusEnum.PREPARE.getStatus());
|
});
|
productProduceMapper.insert(produce);
|
|
// mock: 虚拟线边库
|
MesWmWarehouseDO wh = new MesWmWarehouseDO(); wh.setId(1L);
|
MesWmWarehouseLocationDO loc = new MesWmWarehouseLocationDO(); loc.setId(1L);
|
MesWmWarehouseAreaDO area = new MesWmWarehouseAreaDO(); area.setId(1L);
|
when(warehouseService.getWarehouseByCode(any())).thenReturn(wh);
|
when(locationService.getWarehouseLocationByCode(any())).thenReturn(loc);
|
when(areaService.getWarehouseAreaByCode(any())).thenReturn(area);
|
|
// mock: validateProductProduceLineExists 抛异常
|
when(productProduceLineService.validateProductProduceLineExists(sourceLineId))
|
.thenThrow(new ServiceException(WM_PRODUCT_PRODUCE_LINE_NOT_EXISTS));
|
|
// 调用，应该抛异常
|
ServiceException ex = assertThrows(ServiceException.class, () ->
|
productProduceService.splitPendingAndFinishProduce(feedbackId, sourceLineId,
|
BigDecimal.TEN, BigDecimal.ZERO));
|
assertEquals(WM_PRODUCT_PRODUCE_LINE_NOT_EXISTS.getCode(), ex.getCode());
|
}
|
|
}
|
