diff --git a/src/test/java/com/dispose/test/service/DisposeTaskServiceTest.java b/src/test/java/com/dispose/test/service/DisposeTaskServiceTest.java index 50b4cab8..aefb8a7d 100644 --- a/src/test/java/com/dispose/test/service/DisposeTaskServiceTest.java +++ b/src/test/java/com/dispose/test/service/DisposeTaskServiceTest.java @@ -1,12 +1,9 @@ package com.dispose.test.service; -import com.dispose.common.DDoSAttackType; -import com.dispose.common.DisposeCapacityType; -import com.dispose.common.DisposeConfigValue; -import com.dispose.common.ErrorCode; -import com.dispose.common.NetflowDirection; +import com.dispose.common.*; import com.dispose.manager.DisposeTaskManager; import com.dispose.mapper.DisposeDeviceMapper; +import com.dispose.mapper.DisposeTaskMapper; import com.dispose.mapper.UserAccountMapper; import com.dispose.pojo.entity.DisposeTask; import com.dispose.pojo.po.MulReturnType; @@ -76,6 +73,11 @@ public class DisposeTaskServiceTest extends InitTestEnvironment { */ @Resource private DisposeTaskService disposeTaskService; + /** + * The Dispose task mapper. + */ + @Resource + private DisposeTaskMapper disposeTaskMapper; /** * User login test. @@ -127,5 +129,34 @@ public class DisposeTaskServiceTest extends InitTestEnvironment { Assert.assertEquals(ret.getSecondParam().getId(), disposeTaskManager.getDisposeTask(newTask.getDeviceId(), newTask.getDisposeIp(), newTask.getDisposeCapacity()).getId()); } + + /** + * stop task test mul return type. + */ + @Test + public void a2_stopTaskTest() { + Long taskId = disposeTaskMapper.selectAll().get(0).getId(); + + MulReturnType ret = disposeTaskService.stopTask(taskId-1); + if (ret.getFirstParam() == ErrorCode.ERR_NOSUCHTASK) { + Assert.assertNull(disposeTaskManager.getDisposeTaskById(taskId-1)); + } + + ret = disposeTaskService.stopTask(taskId); + DisposeTask task = disposeTaskMapper.getDisposeTaskById(taskId); + + if (ret.getFirstParam() == ErrorCode.ERR_TASKNOTRUNNING) { + Assert.assertNotNull(task); + Assert.assertFalse(task.getCurrentStatus().taskIsRunning()); + } + else if(ret.getFirstParam() == ErrorCode.ERR_DATABASE){ + Assert.assertNull(task); + Assert.assertFalse(disposeTaskManager.changeDisposeTaskStatus(taskId, DisposeTaskStatus.TASK_CANCELED)); + } + else if(ret.getFirstParam() == ErrorCode.ERR_OK){ + Assert.assertNotNull(task); + Assert.assertEquals(task.getCurrentStatus(),DisposeTaskStatus.TASK_CANCELED); + } + } }