Api automated unit testing.

Author: Json031

.gitignore

@@ -15,3 +15,4 @@ buildNumber.properties
 .project
 # JDT-specific (Eclipse Java Development Tools)
 .classpath
+/.idea

pom.xml

@@ -60,19 +60,19 @@
 
   <!-- 只保留 GitHub Packages 配置 -->
   <distributionManagement>
-<!--    <repository>-->
-<!--      <id>github</id>-->
-<!--      <name>GitHub Json031 Apache Maven Packages</name>-->
-<!--      <url>https://maven.pkg.github.com/json031/mcunittests</url>-->
-<!--    </repository>-->
+    <repository>
+      <id>github</id>
+      <name>GitHub Json031 Apache Maven Packages</name>
+      <url>https://maven.pkg.github.com/json031/mcunittests</url>
+    </repository>
     <snapshotRepository>
       <id>ossrh</id>
       <url>https://s01.oss.sonatype.org/content/repositories/snapshots</url>
     </snapshotRepository>
-    <repository>
-      <id>ossrh</id>
-      <url>https://s01.oss.sonatype.org/service/local/staging/deploy/maven2/</url>
-    </repository>
+<!--    <repository>-->
+<!--      <id>ossrh</id>-->
+<!--      <url>https://s01.oss.sonatype.org/service/local/staging/deploy/maven2/</url>-->
+<!--    </repository>-->
   </distributionManagement>
 
   <build>

src/main/java/io/github/json031/JavaBean/HighConcurrencyResult.java

@@ -0,0 +1,24 @@
+package io.github.json031.JavaBean;
+
+public class HighConcurrencyResult {
+    public final int total;
+    public final int success;
+    public final int failed;
+    public final long avgResponseTimeMillis;
+
+    public HighConcurrencyResult(int total, int success, int failed, long avgResponseTimeMillis) {
+        this.total = total;
+        this.success = success;
+        this.failed = failed;
+        this.avgResponseTimeMillis = avgResponseTimeMillis;
+    }
+
+    @Override
+    public String toString() {
+        return "=== Load Test Result ===\n" +
+                "Total Requests: " + total + "\n" +
+                "Successful:     " + success + "\n" +
+                "Failed:         " + failed + "\n" +
+                "Avg Time (ms):  " + avgResponseTimeMillis;
+    }
+}

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

@@ -14,15 +14,40 @@ public class MCApiTests {
      *
      * @param url     完整的 API 地址（包括 http/https）
      * @param timeout 最大允许超时时间（秒）
+     * @return durationMillis 响应时间（毫秒）
      */
-    public void assertApiRespondsWithinTimeout(String url, long timeout) {
+    public long assertApiRespondsWithinTimeout(String url, long timeout, boolean verbose) {
+        return this.assertApiRespondsWithinTimeoutMillis(url, timeout * 1000, verbose);
+    }
+
+    /**
+     * 通用测试方法，验证给定 API 是否能在 timeout 秒内响应
+     *
+     * @param url     完整的 API 地址（包括 http/https）
+     * @param timeoutMillis 最大允许超时时间（毫秒）
+     * @return durationMillis 响应时间（毫秒）
+     */
+    public long assertApiRespondsWithinTimeoutMillis(String url, long timeoutMillis, boolean verbose) {
         try {
-            Assertions.assertTimeout(Duration.ofSeconds(timeout), () -> {
-                String response = restTemplate.getForObject(url, String.class);
+            long start = System.nanoTime();  // 记录开始时间
+
+            String response = this.restTemplate.getForObject(url, String.class);
+
+            long end = System.nanoTime();  // 记录结束时间
+            long durationMillis = (end - start) / 1_000_000;
+            if (verbose) {
                 System.out.println("API Response: " + response);
-            }, "API did not respond within " + timeout + " seconds");
+                System.out.println("Response time: " + durationMillis + " ms");
+            }
+
+            if (durationMillis >= timeoutMillis) {
+                Assertions.fail("API did not respond within " + timeoutMillis + " milliseconds, url: " +url);
+            }
+
+            return durationMillis;
         } catch (Exception e) {
             Assertions.fail("Connection refused when trying to access: " + url);
         }
+        return -1;
     }
 }

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

@@ -0,0 +1,69 @@
+package io.github.json031;
+
+import io.github.json031.JavaBean.HighConcurrencyResult;
+import org.junit.jupiter.api.Assertions;
+import org.springframework.web.client.RestTemplate;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.concurrent.*;
+import java.util.concurrent.atomic.AtomicInteger;
+import java.util.function.Supplier;
+
+public class MCHighConcurrencyTests {
+
+    private final RestTemplate restTemplate = new RestTemplate();
+    private final MCApiTests mcApiTests = new MCApiTests();
+
+    /**
+     * 运行并发测试
+     * @param url        请求地址
+     * @param threadCount 并发线程数
+     * @return 统计结果
+     */
+    public HighConcurrencyResult highConcurrencyTest(String url, int threadCount, long timeoutMillis, boolean verbose) {
+        ExecutorService executor = Executors.newFixedThreadPool(threadCount);
+        List<Future<Long>> futures = new ArrayList<>();
+        AtomicInteger successCount = new AtomicInteger();
+        AtomicInteger failCount = new AtomicInteger();
+
+        for (int i = 0; i < threadCount; i++) {
+            futures.add(executor.submit(() -> {
+                long start = System.nanoTime();
+                try {
+                    long rsptime = this.mcApiTests.assertApiRespondsWithinTimeoutMillis(url, timeoutMillis, verbose); // 执行请求
+                    System.out.println("API Response time: " + rsptime);
+                    if (rsptime >= timeoutMillis) {
+                        failCount.incrementAndGet();
+                    } else {
+                        successCount.incrementAndGet();
+                    }
+                } catch (Exception e) {
+                    failCount.incrementAndGet();
+                }
+                return (System.nanoTime() - start) / 1_000_000;
+            }));
+        }
+
+        executor.shutdown();
+        try {
+            executor.awaitTermination(60, TimeUnit.SECONDS);
+        } catch (Exception e) {
+            Assertions.fail("High concurrency test failed for url: " + url);
+        }
+
+        long totalTime = 0;
+        int completed = 0;
+        for (Future<Long> future : futures) {
+            try {
+                long time = future.get();
+                totalTime += time;
+                completed++;
+            } catch (Exception ignored) {
+            }
+        }
+
+        long avg = completed > 0 ? totalTime / completed : -1;
+        return new HighConcurrencyResult(threadCount, successCount.get(), failCount.get(), avg);
+    }
+}