package cn.iocoder.yudao.module.mes.service.cal.plan;
|
|
import cn.iocoder.yudao.framework.test.core.ut.BaseDbUnitTest;
|
import cn.iocoder.yudao.module.mes.controller.admin.cal.plan.vo.MesCalPlanSaveReqVO;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.cal.plan.MesCalPlanDO;
|
import cn.iocoder.yudao.module.mes.dal.mysql.cal.plan.MesCalPlanMapper;
|
import cn.iocoder.yudao.module.mes.enums.cal.MesCalPlanStatusEnum;
|
import cn.iocoder.yudao.module.mes.enums.cal.MesCalShiftMethodEnum;
|
import cn.iocoder.yudao.module.mes.enums.cal.MesCalShiftTypeEnum;
|
import cn.iocoder.yudao.module.mes.service.cal.team.MesCalTeamShiftService;
|
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.time.LocalDateTime;
|
|
import static cn.iocoder.yudao.framework.test.core.util.RandomUtils.randomPojo;
|
import static org.junit.jupiter.api.Assertions.*;
|
import static org.mockito.Mockito.*;
|
|
/**
|
* {@link MesCalPlanServiceImpl} 的单元测试
|
*
|
* @author 芋道源码
|
*/
|
@Import(MesCalPlanServiceImpl.class)
|
public class MesCalPlanServiceImplTest extends BaseDbUnitTest {
|
|
@Resource
|
private MesCalPlanServiceImpl planService;
|
|
@Resource
|
private MesCalPlanMapper planMapper;
|
|
@MockitoBean
|
private MesCalPlanShiftService planShiftService;
|
@MockitoBean
|
private MesCalPlanTeamService planTeamService;
|
@MockitoBean
|
private MesCalTeamShiftService teamShiftService;
|
|
@Test
|
public void testCreatePlan_singleShift_addDefaultShift() {
|
// 准备参数
|
MesCalPlanSaveReqVO reqVO = new MesCalPlanSaveReqVO();
|
reqVO.setCode("PLAN-SINGLE-001");
|
reqVO.setName("单白班默认班次");
|
reqVO.setCalendarType(1);
|
reqVO.setStartDate(LocalDateTime.of(2026, 4, 1, 0, 0));
|
reqVO.setEndDate(LocalDateTime.of(2026, 4, 30, 0, 0));
|
reqVO.setShiftType(MesCalShiftTypeEnum.SINGLE.getType());
|
reqVO.setShiftMethod(MesCalShiftMethodEnum.DAY.getMethod());
|
reqVO.setShiftCount(1);
|
|
// 调用
|
Long planId = planService.createPlan(reqVO);
|
|
// 断言 1：计划状态为草稿
|
MesCalPlanDO plan = planMapper.selectById(planId);
|
assertEquals(MesCalPlanStatusEnum.PREPARE.getStatus(), plan.getStatus());
|
// 断言 2：自动生成默认班次
|
verify(planShiftService).addDefaultPlanShift(planId, MesCalShiftTypeEnum.SINGLE.getType());
|
}
|
|
@Test
|
public void testConfirmPlan_syncGenerateTeamShiftRecords() {
|
// mock 数据：插入一条草稿状态的排班计划
|
MesCalPlanDO plan = randomPojo(MesCalPlanDO.class, o -> {
|
o.setCalendarType(1);
|
o.setStartDate(LocalDateTime.of(2026, 4, 1, 0, 0));
|
o.setEndDate(LocalDateTime.of(2026, 4, 3, 0, 0));
|
o.setStatus(MesCalPlanStatusEnum.PREPARE.getStatus());
|
o.setShiftType(MesCalShiftTypeEnum.SINGLE.getType());
|
o.setShiftMethod(MesCalShiftMethodEnum.DAY.getMethod());
|
o.setShiftCount(1);
|
});
|
planMapper.insert(plan);
|
// mock 方法：班组数量满足要求
|
when(planTeamService.getPlanTeamCountByPlanId(plan.getId())).thenReturn(1L);
|
|
// 调用
|
planService.confirmPlan(plan.getId());
|
|
// 断言 1：计划状态更新为已确认
|
MesCalPlanDO updatePlan = planMapper.selectById(plan.getId());
|
assertEquals(MesCalPlanStatusEnum.CONFIRMED.getStatus(), updatePlan.getStatus());
|
// 断言 2：触发生成班组排班记录
|
verify(teamShiftService).generateTeamShiftRecords(plan.getId());
|
}
|
|
@Test
|
public void testDeletePlan_notCascadeDeleteTeamShift() {
|
// mock 数据：插入一条草稿状态的排班计划
|
MesCalPlanDO plan = randomPojo(MesCalPlanDO.class, o -> {
|
o.setCalendarType(1);
|
o.setStartDate(LocalDateTime.of(2026, 4, 1, 0, 0));
|
o.setEndDate(LocalDateTime.of(2026, 4, 3, 0, 0));
|
o.setStatus(MesCalPlanStatusEnum.PREPARE.getStatus());
|
o.setShiftType(MesCalShiftTypeEnum.SINGLE.getType());
|
o.setShiftMethod(MesCalShiftMethodEnum.DAY.getMethod());
|
o.setShiftCount(1);
|
});
|
planMapper.insert(plan);
|
|
// 调用
|
planService.deletePlan(plan.getId());
|
|
// 断言 1：计划被删除
|
assertNull(planMapper.selectById(plan.getId()));
|
// 断言 2：级联删除班次和班组关联
|
verify(planShiftService).deletePlanShiftByPlanId(plan.getId());
|
verify(planTeamService).deleteByPlanId(plan.getId());
|
// 断言 3：不级联删除班组排班记录
|
verify(teamShiftService, never()).deleteByPlanId(plan.getId());
|
}
|
|
}
|
