diff --git a/src/main/java/com/dispose/controller/DisposeTaskController.java b/src/main/java/com/dispose/controller/DisposeTaskController.java index 082e8aef..20263282 100644 --- a/src/main/java/com/dispose/controller/DisposeTaskController.java +++ b/src/main/java/com/dispose/controller/DisposeTaskController.java @@ -399,7 +399,7 @@ public class DisposeTaskController { taskData.setId(v.getId().toString()); AtomicBoolean bHasData = new AtomicBoolean(false); - taskService.getNodeAllTask(v.getId()).parallelStream().forEach(k -> { + taskService.getNodeAllRunningTask(v.getId()).forEach(k -> { if (v.getType().equals(reqInfo.getType()) || reqInfo.getType() == -1) { int diff = Helper.getTimestampDiff(k.getBeginTime(), k.getPlanEndTime()); TaskInfoData taskInfoData = TaskInfoData.builder() @@ -421,8 +421,8 @@ public class DisposeTaskController { taskData.setStatus(ErrorCode.ERR_OK.getCode()); taskData.setMessage(ErrorCode.ERR_OK.getMsg()); } else { - taskData.setStatus(ErrorCode.ERR_NOSUCHTYPE.getCode()); - taskData.setMessage(ErrorCode.ERR_NOSUCHTYPE.getMsg()); + taskData.setStatus(ErrorCode.ERR_NOSUCHTASK.getCode()); + taskData.setMessage(ErrorCode.ERR_NOSUCHTASK.getMsg()); taskData.setTaskArray(null); } }); diff --git a/src/test/java/com/dispose/controller/TaskControllerTest.java b/src/test/java/com/dispose/controller/TaskControllerTest.java index 525cdaf6..3b5230da 100644 --- a/src/test/java/com/dispose/controller/TaskControllerTest.java +++ b/src/test/java/com/dispose/controller/TaskControllerTest.java @@ -7,9 +7,12 @@ import com.dispose.manager.TaskCacheManager; import com.dispose.mapper.DisposeDeviceMapper; import com.dispose.pojo.dto.ProtocolReqDTO; import com.dispose.pojo.entity.DisposeDevice; +import com.dispose.pojo.vo.common.DisposeCapacity; import com.dispose.pojo.vo.common.IDArrayReq; import com.dispose.pojo.vo.common.TaskInfoData; import com.dispose.pojo.vo.common.TaskInfoDetail; +import com.dispose.pojo.vo.information.NodeTaskData; +import com.dispose.pojo.vo.information.NodeTaskRsp; import com.dispose.pojo.vo.task.*; import com.fasterxml.jackson.databind.ObjectMapper; @@ -108,7 +111,7 @@ public class TaskControllerTest extends InitTestEnvironment { StartTaskReq reqData = StartTaskReq.builder() .id(-1) .type(DeviceCapacity.CLEANUP.getCode()) - .disposeIp("192.168.5.2") + .disposeIp("192.168.3.4") .disposeTime(60) .build(); @@ -510,13 +513,14 @@ public class TaskControllerTest extends InitTestEnvironment { .type(0) .build(); + Long reqTimeStamp = System.currentTimeMillis(); ProtocolReqDTO reqInfo = new ProtocolReqDTO(); reqInfo.setVer(ConstValue.Protocol.VERSION); reqInfo.setCryptoType(ConstValue.Protocol.CRYPTO_NONE); - reqInfo.setTimeStamp(System.currentTimeMillis()); + reqInfo.setTimeStamp(reqTimeStamp); reqInfo.setMsgContent(objectMapper.writeValueAsString(reqData)); - mockMvc.perform(MockMvcRequestBuilders + String getTask = mockMvc.perform(MockMvcRequestBuilders .post("/task/get_node") .contentType(MediaType.APPLICATION_JSON) .header("Authorization", "Bearer " + getLogToken()) @@ -526,6 +530,51 @@ public class TaskControllerTest extends InitTestEnvironment { .andReturn() .getResponse() .getContentAsString(); + + String regex = "\\A(25[0-5]|2[0-4]\\d|[0-1]?\\d?\\d)(\\.(25[0-5]|2[0-4]\\d|[0-1]?\\d?\\d)){3}\\z"; + String msgContent = verifyRep(getTask, reqTimeStamp); + System.out.print("msgContent=" + msgContent); + + //将json字符串转为NodeTaskRsp类 + NodeTaskRsp nodeTaskRsp = objectMapper.readValue(msgContent, NodeTaskRsp.class); + + List taskInfoList = nodeTaskRsp.getItems(); + for (NodeTaskData taskNode : taskInfoList + ) { + Assert.assertNotNull(taskNode.getId()); + Assert.assertNotNull(taskNode.getStatus()); + Assert.assertNotNull(taskNode.getMessage()); + if (taskNode.getStatus() == 0) { + Assert.assertEquals(String.valueOf(taskNode.getStatus()), "0"); + Assert.assertEquals(taskNode.getMessage(), "成功"); + } else { + Assert.assertNotEquals(String.valueOf(taskNode.getStatus()), "0"); + Assert.assertNotEquals(taskNode.getMessage(), "成功"); + } + + Assert.assertNotNull(taskNode.getTaskArray()); + if (taskNode.getTaskArray().size() != 0) { + for (TaskInfoData task : taskNode.getTaskArray() + ) { + Assert.assertNotNull(task.getTaskId()); + Assert.assertNotNull(task.getType()); + Assert.assertNotNull(task.getStartTime()); + Assert.assertNotEquals(task.getStartTime(), "-1"); + Assert.assertNotNull(task.getStartTime()); + Assert.assertNotEquals(task.getStartTime(), "-1"); + Assert.assertNotNull(task.getDisposeIp()); + Assert.assertTrue(task.getDisposeIp().matches(regex)); + + if (task.getFlowAttack() != null) { + Assert.assertNotEquals(task.getFlowAttack(), "-1"); + } + + if (task.getFlowClean() != null) { + Assert.assertNotEquals(task.getFlowClean(), "-1"); + } + } + } + } } /** @@ -537,16 +586,17 @@ public class TaskControllerTest extends InitTestEnvironment { public void t6_getNodeAllTaskAll() throws Exception { GetTaskReq reqData = GetTaskReq.builder() .id(new String[]{}) - .type(2) + .type(0) .build(); + Long reqTimeStamp = System.currentTimeMillis(); ProtocolReqDTO reqInfo = new ProtocolReqDTO(); reqInfo.setVer(ConstValue.Protocol.VERSION); reqInfo.setCryptoType(ConstValue.Protocol.CRYPTO_NONE); - reqInfo.setTimeStamp(System.currentTimeMillis()); + reqInfo.setTimeStamp(reqTimeStamp); reqInfo.setMsgContent(objectMapper.writeValueAsString(reqData)); - mockMvc.perform(MockMvcRequestBuilders + String getTask = mockMvc.perform(MockMvcRequestBuilders .post("/task/get_node") .contentType(MediaType.APPLICATION_JSON) .header("Authorization", "Bearer " + getLogToken()) @@ -556,6 +606,51 @@ public class TaskControllerTest extends InitTestEnvironment { .andReturn() .getResponse() .getContentAsString(); + + String regex = "\\A(25[0-5]|2[0-4]\\d|[0-1]?\\d?\\d)(\\.(25[0-5]|2[0-4]\\d|[0-1]?\\d?\\d)){3}\\z"; + String msgContent = verifyRep(getTask, reqTimeStamp); + System.out.print("msgContent=" + msgContent); + + //将json字符串转为NodeTaskRsp类 + NodeTaskRsp nodeTaskRsp = objectMapper.readValue(msgContent, NodeTaskRsp.class); + + List taskInfoList = nodeTaskRsp.getItems(); + for (NodeTaskData taskNode : taskInfoList + ) { + Assert.assertNotNull(taskNode.getId()); + Assert.assertNotNull(taskNode.getStatus()); + Assert.assertNotNull(taskNode.getMessage()); + if (taskNode.getStatus() == 0) { + Assert.assertEquals(String.valueOf(taskNode.getStatus()), "0"); + Assert.assertEquals(taskNode.getMessage(), "成功"); + } else { + Assert.assertNotEquals(String.valueOf(taskNode.getStatus()), "0"); + Assert.assertNotEquals(taskNode.getMessage(), "成功"); + } + + Assert.assertNotNull(taskNode.getTaskArray()); + if (taskNode.getTaskArray().size() != 0) { + for (TaskInfoData task : taskNode.getTaskArray() + ) { + Assert.assertNotNull(task.getTaskId()); + Assert.assertNotNull(task.getType()); + Assert.assertNotNull(task.getStartTime()); + Assert.assertNotEquals(task.getStartTime(), "-1"); + Assert.assertNotNull(task.getStartTime()); + Assert.assertNotEquals(task.getStartTime(), "-1"); + Assert.assertNotNull(task.getDisposeIp()); + Assert.assertTrue(task.getDisposeIp().matches(regex)); + + if (task.getFlowAttack() != null) { + Assert.assertNotEquals(task.getFlowAttack(), "-1"); + } + + if (task.getFlowClean() != null) { + Assert.assertNotEquals(task.getFlowClean(), "-1"); + } + } + } + } } /**