test: mock server 补充单测

backend/services/api-test/src/test/java/io/metersphere/api/service/MockServerTestService.java

@@ -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 -> {
+            msHttpResponseList.stream()
-                if (!item.isDefaultFlag()) {
+                    .filter(item -> !item.isDefaultFlag())
-                    //特意使用非默认的响应
+                    .findFirst()
-                    mockResponse.setApiResponseId(item.getId());
+                    .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);
+                    setSendAsBody(false);
-                        this.setFile(new ApiFile());
+                    setFile(new ApiFile());
-                        this.getFile().setFileId(fileId);
+                    getFile().setFileId(fileId);
-                        this.getFile().setFileName(fileName);
+                    getFile().setFileName(fileName);
                 }});
-                    break;
+            } else if ("json".equals(returnType)) {
-                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":
                 body.setBodyType(Body.BodyType.JSON.name());
                 body.setJsonBody(new JsonBody() {{
-                        this.setJsonValue("{\"inputAge\":123, \"testKeyWord\":\"" + valueKeyWord + "\"}");
+                    setJsonValue("{\"inputAge\":123, \"testKeyWord\":\"" + valueKeyWord + "\"}");
                 }});
-                    break;
+            } else if ("xml".equals(returnType)) {
-                case "xml":
                 body.setBodyType(Body.BodyType.XML.name());
                 body.setXmlBody(new XmlBody() {{
-                        this.setValue("<xml>" + valueKeyWord + "</xml>");
+                    setValue("<xml>" + valueKeyWord + "</xml>");
                 }});
-                    break;
+            } else if ("raw".equals(returnType)) {
-                case "raw":
                 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<>() {{
+        List<MsHeader> headers = List.of(
-            this.add(new MsHeader() {{
+                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;
     }
 

backend/services/api-test/src/test/java/io/metersphere/api/socket/WebSocketHandlerTest.java

@@ -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);