Add unit tests

Author: Json031

pom.xml

@@ -5,7 +5,7 @@
   <groupId>io.github.json031</groupId>
   <artifactId>mcunittests</artifactId>
   <packaging>jar</packaging>
-  <version>1.0.26</version>
+  <version>1.0.27</version>
   <name>mcunittests</name>
   <description>mcunittests是SpringBoot项目的一个单元测试库，是一个在MIT许可下分发的开源项目</description>
   <url>https://github.com/Json031/mcunittests</url>
@@ -45,6 +45,14 @@
       <artifactId>jackson-databind</artifactId>
       <version>2.15.0</version>
     </dependency>
+
+    <!-- Mockito core -->
+    <dependency>
+      <groupId>org.mockito</groupId>
+      <artifactId>mockito-core</artifactId>
+      <version>5.11.0</version>
+      <scope>test</scope>
+    </dependency>
   </dependencies>
 
   <licenses>

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

@@ -15,7 +15,7 @@
  */
 public class MCHighConcurrencyTests {
 
-    private final MCApiTests mcApiTests = new MCApiTests();
+    public MCApiTests mcApiTests = new MCApiTests();
 
     /**
      * 运行并发测试

src/test/java/io/github/json031/AppTest.java

@@ -0,0 +1,61 @@
+package io.github.json031;
+
+import io.github.json031.JavaBean.RequestUnitTestsResult;
+import io.github.json031.unittests.DataUnitTests;
+import io.github.json031.unittests.RequestUnitTests;
+import org.junit.jupiter.api.Test;
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.HttpMethod;
+import org.springframework.http.MediaType;
+import org.springframework.http.ResponseEntity;
+
+import static org.junit.jupiter.api.Assertions.*;
+
+public class DataUnitTestsTest {
+
+    @Test
+    public void testIsValidJSON() {
+        HttpHeaders headers = new HttpHeaders();
+        headers.setContentType(MediaType.APPLICATION_JSON);
+        ResponseEntity<String> response = ResponseEntity.ok()
+                .headers(headers)
+                .body("{\"key\":\"value\"}");
+
+        assertTrue(DataUnitTests.isValidJSON(response));
+    }
+
+    @Test
+    public void testIsValidUrl() {
+        DataUnitTests.isValidUrl("https://www.baidu.com");
+    }
+
+    @Test
+    public void testIsJSONContentType_ResponseEntity() {
+        HttpHeaders headers = new HttpHeaders();
+        headers.setContentType(MediaType.APPLICATION_JSON);
+        ResponseEntity<String> response = ResponseEntity.ok()
+                .headers(headers)
+                .body("{\"key\":\"value\"}");
+
+        assertTrue(DataUnitTests.isJSONContentType(response));
+    }
+
+    @Test
+    public void testIsJSONContentType_MediaType() {
+        assertTrue(DataUnitTests.isJSONContentType(MediaType.APPLICATION_JSON));
+        assertFalse(DataUnitTests.isJSONContentType(MediaType.TEXT_PLAIN));
+    }
+
+    @Test
+    public void testRequestUnitTests() {
+        String url = "https://www.baidu.com";
+        // 调用请求方法
+        RequestUnitTestsResult result = RequestUnitTests.requestWitRestTemplate(url, HttpMethod.GET, null, null, true);
+
+        // 验证结果
+        assertNotNull(result);
+        assertTrue(result.durationMillis < 2000);
+        assertEquals(200, result.response.getStatusCode().value());
+    }
+}
+

src/test/java/io/github/json031/DataUnitTestsTest.java

@@ -0,0 +1,91 @@
+package io.github.json031;
+
+import io.github.json031.JavaBean.HighConcurrencyResult;
+import io.github.json031.apitests.MCHighConcurrencyTests;
+import io.github.json031.apitests.MCApiTests;
+import org.junit.jupiter.api.Test;
+import org.mockito.Mockito;
+import org.springframework.http.HttpMethod;
+
+import java.util.HashMap;
+import java.util.Map;
+
+import static org.junit.jupiter.api.Assertions.*;
+
+public class MCHighConcurrencyTestsTest {
+
+    @Test
+    public void testHighConcurrencyTestWithTimeoutSeconds() {
+        // 准备一个假的 MCHighConcurrencyTests 对象，内部的 MCApiTests 被 mock
+        MCHighConcurrencyTests tests = new MCHighConcurrencyTests() {
+            @Override
+            public HighConcurrencyResult highConcurrencyTestWithTimeoutMillis(String url, int threadCount, HttpMethod method,
+                                                                              Map<String, Object> params, Map<String, String> headers,
+                                                                              long timeoutMillis, boolean verbose) {
+                return new HighConcurrencyResult(threadCount, threadCount, 0, 100); // 假设全部成功
+            }
+        };
+
+        Map<String, Object> params = new HashMap<>();
+        params.put("key", "value");
+
+        HighConcurrencyResult result = tests.highConcurrencyTestWithTimeoutSeconds(
+                "https://www.baidu.com",
+                5,
+                HttpMethod.GET,
+                params,
+                null,
+                2, // 2秒超时
+                false
+        );
+
+        assertNotNull(result);
+        assertEquals(5, result.total);
+        assertEquals(5, result.success);
+        assertEquals(0, result.failed);
+        assertTrue(result.avgResponseTimeMillis > 0);
+        assertTrue(result.avgResponseTimeMillis < 2000);
+    }
+
+    @Test
+    public void testHighConcurrencyTestWithTimeoutMillis_RealisticMock() {
+        // 这里对 MCApiTests 进行 mock（为了不真的去请求网络）
+        MCHighConcurrencyTests tests = new MCHighConcurrencyTests();
+        MCApiTests mockApiTests = Mockito.mock(MCApiTests.class);
+        tests.mcApiTests = mockApiTests;
+
+        try {
+            // 设定 mock 返回固定的响应时间
+            Mockito.when(mockApiTests.assertApiRespondsWithinTimeoutMillis(
+                    Mockito.anyString(),
+                    Mockito.any(HttpMethod.class),
+                    Mockito.anyMap(),
+                    Mockito.anyMap(),
+                    Mockito.anyLong(),
+                    Mockito.anyBoolean()
+            )).thenReturn(50L); // 50ms
+
+            Map<String, Object> params = new HashMap<>();
+            params.put("test", "value");
+
+            HighConcurrencyResult result = tests.highConcurrencyTestWithTimeoutMillis(
+                    "http://example.com/api",
+                    3,
+                    HttpMethod.GET,
+                    params,
+                    null,
+                    1000,
+                    false
+            );
+
+            assertNotNull(result);
+            assertEquals(3, result.total);
+            assertEquals(3, result.success);
+            assertEquals(0, result.failed);
+            assertTrue(result.avgResponseTimeMillis > 0);
+            assertTrue(result.avgResponseTimeMillis < 2000);
+        } catch (Exception e) {
+            fail("Exception thrown in highConcurrencyTestWithTimeoutMillis: " + e.getMessage());
+        }
+    }
+}