cmhi
/
dispose
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

OCT 

REM:
1. 添加停止处置任务单元测试。
This commit is contained in:
wangyiyun 2020-09-03 11:24:04 +08:00
parent a3aa4cc682
commit c47daeb5cb
1 changed files with 36 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
src/test/java/com/dispose/test/service/DisposeTaskServiceTest.java
View File

@ -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<ErrorCode, DisposeTask> 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);
}
}
}