fix(接口测试): 修复mock设置响应码不生效的缺陷

--bug=1040944 --user=王孝刚【接口测试】Mock-Mock的响应码设置不生效 https://www.tapd.cn/55049933/s/1518359
metersphere · May 22, 2024 · 8e098fd · 8e098fd
1 parent da98861
commit 8e098fd
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/...vices/api-test/src/main/java/io/metersphere/api/service/mockserver/MockServerService.java b/...vices/api-test/src/main/java/io/metersphere/api/service/mockserver/MockServerService.java
@@ -242,15 +242,15 @@ private ResponseEntity<?> getResponseBody(ApiDefinitionMockConfig config, String
                 case "JSON" -> responseEntity(responseCode, responseBody.getJsonBody().getJsonWithSchema(), headers);
                 case "XML" -> responseEntity(responseCode, responseBody.getXmlBody().getValue(), headers);
                 case "RAW" -> responseEntity(responseCode, responseBody.getRawBody().getValue(), headers);
-                case "BINARY" -> handleBinaryBody(responseBody, projectId, resourceId, isMock);
+                case "BINARY" -> handleBinaryBody(responseCode, responseBody, projectId, resourceId, isMock);
                 default -> responseEntity(responseCode, StringUtils.EMPTY, headers);
             };
         }
 
         return requestNotFound();
     }
 
-    private ResponseEntity<?> handleBinaryBody(ResponseBody responseBody, String projectId, String resourceId, boolean isMock) {
+    private ResponseEntity<?> handleBinaryBody(int responseCode, ResponseBody responseBody, String projectId, String resourceId, boolean isMock) {
         String fileId = responseBody.getBinaryBody().getFile().getFileId();
         String fileName = responseBody.getBinaryBody().getFile().getFileName();
         String fileType = StringUtils.substring(fileName, fileName.lastIndexOf(".") + 1);
@@ -291,7 +291,7 @@ private ResponseEntity<?> handleBinaryBody(ResponseBody responseBody, String pro
             }
         }
 
-        return ResponseEntity.ok()
+        return ResponseEntity.status(responseCode)
                 .contentType(mediaType)
                 .header(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename=\"" + fileName + "\"")
                 .body(bytes);