Cwikius-Spring/spring-security
1
0
Fork 0
mirror of synced 2026-08-07 02:38:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add Annotated Support for WebTestClient

Browse Source 
Fixes gh-4457
This commit is contained in:
Rob Winch
2017-07-15 21:55:23 -05:00
parent f50812c385
commit 544f39f826
5 changed files with 300 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/src/main/java/org/springframework/security/test/web/reactive/server/SecurityMockServerConfigurers.java
+11 -2
View File
@@ -25,6 +25,7 @@ import org.springframework.security.core.Authentication;
import org.springframework.security.core.GrantedAuthority;
import org.springframework.security.core.userdetails.User;
import org.springframework.security.core.userdetails.UserDetails;
import org.springframework.security.test.context.TestSecurityContextHolder;
import org.springframework.test.web.reactive.server.MockServerConfigurer;
import org.springframework.test.web.reactive.server.WebTestClient;
import org.springframework.test.web.reactive.server.WebTestClientConfigurer;
@@ -39,6 +40,7 @@ import java.util.Collection;
import java.util.List;
import java.util.function.Consumer;
import java.util.function.Function;
import java.util.function.Supplier;
/**
* Test utilities for working with Spring Security and
@@ -56,7 +58,10 @@ public class SecurityMockServerConfigurers {
public static MockServerConfigurer springSecurity() {
return new MockServerConfigurer() {
public void beforeServerCreated(WebHttpHandlerBuilder builder) {
builder.filters( filters -> filters.add(0, new MutatorFilter()));
builder.filters( filters -> {
filters.add(0, new MutatorFilter());
filters.add(0, new SetupMutatorFilter(createMutator( () -> TestSecurityContextHolder.getContext().getAuthentication())));
});
}
};
}
@@ -68,7 +73,7 @@ public class SecurityMockServerConfigurers {
* @return the {@link WebTestClientConfigurer} to use
*/
public static <T extends WebTestClientConfigurer & MockServerConfigurer> T mockPrincipal(Principal principal) {
return (T) new MutatorWebTestClientConfigurer(m -> m.mutate().principal(Mono.just(principal)).build());
return (T) new MutatorWebTestClientConfigurer(createMutator(() -> principal));
}
/**
@@ -115,6 +120,10 @@ public class SecurityMockServerConfigurers {
return new UserExchangeMutator(username);
}
private static Function<ServerWebExchange, ServerWebExchange> createMutator(Supplier<Principal> principal) {
return m -> principal.get() == null ? m : m.mutate().principal(Mono.just(principal.get())).build();
}
/**
* Updates the WebServerExchange using {@code {@link SecurityMockServerConfigurers#mockUser(UserDetails)}. Defaults to use a
* password of "password" and granted authorities of "ROLE_USER".