13 changed files with 948 additions and 14 deletions
@ -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 @@ |
|||||||
|
/* |
||||||
|
* 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 @@ |
|||||||
|
/* |
||||||
|
* 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 @@ |
|||||||
|
/* |
||||||
|
* 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 @@ |
|||||||
|
/* |
||||||
|
* 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 @@ |
|||||||
|
/* |
||||||
|
* 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 @@ |
|||||||
|
/* |
||||||
|
* 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 @@ |
|||||||
|
/* |
||||||
|
* 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