chenlinghy
parent
1b73f87d8e
commit
0667a3d889
|
|
@ -0,0 +1,355 @@
|
|||
package com.dispose.test.controller;
|
||||
|
||||
import com.dispose.common.*;
|
||||
import com.dispose.mapper.DisposeDeviceMapper;
|
||||
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.device.manager.*;
|
||||
import com.dispose.pojo.entity.DisposeDevice;
|
||||
import com.dispose.pojo.po.MulReturnType;
|
||||
import com.dispose.test.Global.InitTestEnvironment;
|
||||
import com.fasterxml.jackson.core.type.TypeReference;
|
||||
import com.fasterxml.jackson.databind.ObjectMapper;
|
||||
import jodd.net.HttpStatus;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.junit.Assert;
|
||||
import org.junit.FixMethodOrder;
|
||||
import org.junit.Test;
|
||||
import org.junit.runner.RunWith;
|
||||
import org.junit.runners.MethodSorters;
|
||||
import org.springframework.boot.test.autoconfigure.web.servlet.AutoConfigureMockMvc;
|
||||
import org.springframework.boot.test.context.SpringBootTest;
|
||||
import org.springframework.http.MediaType;
|
||||
import org.springframework.test.annotation.DirtiesContext;
|
||||
import org.springframework.test.context.junit4.SpringRunner;
|
||||
import org.springframework.test.web.servlet.MockMvc;
|
||||
import org.springframework.test.web.servlet.request.MockMvcRequestBuilders;
|
||||
|
||||
import javax.annotation.Resource;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import java.util.stream.Collectors;
|
||||
|
||||
import static org.springframework.test.web.servlet.result.MockMvcResultHandlers.print;
|
||||
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.jsonPath;
|
||||
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
|
||||
|
||||
/**
|
||||
* The type Auth controller test.
|
||||
*
|
||||
* @author <huangxin@cmhi.chinamoblie.com>
|
||||
*/
|
||||
@AutoConfigureMockMvc
|
||||
@RunWith(SpringRunner.class)
|
||||
@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
|
||||
@DirtiesContext(classMode = DirtiesContext.ClassMode.AFTER_EACH_TEST_METHOD)
|
||||
@FixMethodOrder(MethodSorters.NAME_ASCENDING)
|
||||
@Slf4j
|
||||
public class DisposeDeviceManagerControllerTest extends InitTestEnvironment {
|
||||
/**
|
||||
* The Mock mvc.
|
||||
*/
|
||||
@Resource
|
||||
private MockMvc mockMvc;
|
||||
/**
|
||||
* The Object mapper.
|
||||
*/
|
||||
@Resource
|
||||
private ObjectMapper objectMapper;
|
||||
/**
|
||||
* The dispose device mapper.
|
||||
*/
|
||||
@Resource
|
||||
private DisposeDeviceMapper disposeDeviceMapper;
|
||||
|
||||
/**
|
||||
* A 1 add dispose device.
|
||||
*
|
||||
* @throws Exception the exception
|
||||
*/
|
||||
@Test
|
||||
public void a1_addDisposeDevice() throws Exception {
|
||||
AddDeviceReq addReq = AddDeviceReq.builder()
|
||||
.items(new ArrayList<>())
|
||||
.build();
|
||||
|
||||
List<AddCapacityInfo> newCapList = new ArrayList<>();
|
||||
|
||||
newCapList.add(AddCapacityInfo.builder()
|
||||
.capacityType(DisposeCapacityType.CLEANUP.getValue())
|
||||
.objectType(DisposeObjectType.IP.getValue())
|
||||
.ipType(IpAddrType.IPV4_IPV6.getValue())
|
||||
.protectIp("")
|
||||
.build());
|
||||
|
||||
newCapList.add(AddCapacityInfo.builder()
|
||||
.capacityType(DisposeCapacityType.BLOCKING.getValue())
|
||||
.objectType(DisposeObjectType.URL.getValue())
|
||||
.ipType(IpAddrType.IPV4_IPV6.getValue())
|
||||
.protectIp("")
|
||||
.build());
|
||||
|
||||
addReq.getItems().add(AddDeviceInfo.builder()
|
||||
.ipAddr("10.88.77.15")
|
||||
.ipPort("")
|
||||
.deviceType(DisposeDeviceType.DPTECH_UMC.getValue())
|
||||
.areaCode(0)
|
||||
.deviceName("中移杭研实验室迪普清洗设备")
|
||||
.manufacturer("DPTech")
|
||||
.model("UMC")
|
||||
.version("5.7.13")
|
||||
.userName("admin")
|
||||
.password("UMCAdministrator")
|
||||
.urlPath("UMC/service/AbnormalFlowCleaningService")
|
||||
.urlType(HttpType.HTTP.getValue())
|
||||
.readme("实验室测试设备")
|
||||
.capacity(newCapList)
|
||||
.build());
|
||||
|
||||
ProtocolReqDTO<AddDeviceReq> reqInfo = new ProtocolReqDTO<>();
|
||||
|
||||
reqInfo.setVer(ConstValue.Protocol.VERSION);
|
||||
reqInfo.setCryptoType(ProtoCryptoType.CRYPTO_NONE.getCode());
|
||||
reqInfo.setTimeStamp(System.currentTimeMillis());
|
||||
reqInfo.setMsgContent(addReq);
|
||||
|
||||
List<DisposeDevice> decsBef = disposeDeviceMapper.selectAll();
|
||||
|
||||
String ret = mockMvc.perform(MockMvcRequestBuilders
|
||||
.put("/manager/device")
|
||||
.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<AddDeviceRsp> rspInfo = objectMapper.readValue(ret,
|
||||
new TypeReference<ProtocolRespDTO<AddDeviceRsp>>() {
|
||||
});
|
||||
|
||||
verifyRespProtocol(rspInfo);
|
||||
|
||||
log.debug(objectMapper.writerWithDefaultPrettyPrinter().writeValueAsString(rspInfo));
|
||||
|
||||
List<DisposeDevice> decsAfter = disposeDeviceMapper.selectAll();
|
||||
|
||||
for (AddDeviceRet d : rspInfo.getMsgContent().getItems()
|
||||
) {
|
||||
//before: no device information after: device information exists
|
||||
boolean before = false;
|
||||
if (decsBef != null && decsBef.size() > 0) {
|
||||
for (DisposeDevice v : decsBef) {
|
||||
if (v.getIpAddr().equals(d.getIpAddr()) && v.getIpPort().equals(d.getIpPort())) {
|
||||
before = true;
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
boolean after = false;
|
||||
if (decsAfter != null && decsAfter.size() > 0) {
|
||||
for (DisposeDevice v : decsAfter) {
|
||||
if (v.getIpAddr().equals(d.getIpAddr()) && v.getIpPort().equals(d.getIpPort())) {
|
||||
after = true;
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Assert.assertNotNull(d.getIpAddr());
|
||||
Assert.assertNotNull(d.getIpPort());
|
||||
Assert.assertNotNull(d.getStatus());
|
||||
Assert.assertNotNull(d.getMessage());
|
||||
|
||||
if ((!before) && after) {
|
||||
Assert.assertEquals(Long.valueOf(d.getStatus()), Long.valueOf(ErrorCode.ERR_OK.getCode()));
|
||||
} else {
|
||||
Assert.assertNotEquals(Long.valueOf(d.getStatus()), Long.valueOf(ErrorCode.ERR_OK.getCode()));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* B 1 remove dispose device.
|
||||
*
|
||||
* @throws Exception the exception
|
||||
*/
|
||||
@Test
|
||||
public void b1_removeDisposeDevice() throws Exception {
|
||||
IdArraysReq reqData = IdArraysReq.builder()
|
||||
.id(new String[]{"1", "2"})
|
||||
.build();
|
||||
|
||||
ProtocolReqDTO<IdArraysReq> reqInfo = new ProtocolReqDTO<>();
|
||||
|
||||
reqInfo.setVer(ConstValue.Protocol.VERSION);
|
||||
reqInfo.setCryptoType(ProtoCryptoType.CRYPTO_NONE.getCode());
|
||||
reqInfo.setTimeStamp(System.currentTimeMillis());
|
||||
reqInfo.setMsgContent(reqData);
|
||||
|
||||
List<DisposeDevice> decsBef = disposeDeviceMapper.selectAll();
|
||||
|
||||
String ret = mockMvc.perform(MockMvcRequestBuilders
|
||||
.delete("/manager/device")
|
||||
.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<DeviceInfoRsp>> rspInfo = objectMapper.readValue(ret,
|
||||
new TypeReference<ProtocolRespDTO<List<DeviceInfoRsp>>>() {
|
||||
});
|
||||
|
||||
verifyRespProtocol(rspInfo);
|
||||
|
||||
log.debug(objectMapper.writerWithDefaultPrettyPrinter().writeValueAsString(rspInfo));
|
||||
|
||||
List<DisposeDevice> decsAfter = disposeDeviceMapper.selectAll();
|
||||
|
||||
for (DeviceInfoRsp d : rspInfo.getMsgContent()
|
||||
) {
|
||||
//before: device information exists after: no device information
|
||||
boolean before = false;
|
||||
if (decsBef != null && decsBef.size() > 0) {
|
||||
for (DisposeDevice v : decsBef) {
|
||||
if (v.getId().equals(Long.valueOf(d.getId()))) {
|
||||
before = true;
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
boolean after = false;
|
||||
if (decsAfter != null && decsAfter.size() > 0) {
|
||||
for (DisposeDevice v : decsAfter) {
|
||||
if (v.getId().equals(Long.valueOf(d.getId())) && (v.getStatus() == ObjectStatus.DELETED)) {
|
||||
after = true;
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Assert.assertNotNull(d.getId());
|
||||
Assert.assertNotNull(d.getStatus());
|
||||
Assert.assertNotNull(d.getMessage());
|
||||
|
||||
if (before && after) {
|
||||
Assert.assertEquals(Long.valueOf(d.getStatus()), Long.valueOf(ErrorCode.ERR_OK.getCode()));
|
||||
Assert.assertEquals(d.getDevStatus(), ObjectStatus.DELETED);
|
||||
} else {
|
||||
Assert.assertNotEquals(Long.valueOf(d.getStatus()), Long.valueOf(ErrorCode.ERR_OK.getCode()));
|
||||
Assert.assertNotEquals(d.getDevStatus(), ObjectStatus.DELETED);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* C 1 upgrade dispose device.
|
||||
*
|
||||
* @throws Exception the exception
|
||||
*/
|
||||
@Test
|
||||
public void c1_upgradeDisposeDevice() throws Exception {
|
||||
AddDeviceReq addReq = AddDeviceReq.builder()
|
||||
.items(new ArrayList<>())
|
||||
.build();
|
||||
|
||||
List<AddCapacityInfo> newCapList = new ArrayList<>();
|
||||
|
||||
List<MulReturnType<Long, String>> beforeRed = new ArrayList<>();
|
||||
|
||||
disposeDeviceMapper.selectAll().forEach(v -> {
|
||||
if (v.getDevCapacity().size() == 0) {
|
||||
newCapList.add(AddCapacityInfo.builder()
|
||||
.capacityType(DisposeCapacityType.CLEANUP.getValue())
|
||||
.objectType(DisposeObjectType.IP.getValue())
|
||||
.ipType(IpAddrType.IPV4_IPV6.getValue())
|
||||
.protectIp("")
|
||||
.build());
|
||||
}
|
||||
|
||||
beforeRed.add(new MulReturnType<>(v.getId(), v.getVersion()));
|
||||
|
||||
if (v.getVersion().equals("B5.2.27.7")) {
|
||||
v.setVersion("5.7.13");
|
||||
} else {
|
||||
v.setVersion("B5.2.27.7");
|
||||
}
|
||||
|
||||
addReq.getItems().add(AddDeviceInfo.builder()
|
||||
.ipAddr(v.getIpAddr())
|
||||
.ipPort(v.getIpPort())
|
||||
.deviceType(v.getDeviceType().getValue())
|
||||
.areaCode(v.getAreaCode())
|
||||
.deviceName(v.getDeviceName())
|
||||
.manufacturer(v.getManufacturer())
|
||||
.model(v.getModel())
|
||||
.version(v.getVersion())
|
||||
.userName(v.getUserName())
|
||||
.password(v.getPassword())
|
||||
.urlPath(v.getUrlPath())
|
||||
.urlType(v.getUrlType().getValue())
|
||||
.readme(v.getReadme())
|
||||
.capacity(newCapList)
|
||||
.build());
|
||||
});
|
||||
|
||||
ProtocolReqDTO<AddDeviceReq> reqInfo = new ProtocolReqDTO<>();
|
||||
|
||||
reqInfo.setVer(ConstValue.Protocol.VERSION);
|
||||
reqInfo.setCryptoType(ProtoCryptoType.CRYPTO_NONE.getCode());
|
||||
reqInfo.setTimeStamp(System.currentTimeMillis());
|
||||
reqInfo.setMsgContent(addReq);
|
||||
|
||||
String ret = mockMvc.perform(MockMvcRequestBuilders
|
||||
.post("/manager/upgrade")
|
||||
.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<DeviceInfoRsp>> rspInfo = objectMapper.readValue(ret,
|
||||
new TypeReference<ProtocolRespDTO<List<DeviceInfoRsp>>>() {
|
||||
});
|
||||
|
||||
verifyRespProtocol(rspInfo);
|
||||
|
||||
log.debug(objectMapper.writerWithDefaultPrettyPrinter().writeValueAsString(rspInfo));
|
||||
|
||||
for (DeviceInfoRsp deviceInfoRsp : rspInfo.getMsgContent()
|
||||
) {
|
||||
|
||||
List<MulReturnType<Long, String>> beforeDev = beforeRed.stream()
|
||||
.filter(k -> k.getFirstParam().equals(Long.valueOf(deviceInfoRsp.getId())))
|
||||
.collect(Collectors.toList());
|
||||
|
||||
DisposeDevice afterDev = disposeDeviceMapper.getDeviceById(Long.valueOf(deviceInfoRsp.getId()));
|
||||
log.debug(objectMapper.writerWithDefaultPrettyPrinter().writeValueAsString(afterDev));
|
||||
|
||||
if (deviceInfoRsp.getStatus() == ErrorCode.ERR_OK.getCode()) {
|
||||
log.info("+++++++++++++++++++++before{}", beforeDev.get(0).getSecondParam());
|
||||
log.info("+++++++++++++++++++++after{}", afterDev.getVersion());
|
||||
|
||||
Assert.assertNotNull(deviceInfoRsp);
|
||||
Assert.assertNotNull(deviceInfoRsp.getId());
|
||||
Assert.assertNotNull(deviceInfoRsp.getMessage());
|
||||
Assert.assertNotEquals(afterDev.getVersion(), beforeDev.get(0).getSecondParam());
|
||||
} else {
|
||||
Assert.assertEquals(afterDev.getVersion(), beforeDev.get(0).getSecondParam());
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue