Просмотр исходного кода

Fix HelloWebflux tests

Fixes gh-4353
Joe Grandja 8 лет назад
Родитель
Сommit
86d0fd77a7

+ 5 - 2
samples/javaconfig/hellowebflux/src/test/java/sample/HelloWebfluxApplicationTests.java

@@ -30,6 +30,7 @@ import org.springframework.test.context.ActiveProfiles;
 import org.springframework.test.context.ContextConfiguration;
 import org.springframework.test.context.junit4.SpringRunner;
 import org.springframework.test.web.reactive.server.ExchangeResult;
+import org.springframework.test.web.reactive.server.MockServerExchangeMutator;
 import org.springframework.test.web.reactive.server.WebTestClient;
 import org.springframework.web.reactive.function.client.ExchangeFilterFunction;
 
@@ -160,8 +161,10 @@ public class HelloWebfluxApplicationTests {
 
 	@Test
 	public void mockSupport() throws Exception {
-		this.rest
-			.exchangeMutator( withUser() )
+		MockServerExchangeMutator exchangeMutator = new MockServerExchangeMutator(withUser());
+		WebTestClient mockRest = WebTestClient.bindToApplicationContext(this.context).webFilter(exchangeMutator).build();
+
+		exchangeMutator.filterClient(mockRest, withUser())
 			.get()
 			.uri("/principal")
 			.exchange()

+ 5 - 2
samples/javaconfig/hellowebfluxfn/src/test/java/sample/HelloWebfluxFnApplicationTests.java

@@ -30,6 +30,7 @@ import org.springframework.test.context.ActiveProfiles;
 import org.springframework.test.context.ContextConfiguration;
 import org.springframework.test.context.junit4.SpringRunner;
 import org.springframework.test.web.reactive.server.ExchangeResult;
+import org.springframework.test.web.reactive.server.MockServerExchangeMutator;
 import org.springframework.test.web.reactive.server.WebTestClient;
 import org.springframework.web.reactive.function.client.ExchangeFilterFunction;
 import org.springframework.web.reactive.function.server.RouterFunction;
@@ -166,8 +167,10 @@ public class HelloWebfluxFnApplicationTests {
 
 	@Test
 	public void mockSupport() throws Exception {
-		this.rest
-			.exchangeMutator( withUser() )
+		MockServerExchangeMutator exchangeMutator = new MockServerExchangeMutator(withUser());
+		WebTestClient mockRest = WebTestClient.bindToRouterFunction(this.routerFunction).webFilter(exchangeMutator).build();
+
+		exchangeMutator.filterClient(mockRest, withUser())
 			.get()
 			.uri("/principal")
 			.exchange()