package cn.iocoder.yudao.module.infra.service.job;
|
|
import cn.hutool.extra.spring.SpringUtil;
|
import cn.iocoder.yudao.framework.common.pojo.PageResult;
|
import cn.iocoder.yudao.framework.quartz.core.scheduler.SchedulerManager;
|
import cn.iocoder.yudao.framework.test.core.ut.BaseDbUnitTest;
|
import cn.iocoder.yudao.module.infra.controller.admin.job.vo.job.JobPageReqVO;
|
import cn.iocoder.yudao.module.infra.controller.admin.job.vo.job.JobSaveReqVO;
|
import cn.iocoder.yudao.module.infra.dal.dataobject.job.JobDO;
|
import cn.iocoder.yudao.module.infra.dal.mysql.job.JobMapper;
|
import cn.iocoder.yudao.module.infra.enums.job.JobStatusEnum;
|
import cn.iocoder.yudao.module.infra.job.job.JobLogCleanJob;
|
import jakarta.annotation.Resource;
|
import org.junit.jupiter.api.Test;
|
import org.mockito.MockedStatic;
|
import org.quartz.SchedulerException;
|
import org.springframework.context.annotation.Import;
|
import org.springframework.test.context.bean.override.mockito.MockitoBean;
|
|
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.AssertUtils.assertServiceException;
|
import static cn.iocoder.yudao.framework.test.core.util.RandomUtils.randomPojo;
|
import static cn.iocoder.yudao.framework.test.core.util.RandomUtils.randomString;
|
import static cn.iocoder.yudao.module.infra.enums.ErrorCodeConstants.*;
|
import static org.junit.jupiter.api.Assertions.*;
|
import static org.mockito.ArgumentMatchers.eq;
|
import static org.mockito.Mockito.mockStatic;
|
import static org.mockito.Mockito.verify;
|
|
@Import(JobServiceImpl.class)
|
public class JobServiceImplTest extends BaseDbUnitTest {
|
|
@Resource
|
private JobServiceImpl jobService;
|
@Resource
|
private JobMapper jobMapper;
|
@MockitoBean
|
private SchedulerManager schedulerManager;
|
|
@MockitoBean
|
private JobLogCleanJob jobLogCleanJob;
|
|
@Test
|
public void testCreateJob_cronExpressionValid() {
|
// 准备参数。Cron 表达式为 String 类型，默认随机字符串。
|
JobSaveReqVO reqVO = randomPojo(JobSaveReqVO.class);
|
|
// 调用，并断言异常
|
assertServiceException(() -> jobService.createJob(reqVO), JOB_CRON_EXPRESSION_VALID);
|
}
|
|
@Test
|
public void testCreateJob_jobHandlerExists() throws SchedulerException {
|
// 准备参数 指定 Cron 表达式
|
JobSaveReqVO reqVO = randomPojo(JobSaveReqVO.class, o -> o.setCronExpression("0 0/1 * * * ? *"));
|
try (MockedStatic<SpringUtil> springUtilMockedStatic = mockStatic(SpringUtil.class)) {
|
springUtilMockedStatic.when(() -> SpringUtil.getBean(eq(reqVO.getHandlerName())))
|
.thenReturn(jobLogCleanJob);
|
|
// 调用
|
jobService.createJob(reqVO);
|
// 调用，并断言异常
|
assertServiceException(() -> jobService.createJob(reqVO), JOB_HANDLER_EXISTS);
|
}
|
}
|
|
@Test
|
public void testCreateJob_success() throws SchedulerException {
|
// 准备参数 指定 Cron 表达式
|
JobSaveReqVO reqVO = randomPojo(JobSaveReqVO.class, o -> o.setCronExpression("0 0/1 * * * ? *"))
|
.setId(null);
|
try (MockedStatic<SpringUtil> springUtilMockedStatic = mockStatic(SpringUtil.class)) {
|
springUtilMockedStatic.when(() -> SpringUtil.getBean(eq(reqVO.getHandlerName())))
|
.thenReturn(jobLogCleanJob);
|
|
// 调用
|
Long jobId = jobService.createJob(reqVO);
|
// 断言
|
assertNotNull(jobId);
|
// 校验记录的属性是否正确
|
JobDO job = jobMapper.selectById(jobId);
|
assertPojoEquals(reqVO, job, "id");
|
assertEquals(JobStatusEnum.NORMAL.getStatus(), job.getStatus());
|
// 校验调用
|
verify(schedulerManager).addJob(eq(job.getId()), eq(job.getHandlerName()), eq(job.getHandlerParam()),
|
eq(job.getCronExpression()), eq(reqVO.getRetryCount()), eq(reqVO.getRetryInterval()));
|
}
|
}
|
|
@Test
|
public void testUpdateJob_jobNotExists(){
|
// 准备参数
|
JobSaveReqVO reqVO = randomPojo(JobSaveReqVO.class, o -> o.setCronExpression("0 0/1 * * * ? *"));
|
|
// 调用，并断言异常
|
assertServiceException(() -> jobService.updateJob(reqVO), JOB_NOT_EXISTS);
|
}
|
|
@Test
|
public void testUpdateJob_onlyNormalStatus(){
|
// mock 数据
|
JobDO job = randomPojo(JobDO.class, o -> o.setStatus(JobStatusEnum.INIT.getStatus()));
|
jobMapper.insert(job);
|
// 准备参数
|
JobSaveReqVO updateReqVO = randomPojo(JobSaveReqVO.class, o -> {
|
o.setId(job.getId());
|
o.setCronExpression("0 0/1 * * * ? *");
|
});
|
|
// 调用，并断言异常
|
assertServiceException(() -> jobService.updateJob(updateReqVO),
|
JOB_UPDATE_ONLY_NORMAL_STATUS);
|
}
|
|
@Test
|
public void testUpdateJob_success() throws SchedulerException {
|
// mock 数据
|
JobDO job = randomPojo(JobDO.class, o -> o.setStatus(JobStatusEnum.NORMAL.getStatus()));
|
jobMapper.insert(job);
|
// 准备参数
|
JobSaveReqVO updateReqVO = randomPojo(JobSaveReqVO.class, o -> {
|
o.setId(job.getId());
|
o.setCronExpression("0 0/1 * * * ? *");
|
});
|
try (MockedStatic<SpringUtil> springUtilMockedStatic = mockStatic(SpringUtil.class)) {
|
springUtilMockedStatic.when(() -> SpringUtil.getBean(eq(updateReqVO.getHandlerName())))
|
.thenReturn(jobLogCleanJob);
|
|
// 调用
|
jobService.updateJob(updateReqVO);
|
// 校验记录的属性是否正确
|
JobDO updateJob = jobMapper.selectById(updateReqVO.getId());
|
assertPojoEquals(updateReqVO, updateJob);
|
// 校验调用
|
verify(schedulerManager).updateJob(eq(job.getHandlerName()), eq(updateReqVO.getHandlerParam()),
|
eq(updateReqVO.getCronExpression()), eq(updateReqVO.getRetryCount()), eq(updateReqVO.getRetryInterval()));
|
}
|
}
|
|
@Test
|
public void testUpdateJobStatus_changeStatusInvalid() {
|
// 调用，并断言异常
|
assertServiceException(() -> jobService.updateJobStatus(1L, JobStatusEnum.INIT.getStatus()),
|
JOB_CHANGE_STATUS_INVALID);
|
}
|
|
@Test
|
public void testUpdateJobStatus_changeStatusEquals() {
|
// mock 数据
|
JobDO job = randomPojo(JobDO.class, o -> o.setStatus(JobStatusEnum.NORMAL.getStatus()));
|
jobMapper.insert(job);
|
|
// 调用，并断言异常
|
assertServiceException(() -> jobService.updateJobStatus(job.getId(), job.getStatus()),
|
JOB_CHANGE_STATUS_EQUALS);
|
}
|
|
@Test
|
public void testUpdateJobStatus_stopSuccess() throws SchedulerException {
|
// mock 数据
|
JobDO job = randomPojo(JobDO.class, o -> o.setStatus(JobStatusEnum.NORMAL.getStatus()));
|
jobMapper.insert(job);
|
|
// 调用
|
jobService.updateJobStatus(job.getId(), JobStatusEnum.STOP.getStatus());
|
// 校验记录的属性是否正确
|
JobDO dbJob = jobMapper.selectById(job.getId());
|
assertEquals(JobStatusEnum.STOP.getStatus(), dbJob.getStatus());
|
// 校验调用
|
verify(schedulerManager).pauseJob(eq(job.getHandlerName()));
|
}
|
|
@Test
|
public void testUpdateJobStatus_normalSuccess() throws SchedulerException {
|
// mock 数据
|
JobDO job = randomPojo(JobDO.class, o -> o.setStatus(JobStatusEnum.STOP.getStatus()));
|
jobMapper.insert(job);
|
|
// 调用
|
jobService.updateJobStatus(job.getId(), JobStatusEnum.NORMAL.getStatus());
|
// 校验记录的属性是否正确
|
JobDO dbJob = jobMapper.selectById(job.getId());
|
assertEquals(JobStatusEnum.NORMAL.getStatus(), dbJob.getStatus());
|
// 校验调用
|
verify(schedulerManager).resumeJob(eq(job.getHandlerName()));
|
}
|
|
@Test
|
public void testTriggerJob_success() throws SchedulerException {
|
// mock 数据
|
JobDO job = randomPojo(JobDO.class);
|
jobMapper.insert(job);
|
|
// 调用
|
jobService.triggerJob(job.getId());
|
// 校验调用
|
verify(schedulerManager).triggerJob(eq(job.getId()),
|
eq(job.getHandlerName()), eq(job.getHandlerParam()));
|
}
|
|
@Test
|
public void testDeleteJob_success() throws SchedulerException {
|
// mock 数据
|
JobDO job = randomPojo(JobDO.class);
|
jobMapper.insert(job);
|
|
// 调用
|
jobService.deleteJob(job.getId());
|
// 校验不存在
|
assertNull(jobMapper.selectById(job.getId()));
|
// 校验调用
|
verify(schedulerManager).deleteJob(eq(job.getHandlerName()));
|
}
|
|
@Test
|
public void testGetJobPage() {
|
// mock 数据
|
JobDO dbJob = randomPojo(JobDO.class, o -> {
|
o.setName("定时任务测试");
|
o.setHandlerName("handlerName 单元测试");
|
o.setStatus(JobStatusEnum.INIT.getStatus());
|
});
|
jobMapper.insert(dbJob);
|
// 测试 name 不匹配
|
jobMapper.insert(cloneIgnoreId(dbJob, o -> o.setName("土豆")));
|
// 测试 status 不匹配
|
jobMapper.insert(cloneIgnoreId(dbJob, o -> o.setStatus(JobStatusEnum.NORMAL.getStatus())));
|
// 测试 handlerName 不匹配
|
jobMapper.insert(cloneIgnoreId(dbJob, o -> o.setHandlerName(randomString())));
|
// 准备参数
|
JobPageReqVO reqVo = new JobPageReqVO();
|
reqVo.setName("定时");
|
reqVo.setStatus(JobStatusEnum.INIT.getStatus());
|
reqVo.setHandlerName("单元");
|
|
// 调用
|
PageResult<JobDO> pageResult = jobService.getJobPage(reqVo);
|
// 断言
|
assertEquals(1, pageResult.getTotal());
|
assertEquals(1, pageResult.getList().size());
|
assertPojoEquals(dbJob, pageResult.getList().get(0));
|
}
|
|
@Test
|
public void testGetJob() {
|
// mock 数据
|
JobDO dbJob = randomPojo(JobDO.class);
|
jobMapper.insert(dbJob);
|
// 调用
|
JobDO job = jobService.getJob(dbJob.getId());
|
// 断言
|
assertPojoEquals(dbJob, job);
|
}
|
|
}
|
