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

OCT 

REM:
1. 添加启动处置多任务接口的单测
This commit is contained in:
chenlinghy 2020-09-02 16:14:21 +08:00
parent 38a45d451d
commit c40e1154ae
2 changed files with 61 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/main/java/com/dispose/controller/DisposeTaskController.java
View File

@ -173,7 +173,6 @@ public class DisposeTaskController {
rspInfo.setExpireTime(ret.getSecondParam().getPlanEndTime());
}
rspInfo.setStatus(ret.getFirstParam().getCode());
rspInfo.setMessage(new String[]{ret.getFirstParam().getMsg()});

63
src/test/java/com/dispose/test/controller/DisposeTaskControllerTest.java
View File

@ -15,6 +15,7 @@ import com.dispose.pojo.dto.protocol.base.IdArraysReq;
import com.dispose.pojo.dto.protocol.base.ProtocolReqDTO;
import com.dispose.pojo.dto.protocol.base.ProtocolRespDTO;
import com.dispose.pojo.dto.protocol.task.TaskStartReq;
import com.dispose.pojo.dto.protocol.task.TaskStartRsp;
import com.dispose.pojo.dto.protocol.task.TaskStopRsp;
import com.dispose.pojo.entity.DisposeTask;
import com.dispose.test.Global.InitTestEnvironment;
@ -90,7 +91,7 @@ public class DisposeTaskControllerTest extends InitTestEnvironment {
TaskStartReq req = TaskStartReq.builder()
.id(String.valueOf(disposeDeviceMapper.selectAll().get(0).getId()))
.type(DisposeCapacityType.CLEANUP.getValue())
.disposeIp("192.168.5.3")
.disposeIp("192.168.100.100")
.disposeTime(60)
.flowDirection(NetflowDirection.DIRECTION_IN.getValue())
.attackType(new Integer[]{DDoSAttackType.ACK_FLOOD.getValue()})
@ -148,7 +149,7 @@ public class DisposeTaskControllerTest extends InitTestEnvironment {
DisposeTask disposeTask = disposeTaskManager.selectAll().get(0);
TaskStartReq req = TaskStartReq.builder()
.id(String.valueOf(disposeTask.getId()))
.id(String.valueOf(disposeTask.getDeviceId()))
.type(disposeTask.getDisposeCapacity().getValue())
.disposeIp(disposeTask.getDisposeIp())
.disposeTime(60)
@ -264,4 +265,62 @@ public class DisposeTaskControllerTest extends InitTestEnvironment {
}
}
}
/**
* A 4 stop task protocol resp dto.
*
* @throws Exception the exception
*/
@Test
public void a4_startTaskMulIp() throws Exception {
String[] mulDisposeIp = {"192.168.5.2", "192.168.5.3"};
TaskStartReq req = TaskStartReq.builder()
.id(String.valueOf(disposeDeviceMapper.selectAll().get(0).getId()))
.type(DisposeCapacityType.CLEANUP.getValue())
.mulDisposeIp(mulDisposeIp)
.disposeTime(60)
.flowDirection(NetflowDirection.DIRECTION_IN.getValue())
.attackType(new Integer[]{DDoSAttackType.ACK_FLOOD.getValue()})
.flowBandwidth(DisposeConfigValue.DEFAULT_DISPOSE_BANDWIDTH)
.build();
ProtocolReqDTO<TaskStartReq> reqInfo = new ProtocolReqDTO<>();
reqInfo.setVer(ConstValue.Protocol.VERSION);
reqInfo.setCryptoType(ProtoCryptoType.CRYPTO_NONE.getCode());
reqInfo.setTimeStamp(System.currentTimeMillis());
reqInfo.setMsgContent(req);
String ret = mockMvc.perform(MockMvcRequestBuilders
.post("/task/startMulIp")
.contentType(MediaType.APPLICATION_JSON)
.header("Authorization", ConstValue.STRING_HTTP_AUTH_HEAD + getLoginToken())
.content(objectMapper.writeValueAsString(reqInfo)))
.andDo(print()).andExpect(status().isOk())
.andExpect(jsonPath("$.code").value(HttpStatus.ok().status()))
.andReturn()
.getResponse()
.getContentAsString();
ProtocolRespDTO<List<TaskStartRsp>> rspInfo = objectMapper.readValue(ret,
new TypeReference<ProtocolRespDTO<List<TaskStartRsp>>>() {
});
verifyRespProtocol(rspInfo);
log.debug(objectMapper.writerWithDefaultPrettyPrinter().writeValueAsString(rspInfo));
Assert.assertNotNull(rspInfo.getMsgContent());
rspInfo.getMsgContent().forEach(v->{
Assert.assertNotNull(v.getStatus());
Assert.assertNotNull(v.getMessage());
Assert.assertNotNull(v.getDisposeIp());
if(v.getStatus() == ErrorCode.ERR_OK.getCode() ||
v.getStatus() == ErrorCode.ERR_TASKRUNNING.getCode()){
Assert.assertNotNull(v.getTaskId());
Assert.assertNotNull(v.getExpireTime());
}
});
}
}