package cn.iocoder.yudao.module.mes.service.cal.plan;
|
|
import cn.iocoder.yudao.framework.test.core.ut.BaseDbUnitTest;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.cal.plan.MesCalPlanShiftDO;
|
import cn.iocoder.yudao.module.mes.dal.mysql.cal.plan.MesCalPlanShiftMapper;
|
import cn.iocoder.yudao.module.mes.enums.cal.MesCalShiftTypeEnum;
|
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.util.List;
|
|
import static org.junit.jupiter.api.Assertions.*;
|
|
/**
|
* {@link MesCalPlanShiftServiceImpl} 的单元测试
|
*
|
* @author 芋道源码
|
*/
|
@Import(MesCalPlanShiftServiceImpl.class)
|
public class MesCalPlanShiftServiceImplTest extends BaseDbUnitTest {
|
|
@Resource
|
private MesCalPlanShiftServiceImpl planShiftService;
|
|
@Resource
|
private MesCalPlanShiftMapper planShiftMapper;
|
|
@MockitoBean
|
private MesCalPlanService planService;
|
|
@Test
|
public void testAddDefaultPlanShift_singleShift_use18Clock() {
|
// 准备参数
|
Long planId = 1001L;
|
|
// 调用
|
planShiftService.addDefaultPlanShift(planId, MesCalShiftTypeEnum.SINGLE.getType());
|
|
// 断言：生成 1 条默认班次（白班 08:00~18:00）
|
List<MesCalPlanShiftDO> shifts = planShiftMapper.selectListByPlanId(planId);
|
assertEquals(1, shifts.size());
|
MesCalPlanShiftDO shift = shifts.get(0);
|
assertEquals(planId, shift.getPlanId());
|
assertEquals(1, shift.getSort());
|
assertEquals("白班", shift.getName());
|
assertEquals("08:00", shift.getStartTime());
|
assertEquals("18:00", shift.getEndTime());
|
}
|
|
}
|
