13 changed files with 948 additions and 14 deletions
@ -0,0 +1,59 @@
@@ -0,0 +1,59 @@
|
||||
/* |
||||
* Copyright 2002-2017 the original author or authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
package org.springframework.security.oauth2.jwt; |
||||
|
||||
import static org.assertj.core.api.Assertions.assertThat; |
||||
|
||||
import org.junit.Test; |
||||
|
||||
/** |
||||
* Tests for {@link Jwt.Builder}. |
||||
*/ |
||||
public class JwtBuilderTests { |
||||
|
||||
@Test() |
||||
public void builderCanBeReused() { |
||||
final Jwt.Builder<?> tokensBuilder = Jwt.builder(); |
||||
|
||||
final Jwt first = tokensBuilder |
||||
.tokenValue("V1") |
||||
.header("TEST_HEADER_1", "H1") |
||||
.claim("TEST_CLAIM_1", "C1") |
||||
.build(); |
||||
|
||||
final Jwt second = tokensBuilder |
||||
.tokenValue("V2") |
||||
.header("TEST_HEADER_1", "H2") |
||||
.header("TEST_HEADER_2", "H3") |
||||
.claim("TEST_CLAIM_1", "C2") |
||||
.claim("TEST_CLAIM_2", "C3") |
||||
.build(); |
||||
|
||||
assertThat(first.getHeaders()).hasSize(1); |
||||
assertThat(first.getHeaders().get("TEST_HEADER_1")).isEqualTo("H1"); |
||||
assertThat(first.getClaims()).hasSize(1); |
||||
assertThat(first.getClaims().get("TEST_CLAIM_1")).isEqualTo("C1"); |
||||
assertThat(first.getTokenValue()).isEqualTo("V1"); |
||||
|
||||
assertThat(second.getHeaders()).hasSize(2); |
||||
assertThat(second.getHeaders().get("TEST_HEADER_1")).isEqualTo("H2"); |
||||
assertThat(second.getHeaders().get("TEST_HEADER_2")).isEqualTo("H3"); |
||||
assertThat(second.getClaims()).hasSize(2); |
||||
assertThat(second.getClaims().get("TEST_CLAIM_1")).isEqualTo("C2"); |
||||
assertThat(second.getClaims().get("TEST_CLAIM_2")).isEqualTo("C3"); |
||||
assertThat(second.getTokenValue()).isEqualTo("V2"); |
||||
} |
||||
} |
||||
@ -0,0 +1,71 @@
@@ -0,0 +1,71 @@
|
||||
/* |
||||
* Copyright 2002-2019 the original author or authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
package sample; |
||||
|
||||
import static org.hamcrest.CoreMatchers.is; |
||||
import static org.springframework.security.test.web.servlet.request.SecurityMockMvcRequestPostProcessors.jwt; |
||||
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get; |
||||
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.content; |
||||
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status; |
||||
|
||||
import org.junit.Test; |
||||
import org.junit.runner.RunWith; |
||||
import org.springframework.beans.factory.annotation.Autowired; |
||||
import org.springframework.boot.test.autoconfigure.web.servlet.WebMvcTest; |
||||
import org.springframework.boot.test.mock.mockito.MockBean; |
||||
import org.springframework.security.core.authority.SimpleGrantedAuthority; |
||||
import org.springframework.security.oauth2.jwt.JwtDecoder; |
||||
import org.springframework.test.context.junit4.SpringRunner; |
||||
import org.springframework.test.web.servlet.MockMvc; |
||||
|
||||
/** |
||||
* |
||||
* @author Jérôme Wacongne <ch4mp@c4-soft.com> |
||||
* @since 5.2.0 |
||||
* |
||||
*/ |
||||
@RunWith(SpringRunner.class) |
||||
@WebMvcTest(OAuth2ResourceServerController.class) |
||||
public class OAuth2ResourceServerControllerTests { |
||||
|
||||
@Autowired |
||||
MockMvc mockMvc; |
||||
|
||||
@MockBean |
||||
JwtDecoder jwtDecoder; |
||||
|
||||
@Test |
||||
public void indexGreetsAuthenticatedUser() throws Exception { |
||||
mockMvc.perform(get("/").with(jwt().name("ch4mpy"))) |
||||
.andExpect(content().string(is("Hello, ch4mpy!"))); |
||||
} |
||||
|
||||
@Test |
||||
public void messageCanBeReadWithScopeMessageReadAuthority() throws Exception { |
||||
mockMvc.perform(get("/message").with(jwt().scopes("message:read"))) |
||||
.andExpect(content().string(is("secret message"))); |
||||
|
||||
mockMvc.perform(get("/message").with(jwt().authorities(new SimpleGrantedAuthority(("SCOPE_message:read"))))) |
||||
.andExpect(content().string(is("secret message"))); |
||||
} |
||||
|
||||
@Test |
||||
public void messageCanNotBeReadWithoutScopeMessageReadAuthority() throws Exception { |
||||
mockMvc.perform(get("/message").with(jwt())) |
||||
.andExpect(status().isForbidden()); |
||||
} |
||||
|
||||
} |
||||
@ -0,0 +1,140 @@
@@ -0,0 +1,140 @@
|
||||
/* |
||||
* Copyright 2002-2019 the original author or authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with |
||||
* the License. You may obtain a copy of the License at |
||||
* |
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on |
||||
* an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the |
||||
* specific language governing permissions and limitations under the License. |
||||
*/ |
||||
package org.springframework.security.test.support; |
||||
|
||||
import java.util.Collection; |
||||
import java.util.HashSet; |
||||
import java.util.Set; |
||||
import java.util.stream.Collectors; |
||||
import java.util.stream.Stream; |
||||
|
||||
import org.springframework.core.convert.converter.Converter; |
||||
import org.springframework.security.core.GrantedAuthority; |
||||
import org.springframework.security.core.authority.SimpleGrantedAuthority; |
||||
import org.springframework.security.oauth2.jwt.Jwt; |
||||
import org.springframework.security.oauth2.jwt.JwtClaimNames; |
||||
import org.springframework.security.oauth2.server.resource.authentication.JwtAuthenticationToken; |
||||
import org.springframework.security.oauth2.server.resource.authentication.JwtGrantedAuthoritiesConverter; |
||||
import org.springframework.util.StringUtils; |
||||
|
||||
/** |
||||
* @author Jérôme Wacongne <ch4mp@c4-soft.com> |
||||
* @since 5.2 |
||||
*/ |
||||
public class JwtAuthenticationTokenTestingBuilder<T extends JwtAuthenticationTokenTestingBuilder<T>> |
||||
extends |
||||
JwtAuthenticationToken.Builder<T> { |
||||
|
||||
private static final String[] DEFAULT_SCOPES = { "USER" }; |
||||
|
||||
private final Set<GrantedAuthority> addedAuthorities; |
||||
|
||||
public JwtAuthenticationTokenTestingBuilder(Converter<Jwt, Collection<GrantedAuthority>> authoritiesConverter) { |
||||
super(new JwtTestingBuilder(), authoritiesConverter); |
||||
this.addedAuthorities = new HashSet<>(); |
||||
scopes(DEFAULT_SCOPES); |
||||
} |
||||
|
||||
public JwtAuthenticationTokenTestingBuilder() { |
||||
this(new JwtGrantedAuthoritiesConverter()); |
||||
} |
||||
|
||||
/** |
||||
* How to extract authorities from token |
||||
* @param authoritiesConverter JWT to granted-authorities converter |
||||
* @return this builder to further configure |
||||
*/ |
||||
public T authorities(Converter<Jwt, Collection<GrantedAuthority>> authoritiesConverter) { |
||||
return authoritiesConverter(authoritiesConverter); |
||||
} |
||||
|
||||
/** |
||||
* Adds authorities to what is extracted from the token.<br> |
||||
* Please consider using {@link #authorities(Converter)} instead. |
||||
* @param authorities authorities to add to token ones |
||||
* @return this builder to further configure |
||||
*/ |
||||
public T authorities(Stream<GrantedAuthority> authorities) { |
||||
addedAuthorities.addAll(authorities.collect(Collectors.toSet())); |
||||
return downcast(); |
||||
} |
||||
|
||||
/** |
||||
* Adds authorities to what is extracted from the token.<br> |
||||
* Please consider using {@link #authorities(Converter)} instead. |
||||
* @param authorities authorities to add to token ones |
||||
* @return this builder to further configure |
||||
*/ |
||||
public T authorities(GrantedAuthority... authorities) { |
||||
return authorities(Stream.of(authorities)); |
||||
} |
||||
|
||||
/** |
||||
* Adds authorities to what is extracted from the token.<br> |
||||
* Please consider using {@link #authorities(Converter)} instead. |
||||
* @param authorities authorities to add to token ones |
||||
* @return this builder to further configure |
||||
*/ |
||||
public T authorities(String... authorities) { |
||||
return authorities(Stream.of(authorities).map(SimpleGrantedAuthority::new)); |
||||
} |
||||
|
||||
@Override |
||||
public JwtAuthenticationToken build() { |
||||
final Jwt token = getToken(); |
||||
|
||||
return new JwtAuthenticationToken(token, getAuthorities(token)); |
||||
} |
||||
|
||||
@Override |
||||
protected Collection<GrantedAuthority> getAuthorities(Jwt token) { |
||||
final Collection<GrantedAuthority> principalAuthorities = super.getAuthorities(token); |
||||
|
||||
return addedAuthorities.isEmpty() ? principalAuthorities |
||||
: Stream.concat(principalAuthorities.stream(), addedAuthorities.stream()).collect(Collectors.toSet()); |
||||
} |
||||
|
||||
/** |
||||
* @author Jérôme Wacongne <ch4mp@c4-soft.com> |
||||
* @since 5.2 |
||||
*/ |
||||
static class JwtTestingBuilder extends Jwt.Builder<JwtTestingBuilder> { |
||||
|
||||
private static final String DEFAULT_SUBJECT = "user"; |
||||
|
||||
private static final String DEFAULT_TOKEN_VALUE = "test.jwt.value"; |
||||
|
||||
private static final String DEFAULT_HEADER_NAME = "test-header"; |
||||
|
||||
private static final String DEFAULT_HEADER_VALUE = "test-header-value"; |
||||
|
||||
public JwtTestingBuilder() { |
||||
super(); |
||||
} |
||||
|
||||
@Override |
||||
public Jwt build() { |
||||
final Object subjectClaim = claims.get(JwtClaimNames.SUB); |
||||
if (!StringUtils.hasLength(tokenValue)) { |
||||
tokenValue(DEFAULT_TOKEN_VALUE); |
||||
} |
||||
if (!StringUtils.hasLength((String) subjectClaim)) { |
||||
claim(JwtClaimNames.SUB, DEFAULT_SUBJECT); |
||||
} |
||||
if (headers.size() == 0) { |
||||
header(DEFAULT_HEADER_NAME, DEFAULT_HEADER_VALUE); |
||||
} |
||||
return super.build(); |
||||
} |
||||
} |
||||
} |
||||
@ -0,0 +1,83 @@
@@ -0,0 +1,83 @@
|
||||
/* |
||||
* Copyright 2002-2019 the original author or authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
package org.springframework.security.test.support; |
||||
|
||||
import static org.assertj.core.api.Assertions.assertThat; |
||||
|
||||
import org.junit.Test; |
||||
import org.springframework.security.core.authority.SimpleGrantedAuthority; |
||||
import org.springframework.security.oauth2.jwt.Jwt; |
||||
import org.springframework.security.oauth2.jwt.JwtClaimNames; |
||||
import org.springframework.security.oauth2.server.resource.authentication.JwtAuthenticationToken; |
||||
|
||||
/** |
||||
* @author Jérôme Wacongne <ch4mp@c4-soft.com> |
||||
* @since 5.2 |
||||
*/ |
||||
public class JwtAuthenticationTokenTestingBuilderTests { |
||||
|
||||
@Test |
||||
public void untouchedBuilderSetsDefaultValues() { |
||||
final JwtAuthenticationToken actual = new JwtAuthenticationTokenTestingBuilder<>().build(); |
||||
|
||||
assertThat(actual.getName()).isEqualTo("user"); |
||||
assertThat(actual.getAuthorities()).containsExactly(new SimpleGrantedAuthority("SCOPE_USER")); |
||||
assertThat(actual.getPrincipal()).isInstanceOf(Jwt.class); |
||||
assertThat(actual.getCredentials()).isInstanceOf(Jwt.class); |
||||
assertThat(actual.getDetails()).isNull(); |
||||
|
||||
// Token default values are tested in JwtTestingBuilderTests
|
||||
assertThat(actual.getToken()).isEqualTo(new JwtAuthenticationTokenTestingBuilder.JwtTestingBuilder().build()); |
||||
} |
||||
|
||||
@Test |
||||
public void nameOverridesDefaultValue() { |
||||
assertThat(new JwtAuthenticationTokenTestingBuilder<>().name("ch4mpy").build().getName()).isEqualTo("ch4mpy"); |
||||
} |
||||
|
||||
@Test |
||||
public void authoritiesAddsToDefaultValue() { |
||||
assertThat(new JwtAuthenticationTokenTestingBuilder<>().authorities("TEST").build().getAuthorities()) |
||||
.containsExactlyInAnyOrder(new SimpleGrantedAuthority("SCOPE_USER"), new SimpleGrantedAuthority("TEST")); |
||||
} |
||||
|
||||
@Test |
||||
public void scopesOveridesDefaultValue() { |
||||
assertThat(new JwtAuthenticationTokenTestingBuilder<>().scopes("TEST").build().getAuthorities()) |
||||
.containsExactly(new SimpleGrantedAuthority("SCOPE_TEST")); |
||||
} |
||||
|
||||
@Test |
||||
public void nameSetsAuthenticationNameAndTokenSubjectClaim() { |
||||
final JwtAuthenticationToken actual = new JwtAuthenticationTokenTestingBuilder<>().name("ch4mpy").build(); |
||||
|
||||
assertThat(actual.getName()).isEqualTo("ch4mpy"); |
||||
assertThat(actual.getTokenAttributes().get(JwtClaimNames.SUB)).isEqualTo("ch4mpy"); |
||||
} |
||||
|
||||
@Test |
||||
public void buildMergesConvertedClaimsAndAuthorities() { |
||||
final JwtAuthenticationToken actual = new JwtAuthenticationTokenTestingBuilder<>().name("ch4mpy") |
||||
.authorities(new SimpleGrantedAuthority("TEST_AUTHORITY")) |
||||
.scopes("scope:claim") |
||||
.build(); |
||||
|
||||
assertThat(actual.getAuthorities()).containsExactlyInAnyOrder( |
||||
new SimpleGrantedAuthority("TEST_AUTHORITY"), |
||||
new SimpleGrantedAuthority("SCOPE_scope:claim")); |
||||
} |
||||
|
||||
} |
||||
@ -0,0 +1,56 @@
@@ -0,0 +1,56 @@
|
||||
/* |
||||
* Copyright 2002-2019 the original author or authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
package org.springframework.security.test.support; |
||||
|
||||
import static org.assertj.core.api.Assertions.assertThat; |
||||
|
||||
import java.time.Instant; |
||||
|
||||
import org.junit.Test; |
||||
import org.springframework.security.oauth2.jwt.Jwt; |
||||
import org.springframework.security.oauth2.jwt.JwtClaimNames; |
||||
import org.springframework.security.test.support.JwtAuthenticationTokenTestingBuilder.JwtTestingBuilder; |
||||
|
||||
/** |
||||
* |
||||
* |
||||
* @author Jérôme Wacongne <ch4mp@c4-soft.com> |
||||
*/ |
||||
public class JwtTestingBuilderTests { |
||||
|
||||
@Test |
||||
public void testDefaultValuesAreSet() { |
||||
final Jwt actual = new JwtTestingBuilder().build(); |
||||
|
||||
assertThat(actual.getTokenValue()).isEqualTo("test.jwt.value"); |
||||
assertThat(actual.getClaimAsString(JwtClaimNames.SUB)).isEqualTo("user"); |
||||
assertThat(actual.getHeaders()).hasSize(1); |
||||
} |
||||
|
||||
@Test |
||||
public void iatClaimAndExpClaimSetIssuedAtAndExpiresAt() { |
||||
final Jwt actual = new JwtTestingBuilder() |
||||
.claim(JwtClaimNames.IAT, Instant.parse("2019-03-21T13:52:25Z")) |
||||
.claim(JwtClaimNames.EXP, Instant.parse("2019-03-22T13:52:25Z")) |
||||
.build(); |
||||
|
||||
assertThat(actual.getIssuedAt()).isEqualTo(Instant.parse("2019-03-21T13:52:25Z")); |
||||
assertThat(actual.getExpiresAt()).isEqualTo(Instant.parse("2019-03-22T13:52:25Z")); |
||||
assertThat(actual.getClaimAsInstant(JwtClaimNames.IAT)).isEqualTo(Instant.parse("2019-03-21T13:52:25Z")); |
||||
assertThat(actual.getClaimAsInstant(JwtClaimNames.EXP)).isEqualTo(Instant.parse("2019-03-22T13:52:25Z")); |
||||
} |
||||
|
||||
} |
||||
@ -0,0 +1,65 @@
@@ -0,0 +1,65 @@
|
||||
/* |
||||
* Copyright 2002-2019 the original author or authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
package org.springframework.security.test.web.reactive.server; |
||||
|
||||
import static org.springframework.security.test.web.reactive.server.SecurityMockServerConfigurers.mockJwt; |
||||
|
||||
import org.junit.Test; |
||||
|
||||
/** |
||||
* @author Jérôme Wacongne <ch4mp@c4-soft.com> |
||||
* @since 5.2 |
||||
*/ |
||||
public class JwtMutatorTests { |
||||
// @formatter:off
|
||||
@Test |
||||
public void defaultJwtConfigurerConfiguresAuthenticationDefaultNameAndAuthorities() { |
||||
TestController.clientBuilder() |
||||
.apply(mockJwt()).build() |
||||
.get().uri("/greet").exchange() |
||||
.expectStatus().isOk() |
||||
.expectBody().toString().equals("Hello user!"); |
||||
|
||||
TestController.clientBuilder() |
||||
.apply(mockJwt()).build() |
||||
.get().uri("/authorities").exchange() |
||||
.expectStatus().isOk() |
||||
.expectBody().toString().equals("[\"ROLE_USER\"]"); |
||||
} |
||||
|
||||
@Test |
||||
public void nameAndScopesConfigureAuthenticationNameAndAuthorities() { |
||||
TestController.clientBuilder() |
||||
.apply(mockJwt().name("ch4mpy").scopes("message:read")).build() |
||||
.get().uri("/greet").exchange() |
||||
.expectStatus().isOk() |
||||
.expectBody().toString().equals("Hello ch4mpy!"); |
||||
|
||||
TestController.clientBuilder() |
||||
.apply(mockJwt().name("ch4mpy").scopes("message:read")).build() |
||||
.get().uri("/authorities").exchange() |
||||
.expectStatus().isOk() |
||||
.expectBody().toString().equals("[\"SCOPE_message:read\"]"); |
||||
|
||||
TestController.clientBuilder() |
||||
.apply(mockJwt().name("ch4mpy").scopes("message:read")).build() |
||||
.get().uri("/jwt").exchange() |
||||
.expectStatus().isOk() |
||||
.expectBody().toString().equals( |
||||
"Hello,ch4mpy! You are sucessfully authenticated and granted with [message:read] scopes using a JavaWebToken."); |
||||
} |
||||
// @formatter:on
|
||||
} |
||||
@ -0,0 +1,78 @@
@@ -0,0 +1,78 @@
|
||||
/* |
||||
* Copyright 2002-2019 the original author or authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
package org.springframework.security.test.web.reactive.server; |
||||
|
||||
import static org.springframework.security.test.web.reactive.server.SecurityMockServerConfigurers.springSecurity; |
||||
|
||||
import java.security.Principal; |
||||
import java.util.stream.Collectors; |
||||
|
||||
import org.springframework.http.HttpHeaders; |
||||
import org.springframework.http.MediaType; |
||||
import org.springframework.security.core.Authentication; |
||||
import org.springframework.security.core.GrantedAuthority; |
||||
import org.springframework.security.oauth2.jwt.Jwt; |
||||
import org.springframework.security.web.server.context.SecurityContextServerWebExchangeWebFilter; |
||||
import org.springframework.security.web.server.csrf.CsrfWebFilter; |
||||
import org.springframework.test.web.reactive.server.WebTestClient; |
||||
import org.springframework.web.bind.annotation.GetMapping; |
||||
import org.springframework.web.bind.annotation.RestController; |
||||
|
||||
/** |
||||
* @author Jérôme Wacongne <ch4mp@c4-soft.com> |
||||
* @since 5.2 |
||||
*/ |
||||
@RestController |
||||
public class TestController { |
||||
|
||||
@GetMapping("/greet") |
||||
public String greet(final Principal authentication) { |
||||
return String.format("Hello, %s!", authentication.getName()); |
||||
} |
||||
|
||||
@GetMapping("/authorities") |
||||
public String authentication(final Authentication authentication) { |
||||
return authentication.getAuthorities() |
||||
.stream() |
||||
.map(GrantedAuthority::getAuthority) |
||||
.collect(Collectors.toList()) |
||||
.toString(); |
||||
} |
||||
|
||||
@GetMapping("/jwt") |
||||
// TODO: investigate why "@AuthenticationPrincipal Jwt token" does not work here
|
||||
public String jwt(final Authentication authentication) { |
||||
final Jwt token = (Jwt) authentication.getPrincipal(); |
||||
final String scopes = token.getClaimAsString("scope"); |
||||
|
||||
return String.format( |
||||
"Hello, %s! You are sucessfully authenticated and granted with %s scopes using a Jwt.", |
||||
token.getSubject(), |
||||
scopes); |
||||
} |
||||
|
||||
public static WebTestClient.Builder clientBuilder() { |
||||
return WebTestClient.bindToController(new TestController()) |
||||
.webFilter(new CsrfWebFilter(), new SecurityContextServerWebExchangeWebFilter()) |
||||
.apply(springSecurity()) |
||||
.configureClient() |
||||
.defaultHeader(HttpHeaders.ACCEPT, MediaType.APPLICATION_JSON_VALUE); |
||||
} |
||||
|
||||
public static WebTestClient client() { |
||||
return (WebTestClient) clientBuilder().build(); |
||||
} |
||||
} |
||||
@ -0,0 +1,67 @@
@@ -0,0 +1,67 @@
|
||||
/* |
||||
* Copyright 2002-2019 the original author or authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
package org.springframework.security.test.web.servlet.request; |
||||
|
||||
import static org.assertj.core.api.Assertions.assertThat; |
||||
import static org.springframework.security.test.web.servlet.request.SecurityMockMvcRequestPostProcessors.jwt; |
||||
|
||||
import org.junit.Before; |
||||
import org.junit.Test; |
||||
import org.mockito.Mock; |
||||
import org.springframework.mock.web.MockHttpServletRequest; |
||||
import org.springframework.security.core.Authentication; |
||||
import org.springframework.security.core.authority.SimpleGrantedAuthority; |
||||
import org.springframework.security.core.context.SecurityContext; |
||||
import org.springframework.security.oauth2.server.resource.authentication.JwtAuthenticationToken; |
||||
import org.springframework.security.test.web.servlet.request.SecurityMockMvcRequestPostProcessors.JwtRequestPostProcessor; |
||||
import org.springframework.security.test.web.servlet.request.SecurityMockMvcRequestPostProcessors.SecurityContextRequestPostProcessorSupport.TestSecurityContextRepository; |
||||
|
||||
/** |
||||
* @author Jérôme Wacongne <ch4mp@c4-soft.com> |
||||
* @since 5.2 |
||||
*/ |
||||
public class JwtRequestPostProcessorTests { |
||||
@Mock |
||||
MockHttpServletRequest request; |
||||
|
||||
final static String TEST_NAME = "ch4mpy"; |
||||
final static String[] TEST_AUTHORITIES = { "TEST_AUTHORITY" }; |
||||
|
||||
@Before |
||||
public void setup() throws Exception { |
||||
request = new MockHttpServletRequest(); |
||||
} |
||||
|
||||
@Test |
||||
public void nameAndAuthoritiesAndClaimsConfigureSecurityContextAuthentication() { |
||||
final JwtRequestPostProcessor rpp = |
||||
jwt().name(TEST_NAME).authorities(TEST_AUTHORITIES).scopes("test:claim"); |
||||
|
||||
final JwtAuthenticationToken actual = (JwtAuthenticationToken) authentication(rpp.postProcessRequest(request)); |
||||
|
||||
assertThat(actual.getName()).isEqualTo(TEST_NAME); |
||||
assertThat(actual.getAuthorities()).containsExactlyInAnyOrder( |
||||
new SimpleGrantedAuthority("TEST_AUTHORITY"), |
||||
new SimpleGrantedAuthority("SCOPE_test:claim")); |
||||
assertThat(actual.getTokenAttributes().get("scope")).isEqualTo("test:claim"); |
||||
} |
||||
|
||||
static Authentication authentication(final MockHttpServletRequest req) { |
||||
final SecurityContext securityContext = (SecurityContext) req.getAttribute(TestSecurityContextRepository.ATTR_NAME); |
||||
return securityContext == null ? null : securityContext.getAuthentication(); |
||||
} |
||||
|
||||
} |
||||
Loading…
Reference in new issue