package cn.iocoder.yudao.module.mes.service.cal.team;
|
|
import cn.hutool.core.collection.ListUtil;
|
import cn.iocoder.yudao.framework.test.core.ut.BaseDbUnitTest;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.cal.plan.MesCalPlanDO;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.cal.plan.MesCalPlanShiftDO;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.cal.plan.MesCalPlanTeamDO;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.cal.team.MesCalTeamShiftDO;
|
import cn.iocoder.yudao.module.mes.dal.mysql.cal.team.MesCalTeamShiftMapper;
|
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.plan.MesCalPlanService;
|
import cn.iocoder.yudao.module.mes.service.cal.plan.MesCalPlanShiftService;
|
import cn.iocoder.yudao.module.mes.service.cal.plan.MesCalPlanTeamService;
|
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 java.util.Comparator;
|
import java.util.List;
|
|
import static org.junit.jupiter.api.Assertions.*;
|
import static org.mockito.Mockito.*;
|
|
/**
|
* {@link MesCalTeamShiftServiceImpl} 的单元测试
|
*
|
* @author 芋道源码
|
*/
|
@Import(MesCalTeamShiftServiceImpl.class)
|
public class MesCalTeamShiftServiceImplTest extends BaseDbUnitTest {
|
|
@Resource
|
private MesCalTeamShiftServiceImpl teamShiftService;
|
|
@Resource
|
private MesCalTeamShiftMapper teamShiftMapper;
|
|
@MockitoBean
|
private MesCalPlanService planService;
|
@MockitoBean
|
private MesCalPlanShiftService planShiftService;
|
@MockitoBean
|
private MesCalPlanTeamService planTeamService;
|
|
@Test
|
public void testGenerateTeamShiftRecords_threeShift_rotateThreeStates() {
|
// 准备参数
|
Long planId = 3001L;
|
// mock 方法：排班计划（三班倒、按天轮班、3 天）
|
MesCalPlanDO plan = MesCalPlanDO.builder()
|
.id(planId)
|
.code("PLAN-THREE-001")
|
.name("三班倒轮转")
|
.calendarType(1)
|
.startDate(LocalDateTime.of(2026, 4, 1, 0, 0))
|
.endDate(LocalDateTime.of(2026, 4, 3, 0, 0))
|
.shiftType(MesCalShiftTypeEnum.THREE.getType())
|
.shiftMethod(MesCalShiftMethodEnum.DAY.getMethod())
|
.shiftCount(1)
|
.build();
|
List<MesCalPlanShiftDO> shifts = ListUtil.of(
|
MesCalPlanShiftDO.builder().id(11L).planId(planId).sort(1).name("白班").startTime("08:00").endTime("16:00").build(),
|
MesCalPlanShiftDO.builder().id(12L).planId(planId).sort(2).name("中班").startTime("16:00").endTime("00:00").build(),
|
MesCalPlanShiftDO.builder().id(13L).planId(planId).sort(3).name("夜班").startTime("00:00").endTime("08:00").build()
|
);
|
List<MesCalPlanTeamDO> teams = ListUtil.of(
|
MesCalPlanTeamDO.builder().id(21L).planId(planId).teamId(101L).build(),
|
MesCalPlanTeamDO.builder().id(22L).planId(planId).teamId(102L).build(),
|
MesCalPlanTeamDO.builder().id(23L).planId(planId).teamId(103L).build()
|
);
|
when(planService.getPlan(planId)).thenReturn(plan);
|
when(planShiftService.getPlanShiftListByPlanId(planId)).thenReturn(shifts);
|
when(planTeamService.getPlanTeamListByPlanId(planId)).thenReturn(teams);
|
|
// 调用
|
teamShiftService.generateTeamShiftRecords(planId);
|
|
// 断言：3 天 × 3 班次 = 9 条排班记录，班组按天轮转
|
List<MesCalTeamShiftDO> records = teamShiftMapper.selectListByPlanId(planId);
|
records.sort(Comparator.comparing(MesCalTeamShiftDO::getDay).thenComparing(MesCalTeamShiftDO::getSort));
|
assertEquals(9, records.size());
|
// 断言：第 1 天（shiftIndex=0）：A 白班、B 中班、C 夜班
|
assertRecord(records.get(0), LocalDateTime.of(2026, 4, 1, 0, 0), 1, 101L, 11L);
|
assertRecord(records.get(1), LocalDateTime.of(2026, 4, 1, 0, 0), 2, 102L, 12L);
|
assertRecord(records.get(2), LocalDateTime.of(2026, 4, 1, 0, 0), 3, 103L, 13L);
|
// 断言：第 2 天（shiftIndex=1）：C 白班、A 中班、B 夜班
|
assertRecord(records.get(3), LocalDateTime.of(2026, 4, 2, 0, 0), 1, 103L, 11L);
|
assertRecord(records.get(4), LocalDateTime.of(2026, 4, 2, 0, 0), 2, 101L, 12L);
|
assertRecord(records.get(5), LocalDateTime.of(2026, 4, 2, 0, 0), 3, 102L, 13L);
|
// 断言：第 3 天（shiftIndex=2）：B 白班、C 中班、A 夜班
|
assertRecord(records.get(6), LocalDateTime.of(2026, 4, 3, 0, 0), 1, 102L, 11L);
|
assertRecord(records.get(7), LocalDateTime.of(2026, 4, 3, 0, 0), 2, 103L, 12L);
|
assertRecord(records.get(8), LocalDateTime.of(2026, 4, 3, 0, 0), 3, 101L, 13L);
|
}
|
|
private static void assertRecord(MesCalTeamShiftDO actual, LocalDateTime day, Integer sort,
|
Long teamId, Long shiftId) {
|
assertEquals(day, actual.getDay());
|
assertEquals(sort, actual.getSort());
|
assertEquals(teamId, actual.getTeamId());
|
assertEquals(shiftId, actual.getShiftId());
|
}
|
|
}
|
