feat: WebReactiveIntegrationTest增强

smart-cloud · Apr 24, 2024 · 555e75d · 555e75d
1 parent b39f476
commit 555e75d
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/...core/src/main/java/io/github/smart/cloud/test/core/integration/WebMvcIntegrationTest.java b/...core/src/main/java/io/github/smart/cloud/test/core/integration/WebMvcIntegrationTest.java
@@ -64,7 +64,9 @@ public void initMock() {
             filters[i++] = entry.getValue();
         }
 
-        mockMvc = MockMvcBuilders.webAppContextSetup((WebApplicationContext) applicationContext).addFilters(filters).build();
+        mockMvc = MockMvcBuilders.webAppContextSetup((WebApplicationContext) applicationContext)
+                .addFilters(filters)
+                .build();
     }
 
     /**

diff --git a/...src/main/java/io/github/smart/cloud/test/core/integration/WebReactiveIntegrationTest.java b/...src/main/java/io/github/smart/cloud/test/core/integration/WebReactiveIntegrationTest.java
@@ -26,6 +26,7 @@
 import org.springframework.test.web.reactive.server.WebTestClient;
 
 import java.nio.charset.StandardCharsets;
+import java.time.Duration;
 import java.util.Iterator;
 import java.util.LinkedHashMap;
 import java.util.Map;
@@ -43,7 +44,10 @@ public class WebReactiveIntegrationTest extends AbstractIntegrationTest implemen
 
     @BeforeEach
     public void initMock() {
-        webTestClient = WebTestClient.bindToApplicationContext(applicationContext).build();
+        webTestClient = WebTestClient.bindToApplicationContext(applicationContext)
+                .configureClient()
+                .responseTimeout(Duration.ofSeconds(60))
+                .build();
     }
 
     @Override
@@ -104,7 +108,7 @@ public <T> T get(String url, Map<String, String> headers, Object req, TypeRefere
             }
         }
 
-        WebTestClient.RequestHeadersSpec requestHeadersSpec = webTestClient.get().uri(url, params);
+        WebTestClient.RequestHeadersSpec requestHeadersSpec = params == null ? webTestClient.get().uri(url) : webTestClient.get().uri(url, params);
         if (MapUtils.isNotEmpty(headers)) {
             headers.forEach(requestHeadersSpec::header);
         }