DeviceStatusCheckTaskTest.java 1.3 KB

12345678910111213141516171819202122232425262728293031323334353637
  1. package com.hfln.device.application.task;
  2. import com.hfln.device.domain.entity.Device;
  3. import com.hfln.device.domain.service.DeviceManagerService;
  4. import org.junit.jupiter.api.BeforeEach;
  5. import org.junit.jupiter.api.Test;
  6. import org.mockito.InjectMocks;
  7. import org.mockito.Mock;
  8. import org.mockito.MockitoAnnotations;
  9. import java.util.*;
  10. import static org.mockito.Mockito.*;
  11. class DeviceStatusCheckTaskTest {
  12. @Mock
  13. DeviceManagerService deviceManagerService;
  14. @InjectMocks
  15. DeviceStatusCheckTask deviceStatusCheckTask;
  16. @BeforeEach
  17. void setUp() {
  18. MockitoAnnotations.openMocks(this);
  19. }
  20. @Test
  21. void testCheckDeviceKeepAlive() {
  22. Device device = new Device("dev001");
  23. device.setKeepaliveTime(System.currentTimeMillis() - 100000);
  24. when(deviceManagerService.getAllDevicesFromCache()).thenReturn(Collections.singletonList(device));
  25. deviceStatusCheckTask.checkDeviceKeepAlive();
  26. // 可根据业务补充verify
  27. }
  28. @Test
  29. void testCheckAllDeviceAlarmPlan() {
  30. Device device = new Device("dev001");
  31. when(deviceManagerService.getAllDevicesFromCache()).thenReturn(Collections.singletonList(device));
  32. deviceStatusCheckTask.checkAllDeviceAlarmPlan();
  33. // 可根据业务补充verify
  34. }
  35. }