package cn.iocoder.yudao.module.iot.service.rule.scene.timer;
|
|
import cn.hutool.core.collection.ListUtil;
|
import cn.iocoder.yudao.module.iot.dal.dataobject.rule.IotSceneRuleDO;
|
import cn.iocoder.yudao.module.iot.enums.rule.IotSceneRuleTriggerTypeEnum;
|
import cn.iocoder.yudao.module.iot.framework.job.core.IotSchedulerManager;
|
import cn.iocoder.yudao.module.iot.job.rule.IotSceneRuleJob;
|
import org.junit.jupiter.api.BeforeEach;
|
import org.junit.jupiter.api.Test;
|
import org.junit.jupiter.api.extension.ExtendWith;
|
import org.mockito.InjectMocks;
|
import org.mockito.Mock;
|
import org.mockito.junit.jupiter.MockitoExtension;
|
import org.quartz.SchedulerException;
|
|
import static org.mockito.ArgumentMatchers.any;
|
import static org.mockito.ArgumentMatchers.eq;
|
import static org.mockito.Mockito.*;
|
|
/**
|
* {@link IotSceneRuleTimerHandler} 的单元测试类
|
*
|
* @author HUIHUI
|
*/
|
@ExtendWith(MockitoExtension.class)
|
public class IotSceneRuleTimerHandlerTest {
|
|
@Mock
|
private IotSchedulerManager schedulerManager;
|
|
@InjectMocks
|
private IotSceneRuleTimerHandler timerHandler;
|
|
@BeforeEach
|
void setUp() {
|
// 重置 Mock 对象
|
reset(schedulerManager);
|
}
|
|
@Test
|
public void testRegisterTimerTriggers_success() throws SchedulerException {
|
// 准备参数
|
Long sceneRuleId = 1L;
|
IotSceneRuleDO sceneRule = new IotSceneRuleDO();
|
sceneRule.setId(sceneRuleId);
|
sceneRule.setStatus(0); // 0 表示启用
|
// 创建定时触发器
|
IotSceneRuleDO.Trigger timerTrigger = new IotSceneRuleDO.Trigger();
|
timerTrigger.setType(IotSceneRuleTriggerTypeEnum.TIMER.getType());
|
timerTrigger.setCronExpression("0 0 12 * * ?"); // 每天中午12点
|
sceneRule.setTriggers(ListUtil.toList(timerTrigger));
|
|
// 调用
|
timerHandler.registerTimerTriggers(sceneRule);
|
|
// 验证
|
verify(schedulerManager, times(1)).addOrUpdateJob(
|
eq(IotSceneRuleJob.class),
|
eq("iot_scene_rule_timer_" + sceneRuleId),
|
eq("0 0 12 * * ?"),
|
eq(IotSceneRuleJob.buildJobDataMap(sceneRuleId))
|
);
|
}
|
|
@Test
|
public void testRegisterTimerTriggers_noTimerTrigger() throws SchedulerException {
|
// 准备参数 - 没有定时触发器
|
IotSceneRuleDO sceneRule = new IotSceneRuleDO();
|
sceneRule.setStatus(0); // 0 表示启用
|
// 创建非定时触发器
|
IotSceneRuleDO.Trigger deviceTrigger = new IotSceneRuleDO.Trigger();
|
deviceTrigger.setType(IotSceneRuleTriggerTypeEnum.DEVICE_PROPERTY_POST.getType());
|
sceneRule.setTriggers(ListUtil.toList(deviceTrigger));
|
|
// 调用
|
timerHandler.registerTimerTriggers(sceneRule);
|
|
// 验证 - 不应该调用调度器
|
verify(schedulerManager, never()).addOrUpdateJob(any(), any(), any(), any());
|
}
|
|
@Test
|
public void testRegisterTimerTriggers_emptyCronExpression() throws SchedulerException {
|
// 准备参数 - CRON 表达式为空
|
Long sceneRuleId = 2L;
|
IotSceneRuleDO sceneRule = new IotSceneRuleDO();
|
sceneRule.setId(sceneRuleId);
|
sceneRule.setStatus(0); // 0 表示启用
|
// 创建定时触发器但没有 CRON 表达式
|
IotSceneRuleDO.Trigger timerTrigger = new IotSceneRuleDO.Trigger();
|
timerTrigger.setType(IotSceneRuleTriggerTypeEnum.TIMER.getType());
|
timerTrigger.setCronExpression(""); // 空的 CRON 表达式
|
sceneRule.setTriggers(ListUtil.toList(timerTrigger));
|
|
// 调用
|
timerHandler.registerTimerTriggers(sceneRule);
|
|
// 验证 - 不应该调用调度器
|
verify(schedulerManager, never()).addOrUpdateJob(any(), any(), any(), any());
|
}
|
|
@Test
|
public void testUnregisterTimerTriggers_success() throws SchedulerException {
|
// 准备参数
|
Long sceneRuleId = 3L;
|
|
// 调用
|
timerHandler.unregisterTimerTriggers(sceneRuleId);
|
|
// 验证
|
verify(schedulerManager, times(1)).deleteJob("iot_scene_rule_timer_" + sceneRuleId);
|
}
|
|
@Test
|
public void testPauseTimerTriggers_success() throws SchedulerException {
|
// 准备参数
|
Long sceneRuleId = 4L;
|
|
// 调用
|
timerHandler.pauseTimerTriggers(sceneRuleId);
|
|
// 验证
|
verify(schedulerManager, times(1)).pauseJob("iot_scene_rule_timer_" + sceneRuleId);
|
}
|
|
}
|
