Spring/spring-framework
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Per-request exchange mutating for WebTestClient

Browse Source 
Issue: SPR-15250
This commit is contained in:
Rossen Stoyanchev
2017-03-10 17:01:50 -05:00
parent f36e3d4a0d
commit f84580c32d
11 changed files with 174 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
spring-test/src/test/java/org/springframework/test/web/reactive/server/samples/bind/ApplicationContextTests.java
View File

@@ -16,7 +16,7 @@
package org.springframework.test.web.reactive.server.samples.bind;
import java.security.Principal;
import java.util.function.Function;
import java.util.function.UnaryOperator;
import org.junit.Before;
import org.junit.Test;
@@ -53,11 +53,11 @@ public class ApplicationContextTests {
context.refresh();
this.client = WebTestClient.bindToApplicationContext(context)
.exchangeMutator(identityMutator("Pablo"))
.exchangeMutator(identitySetup("Pablo"))
.build();
}
private Function<ServerWebExchange, ServerWebExchange> identityMutator(String userName) {
private UnaryOperator<ServerWebExchange> identitySetup(String userName) {
return exchange -> {
Principal user = mock(Principal.class);
when(user.getName()).thenReturn(userName);
@@ -67,13 +67,22 @@ public class ApplicationContextTests {
@Test
public void test() throws Exception {
public void basic() throws Exception {
this.client.get().uri("/test")
.exchange()
.expectStatus().isOk()
.expectBody(String.class).value().isEqualTo("Hello Pablo!");
}
@Test
public void perRequestIdentityOverride() throws Exception {
this.client.exchangeMutator(identitySetup("Giovani"))
.get().uri("/test")
.exchange()
.expectStatus().isOk()
.expectBody(String.class).value().isEqualTo("Hello Giovani!");
}
@Configuration
@EnableWebFlux

18
spring-test/src/test/java/org/springframework/test/web/reactive/server/samples/bind/ControllerTests.java
View File

@@ -16,7 +16,7 @@
package org.springframework.test.web.reactive.server.samples.bind;
import java.security.Principal;
import java.util.function.Function;
import java.util.function.UnaryOperator;
import org.junit.Before;
import org.junit.Test;
@@ -43,13 +43,12 @@ public class ControllerTests {
@Before
public void setUp() throws Exception {
this.client = WebTestClient.bindToController(new TestController())
.exchangeMutator(identityMutator("Pablo"))
.exchangeMutator(identitySetup("Pablo"))
.build();
}
private Function<ServerWebExchange, ServerWebExchange> identityMutator(String userName) {
private UnaryOperator<ServerWebExchange> identitySetup(String userName) {
return exchange -> {
Principal user = mock(Principal.class);
when(user.getName()).thenReturn(userName);
@@ -59,13 +58,22 @@ public class ControllerTests {
@Test
public void test() throws Exception {
public void basic() throws Exception {
this.client.get().uri("/test")
.exchange()
.expectStatus().isOk()
.expectBody(String.class).value().isEqualTo("Hello Pablo!");
}
@Test
public void perRequestIdentityOverride() throws Exception {
this.client.exchangeMutator(identitySetup("Giovani"))
.get().uri("/test")
.exchange()
.expectStatus().isOk()
.expectBody(String.class).value().isEqualTo("Hello Giovani!");
}
@RestController
static class TestController {