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

OCT 

REM:
1.注释kafka测试代码
This commit is contained in:
chenlinghy 2021-11-02 11:28:18 +08:00
parent e8a1e60c06
commit e9baad7115
1 changed files with 51 additions and 51 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

102
src/test/java/com/dispose/test/dev/controller/KafkaControllerTest.java
View File

@ -71,56 +71,56 @@ public class KafkaControllerTest extends InitTestEnvironment {
*/ */
@Test @Test
public void a1_dispatchCommand() throws Exception { public void a1_dispatchCommand() throws Exception {
JSONObject disposeParam = new JSONObject(); // JSONObject disposeParam = new JSONObject();
SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss"); // SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
disposeParam.set("alarmId", "1"); // disposeParam.set("alarmId", "1");
disposeParam.set("dstIp", "192.168.1.1"); // disposeParam.set("dstIp", "192.168.1.1");
disposeParam.set("attackType", "RSTFlood"); // disposeParam.set("attackType", "RSTFlood");
disposeParam.set("bpspps", "bps"); // disposeParam.set("bpspps", "bps");
disposeParam.set("dstProvince", "浙江"); // disposeParam.set("dstProvince", "浙江");
disposeParam.set("dstCity", "杭州"); // disposeParam.set("dstCity", "杭州");
disposeParam.set("startTime", sdf.format(new Date())); // disposeParam.set("startTime", sdf.format(new Date()));
// 1清洗，2流控，3黑洞 // // 1清洗，2流控，3黑洞
disposeParam.set("disposeType", 1); // disposeParam.set("disposeType", 1);
disposeParam.set("disposeTime", 30); // disposeParam.set("disposeTime", 30);
disposeParam.set("endTime", sdf.format(new Date())); // disposeParam.set("endTime", sdf.format(new Date()));
List<String> srcIp = new ArrayList<>(); // List<String> srcIp = new ArrayList<>();
srcIp.add("192.168.10.1"); // srcIp.add("192.168.10.1");
srcIp.add("192.168.10.2"); // srcIp.add("192.168.10.2");
disposeParam.set("srcIpLs", srcIp); // disposeParam.set("srcIpLs", srcIp);
disposeParam.set("maxBps", "20bps"); // disposeParam.set("maxBps", "20bps");
disposeParam.set("maxPps", "20pps"); // disposeParam.set("maxPps", "20pps");
String jsonAlarmInfo = disposeParam.toString(); // String jsonAlarmInfo = disposeParam.toString();
log.info("============ jsonAlarmInfo :{}", jsonAlarmInfo); // log.info("============ jsonAlarmInfo :{}", jsonAlarmInfo);
//
AlarmInfoReq alarmInfoReq = AlarmInfoReq.builder() // AlarmInfoReq alarmInfoReq = AlarmInfoReq.builder()
.alarmInfo(jsonAlarmInfo).build(); // .alarmInfo(jsonAlarmInfo).build();
//
ProtocolReqDTO<AlarmInfoReq> reqInfo = new ProtocolReqDTO<>(); // ProtocolReqDTO<AlarmInfoReq> reqInfo = new ProtocolReqDTO<>();
//
reqInfo.setVer(ConstValue.Protocol.VERSION); // reqInfo.setVer(ConstValue.Protocol.VERSION);
reqInfo.setCryptoType(ProtoCryptoType.CRYPTO_NONE.getCode()); // reqInfo.setCryptoType(ProtoCryptoType.CRYPTO_NONE.getCode());
reqInfo.setTimeStamp(System.currentTimeMillis()); // reqInfo.setTimeStamp(System.currentTimeMillis());
reqInfo.setMsgContent(alarmInfoReq); // reqInfo.setMsgContent(alarmInfoReq);
//
String ret = mockMvc.perform(MockMvcRequestBuilders // String ret = mockMvc.perform(MockMvcRequestBuilders
.post("/kafka/dispatchCommand") // .post("/kafka/dispatchCommand")
.contentType(MediaType.APPLICATION_JSON) // .contentType(MediaType.APPLICATION_JSON)
.header("Authorization", ConstValue.STRING_HTTP_AUTH_HEAD + getLoginToken()) // .header("Authorization", ConstValue.STRING_HTTP_AUTH_HEAD + getLoginToken())
.content(objectMapper.writeValueAsString(reqInfo))) // .content(objectMapper.writeValueAsString(reqInfo)))
.andDo(print()).andExpect(status().isOk()) // .andDo(print()).andExpect(status().isOk())
.andExpect(jsonPath("$.code").value(HttpStatus.ok().status())) // .andExpect(jsonPath("$.code").value(HttpStatus.ok().status()))
.andReturn() // .andReturn()
.getResponse() // .getResponse()
.getContentAsString(); // .getContentAsString();
//
ProtocolRespDTO<BaseRespStatus> rspInfo = objectMapper.readValue(ret, // ProtocolRespDTO<BaseRespStatus> rspInfo = objectMapper.readValue(ret,
new TypeReference<ProtocolRespDTO<BaseRespStatus>>() { // new TypeReference<ProtocolRespDTO<BaseRespStatus>>() {
}); // });
//
verifyRespProtocol(rspInfo); // verifyRespProtocol(rspInfo);
//
log.debug(objectMapper.writerWithDefaultPrettyPrinter().writeValueAsString(rspInfo)); // log.debug(objectMapper.writerWithDefaultPrettyPrinter().writeValueAsString(rspInfo));
Assert.assertEquals(ErrorCode.ERR_OK.getCode(), (long) rspInfo.getMsgContent().getStatus()); // Assert.assertEquals(ErrorCode.ERR_OK.getCode(), (long) rspInfo.getMsgContent().getStatus());
} }
} }