Add universal API interface unit testing

Author: Json031

pom.xml

@@ -5,7 +5,7 @@
   <groupId>io.github.json031</groupId>
   <artifactId>mcunittests</artifactId>
   <packaging>jar</packaging>
-  <version>1.0.3</version>
+  <version>1.0.5</version>
   <name>mcunittests</name>
   <description>mcunittests是SpringBoot项目的一个单元测试库，是一个在MIT许可下分发的开源项目</description>
   <url>https://github.com/Json031/mcunittests</url>

src/main/java/io/github/json031/App.java

@@ -1,9 +1,11 @@
 package io.github.json031;
 
 import org.junit.jupiter.api.Assertions;
+import org.springframework.http.*;
 import org.springframework.web.client.RestTemplate;
 
 import java.time.Duration;
+import java.util.Map;
 
 public class MCApiTests {
 
@@ -12,42 +14,88 @@ public class MCApiTests {
     /**
      * 通用测试方法，验证给定 API 是否能在 timeout 秒内响应
      *
-     * @param url     完整的 API 地址（包括 http/https）
+     * @param url           完整的 API 地址（包括 http/https）
+     * @param method        请求方式（GET / POST）
+     * @param params        请求参数（POST body 或 GET 查询参数）
+     * @param headers       请求头（可选）
      * @param timeout 最大允许超时时间（秒）
-     * @return durationMillis 响应时间（毫秒）
+     * @param verbose       是否打印响应
+     * @return 响应时间（毫秒）
      */
-    public long assertApiRespondsWithinTimeout(String url, long timeout, boolean verbose) {
-        return this.assertApiRespondsWithinTimeoutMillis(url, timeout * 1000, verbose);
+    public long assertApiRespondsWithinTimeout(String url,
+                                               HttpMethod method,
+                                               Map<String, Object> params,
+                                               Map<String, String> headers,
+                                               long timeout,
+                                               boolean verbose) {
+        return this.assertApiRespondsWithinTimeoutMillis(url, method, params, headers, timeout * 1000, verbose);
     }
 
     /**
-     * 通用测试方法，验证给定 API 是否能在 timeout 秒内响应
+     * 通用测试方法，验证给定 API 是否能在 timeoutMillis 内响应。
      *
-     * @param url     完整的 API 地址（包括 http/https）
-     * @param timeoutMillis 最大允许超时时间（毫秒）
-     * @return durationMillis 响应时间（毫秒）
+     * @param url           完整的 API 地址（包括 http/https）
+     * @param method        请求方式（GET / POST）
+     * @param params        请求参数（POST body 或 GET 查询参数）
+     * @param headers       请求头（可选）
+     * @param timeoutMillis 超时时间（毫秒）
+     * @param verbose       是否打印响应
+     * @return 响应时间（毫秒）
      */
-    public long assertApiRespondsWithinTimeoutMillis(String url, long timeoutMillis, boolean verbose) {
+    public long assertApiRespondsWithinTimeoutMillis(String url,
+                                                     HttpMethod method,
+                                                     Map<String, Object> params,
+                                                     Map<String, String> headers,
+                                                     long timeoutMillis,
+                                                     boolean verbose) {
         try {
-            long start = System.nanoTime();  // 记录开始时间
+            HttpHeaders httpHeaders = new HttpHeaders();
+            if (headers != null) {
+                headers.forEach(httpHeaders::set);
+            }
+
+            HttpEntity<?> entity;
+            String finalUrl = url;
+
+            if (method == HttpMethod.GET && params != null && !params.isEmpty()) {
+                // 拼接 GET 参数
+                StringBuilder queryBuilder = new StringBuilder(url);
+                queryBuilder.append(url.contains("?") ? "&" : "?");
+                params.forEach((key, value) -> queryBuilder.append(key).append("=").append(value).append("&"));
+                finalUrl = queryBuilder.substring(0, queryBuilder.length() - 1); // 去掉最后一个 &
+                entity = new HttpEntity<>(httpHeaders);
+            } else {
+                // POST 请求体（可为 null）
+                entity = new HttpEntity<>(params, httpHeaders);
+            }
 
-            String response = this.restTemplate.getForObject(url, String.class);
+            long start = System.nanoTime();
 
-            long end = System.nanoTime();  // 记录结束时间
+            ResponseEntity<String> response = restTemplate.exchange(
+                    finalUrl,
+                    method,
+                    entity,
+                    String.class
+            );
+
+            long end = System.nanoTime();
             long durationMillis = (end - start) / 1_000_000;
+
             if (verbose) {
-                System.out.println("API Response: " + response);
+                System.out.println("API Response: " + response.getBody());
                 System.out.println("Response time: " + durationMillis + " ms");
             }
 
-            if (durationMillis >= timeoutMillis) {
-                Assertions.fail("API did not respond within " + timeoutMillis + " milliseconds, url: " +url);
+            if (durationMillis > timeoutMillis) {
+                Assertions.fail("API did not respond within " + timeoutMillis + " ms, url: " + finalUrl);
             }
 
             return durationMillis;
+
         } catch (Exception e) {
-            Assertions.fail("Connection refused when trying to access: " + url);
+            Assertions.fail("API call failed for: " + url + " with error: " + e.getMessage());
         }
+
         return -1;
     }
 }

src/main/java/io/github/json031/MCApiTests.java

@@ -2,10 +2,12 @@
 
 import io.github.json031.JavaBean.HighConcurrencyResult;
 import org.junit.jupiter.api.Assertions;
+import org.springframework.http.HttpMethod;
 import org.springframework.web.client.RestTemplate;
 
 import java.util.ArrayList;
 import java.util.List;
+import java.util.Map;
 import java.util.concurrent.*;
 import java.util.concurrent.atomic.AtomicInteger;
 import java.util.function.Supplier;
@@ -21,7 +23,13 @@ public class MCHighConcurrencyTests {
      * @param threadCount 并发线程数
      * @return 统计结果
      */
-    public HighConcurrencyResult highConcurrencyTest(String url, int threadCount, long timeoutMillis, boolean verbose) {
+    public HighConcurrencyResult highConcurrencyTest(String url,
+                                                     int threadCount,
+                                                     HttpMethod method,
+                                                     Map<String, Object> params,
+                                                     Map<String, String> headers,
+                                                     long timeoutMillis,
+                                                     boolean verbose) {
         ExecutorService executor = Executors.newFixedThreadPool(threadCount);
         List<Future<Long>> futures = new ArrayList<>();
         AtomicInteger successCount = new AtomicInteger();
@@ -31,7 +39,7 @@ public HighConcurrencyResult highConcurrencyTest(String url, int threadCount, lo
             futures.add(executor.submit(() -> {
                 long start = System.nanoTime();
                 try {
-                    long rsptime = this.mcApiTests.assertApiRespondsWithinTimeoutMillis(url, timeoutMillis, verbose); // 执行请求
+                    long rsptime = this.mcApiTests.assertApiRespondsWithinTimeoutMillis(url, method, params, headers, timeoutMillis, verbose); // 执行请求
                     System.out.println("API Response time: " + rsptime);
                     if (rsptime >= timeoutMillis) {
                         failCount.incrementAndGet();