test: mock server 补充单测
This commit is contained in:
fit2-zhao
Craftsman
parent
b9303de3bd
commit
a5156299df
|
|
@ -86,6 +86,7 @@ public class MockServerTestService {
|
|||
if (fileIds != null) {
|
||||
ApiFileResourceService apiFileResourceService = CommonBeanFactory.getBean(ApiFileResourceService.class);
|
||||
// 验证文件的关联关系，以及是否存入对象存储
|
||||
assert apiFileResourceService != null;
|
||||
List<ApiFileResource> apiFileResources = apiFileResourceService.getByResourceId(id);
|
||||
Assertions.assertEquals(apiFileResources.size(), fileIds.size());
|
||||
|
||||
|
|
@ -114,6 +115,7 @@ public class MockServerTestService {
|
|||
*/
|
||||
public static void assertLinkFile(String id, List<String> fileIds) {
|
||||
FileAssociationService fileAssociationService = CommonBeanFactory.getBean(FileAssociationService.class);
|
||||
assert fileAssociationService != null;
|
||||
List<String> linkFileIds = fileAssociationService.getFiles(id)
|
||||
.stream()
|
||||
.map(FileInfo::getFileId)
|
||||
|
|
@ -254,74 +256,62 @@ public class MockServerTestService {
|
|||
public MockResponse genMockResponse(String returnType, int status, String valueKeyWord, String fileId, String fileName, ApiDefinitionBlob apiDefinitionBlob) {
|
||||
MockResponse mockResponse = new MockResponse();
|
||||
mockResponse.setStatusCode(status);
|
||||
|
||||
if (apiDefinitionBlob != null) {
|
||||
mockResponse.setUseApiResponse(true);
|
||||
List<HttpResponse> msHttpResponseList = JSON.parseArray(new String(apiDefinitionBlob.getResponse()), HttpResponse.class);
|
||||
msHttpResponseList.forEach(item -> {
|
||||
if (!item.isDefaultFlag()) {
|
||||
//特意使用非默认的响应
|
||||
mockResponse.setApiResponseId(item.getId());
|
||||
}
|
||||
});
|
||||
msHttpResponseList.stream()
|
||||
.filter(item -> !item.isDefaultFlag())
|
||||
.findFirst()
|
||||
.ifPresent(item -> mockResponse.setApiResponseId(item.getId()));
|
||||
} else {
|
||||
ResponseBody body = new ResponseBody();
|
||||
switch (returnType) {
|
||||
case "file":
|
||||
|
||||
if ("file".equals(returnType) || "file-body".equals(returnType)) {
|
||||
body.setBodyType(Body.BodyType.BINARY.name());
|
||||
body.setBinaryBody(new ResponseBinaryBody() {{
|
||||
this.setSendAsBody(false);
|
||||
this.setFile(new ApiFile());
|
||||
this.getFile().setFileId(fileId);
|
||||
this.getFile().setFileName(fileName);
|
||||
setSendAsBody(false);
|
||||
setFile(new ApiFile());
|
||||
getFile().setFileId(fileId);
|
||||
getFile().setFileName(fileName);
|
||||
}});
|
||||
break;
|
||||
case "file-body":
|
||||
body.setBodyType(Body.BodyType.BINARY.name());
|
||||
body.setBinaryBody(new ResponseBinaryBody() {{
|
||||
this.setSendAsBody(false);
|
||||
this.setFile(new ApiFile());
|
||||
this.getFile().setFileId(fileId);
|
||||
this.getFile().setFileName(fileName);
|
||||
}});
|
||||
break;
|
||||
case "json":
|
||||
} else if ("json".equals(returnType)) {
|
||||
body.setBodyType(Body.BodyType.JSON.name());
|
||||
body.setJsonBody(new JsonBody() {{
|
||||
this.setJsonValue("{\"inputAge\":123, \"testKeyWord\":\"" + valueKeyWord + "\"}");
|
||||
setJsonValue("{\"inputAge\":123, \"testKeyWord\":\"" + valueKeyWord + "\"}");
|
||||
}});
|
||||
break;
|
||||
case "xml":
|
||||
} else if ("xml".equals(returnType)) {
|
||||
body.setBodyType(Body.BodyType.XML.name());
|
||||
body.setXmlBody(new XmlBody() {{
|
||||
this.setValue("<xml>" + valueKeyWord + "</xml>");
|
||||
setValue("<xml>" + valueKeyWord + "</xml>");
|
||||
}});
|
||||
break;
|
||||
case "raw":
|
||||
} else if ("raw".equals(returnType)) {
|
||||
body.setBodyType(Body.BodyType.RAW.name());
|
||||
body.setRawBody(new RawBody() {{
|
||||
this.setValue("Raw body content:" + valueKeyWord);
|
||||
setValue("Raw body content:" + valueKeyWord);
|
||||
}});
|
||||
break;
|
||||
}
|
||||
|
||||
mockResponse.setBody(body);
|
||||
}
|
||||
|
||||
List<MsHeader> headers = new ArrayList<>() {{
|
||||
this.add(new MsHeader() {{
|
||||
List<MsHeader> headers = List.of(
|
||||
new MsHeader() {{
|
||||
this.setKey("rspHeaderA");
|
||||
this.setValue("header-1");
|
||||
}});
|
||||
this.add(new MsHeader() {{
|
||||
}},
|
||||
new MsHeader() {{
|
||||
this.setKey("rspHeaderB");
|
||||
this.setValue("header-2");
|
||||
this.setEnable(false);
|
||||
}});
|
||||
this.add(new MsHeader() {{
|
||||
}},
|
||||
new MsHeader() {{
|
||||
this.setKey("rspHeaderC");
|
||||
this.setValue("header-3");
|
||||
}});
|
||||
}};
|
||||
}}
|
||||
);
|
||||
mockResponse.setHeaders(headers);
|
||||
|
||||
return mockResponse;
|
||||
}
|
||||
|
||||
|
|
|
|||
|
|
@ -34,7 +34,7 @@ public class WebSocketHandlerTest {
|
|||
|
||||
@Test
|
||||
@Order(1)
|
||||
void testOpenSession() throws IOException {
|
||||
void testOpenSession() {
|
||||
// 模拟参数
|
||||
String reportId = "123";
|
||||
when(session.getAsyncRemote()).thenReturn(async);
|
||||
|
|
@ -57,14 +57,11 @@ public class WebSocketHandlerTest {
|
|||
|
||||
// 调用被测试方法
|
||||
webSocketHandler.onMessage(reportId, message);
|
||||
|
||||
// 验证行为
|
||||
// 这里可以添加更多的验证，例如检查 sendMessageSingle 的调用情况等
|
||||
}
|
||||
|
||||
@Test
|
||||
@Order(3)
|
||||
void testOnHeartbeatCheck() throws IOException {
|
||||
void testOnHeartbeatCheck() {
|
||||
// 模拟参数
|
||||
when(session.getAsyncRemote()).thenReturn(async);
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue