package cn.iocoder.yudao.module.mes.dal.mysql.pro.task;
|
|
import cn.hutool.core.collection.ListUtil;
|
import cn.iocoder.yudao.framework.common.pojo.PageResult;
|
import cn.iocoder.yudao.framework.test.core.ut.BaseDbUnitTest;
|
import cn.iocoder.yudao.module.mes.controller.admin.pro.task.vo.MesProTaskPageReqVO;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.pro.route.MesProRouteProcessDO;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.pro.task.MesProTaskDO;
|
import cn.iocoder.yudao.module.mes.dal.mysql.pro.route.MesProRouteProcessMapper;
|
import jakarta.annotation.Resource;
|
import org.junit.jupiter.api.Test;
|
|
import java.math.BigDecimal;
|
import java.util.List;
|
import java.util.function.Consumer;
|
|
import static cn.iocoder.yudao.framework.common.util.object.ObjectUtils.cloneIgnoreId;
|
import static cn.iocoder.yudao.framework.test.core.util.AssertUtils.assertPojoEquals;
|
import static cn.iocoder.yudao.framework.test.core.util.RandomUtils.randomPojo;
|
import static org.junit.jupiter.api.Assertions.assertEquals;
|
|
/**
|
* {@link MesProTaskMapper#selectPage(MesProTaskPageReqVO)} 的单元测试
|
*
|
* @author 芋道源码
|
*/
|
public class MesProTaskMapperTest extends BaseDbUnitTest {
|
|
@Resource
|
private MesProTaskMapper taskMapper;
|
|
@Resource
|
private MesProRouteProcessMapper routeProcessMapper;
|
|
/**
|
* 创建 Task 时统一设置 BigDecimal 字段为 2 位小数，避免 H2 decimal(14,2) 精度不匹配
|
*/
|
private MesProTaskDO createTaskPojo(Consumer<MesProTaskDO> consumer) {
|
return randomPojo(MesProTaskDO.class, o -> {
|
o.setQuantity(new BigDecimal("10.00"));
|
o.setProducedQuantity(new BigDecimal("5.00"));
|
o.setQualifyQuantity(new BigDecimal("4.00"));
|
o.setUnqualifyQuantity(new BigDecimal("1.00"));
|
o.setChangedQuantity(new BigDecimal("0.00"));
|
consumer.accept(o);
|
});
|
}
|
|
// ==================== selectPage 基础过滤 ====================
|
|
@Test
|
public void testSelectPage_byCode() {
|
MesProTaskDO matchTask = createTaskPojo(o -> {
|
o.setCode("PT202501010001");
|
o.setStatus(0);
|
});
|
taskMapper.insert(matchTask);
|
taskMapper.insert(cloneIgnoreId(matchTask, o -> o.setCode("OTHER_CODE")));
|
|
MesProTaskPageReqVO reqVO = new MesProTaskPageReqVO();
|
reqVO.setCode("PT2025");
|
PageResult<MesProTaskDO> result = taskMapper.selectPage(reqVO);
|
|
assertEquals(1, result.getTotal());
|
assertPojoEquals(matchTask, result.getList().get(0));
|
}
|
|
@Test
|
public void testSelectPage_byName() {
|
MesProTaskDO matchTask = createTaskPojo(o -> {
|
o.setName("注塑任务-A");
|
o.setStatus(0);
|
});
|
taskMapper.insert(matchTask);
|
taskMapper.insert(cloneIgnoreId(matchTask, o -> o.setName("OTHER_NAME")));
|
|
MesProTaskPageReqVO reqVO = new MesProTaskPageReqVO();
|
reqVO.setName("注塑");
|
PageResult<MesProTaskDO> result = taskMapper.selectPage(reqVO);
|
|
assertEquals(1, result.getTotal());
|
assertPojoEquals(matchTask, result.getList().get(0));
|
}
|
|
@Test
|
public void testSelectPage_byWorkOrderId() {
|
MesProTaskDO matchTask = createTaskPojo(o -> {
|
o.setWorkOrderId(100L);
|
o.setStatus(0);
|
});
|
taskMapper.insert(matchTask);
|
taskMapper.insert(cloneIgnoreId(matchTask, o -> o.setWorkOrderId(999L)));
|
|
MesProTaskPageReqVO reqVO = new MesProTaskPageReqVO();
|
reqVO.setWorkOrderId(100L);
|
PageResult<MesProTaskDO> result = taskMapper.selectPage(reqVO);
|
|
assertEquals(1, result.getTotal());
|
assertPojoEquals(matchTask, result.getList().get(0));
|
}
|
|
@Test
|
public void testSelectPage_byStatus() {
|
MesProTaskDO matchTask = createTaskPojo(o -> o.setStatus(1));
|
taskMapper.insert(matchTask);
|
taskMapper.insert(cloneIgnoreId(matchTask, o -> o.setStatus(2)));
|
|
MesProTaskPageReqVO reqVO = new MesProTaskPageReqVO();
|
reqVO.setStatus(1);
|
PageResult<MesProTaskDO> result = taskMapper.selectPage(reqVO);
|
|
assertEquals(1, result.getTotal());
|
assertPojoEquals(matchTask, result.getList().get(0));
|
}
|
|
@Test
|
public void testSelectPage_byStatuses() {
|
MesProTaskDO task1 = createTaskPojo(o -> o.setStatus(1));
|
MesProTaskDO task2 = createTaskPojo(o -> o.setStatus(2));
|
MesProTaskDO task3 = createTaskPojo(o -> o.setStatus(3));
|
taskMapper.insert(task1);
|
taskMapper.insert(task2);
|
taskMapper.insert(task3);
|
|
MesProTaskPageReqVO reqVO = new MesProTaskPageReqVO();
|
reqVO.setStatuses(ListUtil.of(1, 2));
|
PageResult<MesProTaskDO> result = taskMapper.selectPage(reqVO);
|
|
assertEquals(2, result.getTotal());
|
}
|
|
@Test
|
public void testSelectPage_noFilter() {
|
MesProTaskDO task1 = createTaskPojo(o -> o.setStatus(0));
|
MesProTaskDO task2 = createTaskPojo(o -> o.setStatus(1));
|
taskMapper.insert(task1);
|
taskMapper.insert(task2);
|
|
MesProTaskPageReqVO reqVO = new MesProTaskPageReqVO();
|
PageResult<MesProTaskDO> result = taskMapper.selectPage(reqVO);
|
|
assertEquals(2, result.getTotal());
|
}
|
|
// ==================== selectPage + checkFlag (MPJ LEFT JOIN) ====================
|
|
@Test
|
public void testSelectPage_checkFlagTrue() {
|
Long routeId = 10L;
|
Long processId1 = 100L;
|
Long processId2 = 200L;
|
|
MesProTaskDO task1 = createTaskPojo(o -> {
|
o.setRouteId(routeId);
|
o.setProcessId(processId1);
|
o.setStatus(0);
|
});
|
MesProTaskDO task2 = createTaskPojo(o -> {
|
o.setRouteId(routeId);
|
o.setProcessId(processId2);
|
o.setStatus(0);
|
});
|
taskMapper.insert(task1);
|
taskMapper.insert(task2);
|
|
routeProcessMapper.insert(randomPojo(MesProRouteProcessDO.class, o -> {
|
o.setRouteId(routeId);
|
o.setProcessId(processId1);
|
o.setCheckFlag(true);
|
}));
|
routeProcessMapper.insert(randomPojo(MesProRouteProcessDO.class, o -> {
|
o.setRouteId(routeId);
|
o.setProcessId(processId2);
|
o.setCheckFlag(false);
|
}));
|
|
// 查询：checkFlag = true，只返回 task1
|
MesProTaskPageReqVO reqVO = new MesProTaskPageReqVO();
|
reqVO.setCheckFlag(true);
|
PageResult<MesProTaskDO> result = taskMapper.selectPage(reqVO);
|
|
assertEquals(1, result.getTotal());
|
assertPojoEquals(task1, result.getList().get(0));
|
}
|
|
@Test
|
public void testSelectPage_checkFlagFalse() {
|
Long routeId = 20L;
|
Long processId1 = 300L;
|
Long processId2 = 400L;
|
|
MesProTaskDO task1 = createTaskPojo(o -> {
|
o.setRouteId(routeId);
|
o.setProcessId(processId1);
|
o.setStatus(0);
|
});
|
MesProTaskDO task2 = createTaskPojo(o -> {
|
o.setRouteId(routeId);
|
o.setProcessId(processId2);
|
o.setStatus(0);
|
});
|
taskMapper.insert(task1);
|
taskMapper.insert(task2);
|
|
routeProcessMapper.insert(randomPojo(MesProRouteProcessDO.class, o -> {
|
o.setRouteId(routeId);
|
o.setProcessId(processId1);
|
o.setCheckFlag(true);
|
}));
|
routeProcessMapper.insert(randomPojo(MesProRouteProcessDO.class, o -> {
|
o.setRouteId(routeId);
|
o.setProcessId(processId2);
|
o.setCheckFlag(false);
|
}));
|
|
// 查询：checkFlag = false，只返回 task2
|
MesProTaskPageReqVO reqVO = new MesProTaskPageReqVO();
|
reqVO.setCheckFlag(false);
|
PageResult<MesProTaskDO> result = taskMapper.selectPage(reqVO);
|
|
assertEquals(1, result.getTotal());
|
assertPojoEquals(task2, result.getList().get(0));
|
}
|
|
@Test
|
public void testSelectPage_checkFlagWithOtherFilter() {
|
Long routeId = 30L;
|
Long processId = 500L;
|
|
MesProTaskDO taskMatch = createTaskPojo(o -> {
|
o.setRouteId(routeId);
|
o.setProcessId(processId);
|
o.setCode("PT-MATCH-001");
|
o.setStatus(0);
|
});
|
MesProTaskDO taskNoMatch = createTaskPojo(o -> {
|
o.setRouteId(routeId);
|
o.setProcessId(processId);
|
o.setCode("OTHER-CODE");
|
o.setStatus(0);
|
});
|
taskMapper.insert(taskMatch);
|
taskMapper.insert(taskNoMatch);
|
|
routeProcessMapper.insert(randomPojo(MesProRouteProcessDO.class, o -> {
|
o.setRouteId(routeId);
|
o.setProcessId(processId);
|
o.setCheckFlag(true);
|
}));
|
|
// 查询：checkFlag=true 且 code like 'PT-MATCH'
|
MesProTaskPageReqVO reqVO = new MesProTaskPageReqVO();
|
reqVO.setCheckFlag(true);
|
reqVO.setCode("PT-MATCH");
|
PageResult<MesProTaskDO> result = taskMapper.selectPage(reqVO);
|
|
assertEquals(1, result.getTotal());
|
assertPojoEquals(taskMatch, result.getList().get(0));
|
}
|
|
@Test
|
public void testSelectPage_checkFlagNull_noJoin() {
|
Long routeId = 40L;
|
Long processId = 600L;
|
|
MesProTaskDO task = createTaskPojo(o -> {
|
o.setRouteId(routeId);
|
o.setProcessId(processId);
|
o.setStatus(0);
|
});
|
taskMapper.insert(task);
|
|
MesProTaskPageReqVO reqVO = new MesProTaskPageReqVO();
|
PageResult<MesProTaskDO> result = taskMapper.selectPage(reqVO);
|
|
assertEquals(1, result.getTotal());
|
assertPojoEquals(task, result.getList().get(0));
|
}
|
|
}
|
