cubetiq/spring-authorization-server
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Polish NimbusJwsEncoderTests

Browse Source 
Issue gh-196
This commit is contained in:
Joe Grandja 2021-01-19 05:22:51 -05:00
parent b7996e26d0
commit 17c20e98d4
1 changed files with 15 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
oauth2-authorization-server/src/test/java/org/springframework/security/oauth2/jwt/NimbusJwsEncoderTests.java
View File

@ -17,6 +17,7 @@ package org.springframework.security.oauth2.jwt;
import java.security.interfaces.ECPrivateKey; import java.security.interfaces.ECPrivateKey;
import java.security.interfaces.ECPublicKey; import java.security.interfaces.ECPublicKey;
import java.util.ArrayList;
import java.util.Arrays; import java.util.Arrays;
import java.util.Collections; import java.util.Collections;
import java.util.List; import java.util.List;
@ -57,13 +58,16 @@ import static org.mockito.Mockito.verify;
*/ */
public class NimbusJwsEncoderTests { public class NimbusJwsEncoderTests {
private List<JWK> jwkList;
private JWKSource<SecurityContext> jwkSource; private JWKSource<SecurityContext> jwkSource;
private NimbusJwsEncoder jwsEncoder; private NimbusJwsEncoder jwsEncoder;
@Before @Before
public void setUp() { public void setUp() {
this.jwkSource = mock(JWKSource.class); this.jwkList = new ArrayList<>();
this.jwkSource = (jwkSelector, securityContext) -> jwkSelector.select(new JWKSet(this.jwkList));
this.jwsEncoder = new NimbusJwsEncoder(this.jwkSource); this.jwsEncoder = new NimbusJwsEncoder(this.jwkSource);
} }
@ -96,9 +100,9 @@ public class NimbusJwsEncoderTests {
} }
@Test @Test
public void encodeWhenCustomizerSetThenCalled() throws Exception { public void encodeWhenCustomizerSetThenCalled() {
RSAKey rsaJwk = TestJwks.DEFAULT_RSA_JWK; RSAKey rsaJwk = TestJwks.DEFAULT_RSA_JWK;
given(this.jwkSource.get(any(), any())).willReturn(Collections.singletonList(rsaJwk)); this.jwkList.add(rsaJwk);
BiConsumer<JoseHeader.Builder, JwtClaimsSet.Builder> jwtCustomizer = mock(BiConsumer.class); BiConsumer<JoseHeader.Builder, JwtClaimsSet.Builder> jwtCustomizer = mock(BiConsumer.class);
this.jwsEncoder.setJwtCustomizer(jwtCustomizer); this.jwsEncoder.setJwtCustomizer(jwtCustomizer);
@ -113,6 +117,8 @@ public class NimbusJwsEncoderTests {
@Test @Test
public void encodeWhenJwkSelectFailedThenThrowJwtEncodingException() throws Exception { public void encodeWhenJwkSelectFailedThenThrowJwtEncodingException() throws Exception {
this.jwkSource = mock(JWKSource.class);
this.jwsEncoder = new NimbusJwsEncoder(this.jwkSource);
given(this.jwkSource.get(any(), any())).willThrow(new KeySourceException("key source error")); given(this.jwkSource.get(any(), any())).willThrow(new KeySourceException("key source error"));
JoseHeader joseHeader = TestJoseHeaders.joseHeader().build(); JoseHeader joseHeader = TestJoseHeaders.joseHeader().build();
@ -126,7 +132,8 @@ public class NimbusJwsEncoderTests {
@Test @Test
public void encodeWhenJwkMultipleSelectedThenThrowJwtEncodingException() throws Exception { public void encodeWhenJwkMultipleSelectedThenThrowJwtEncodingException() throws Exception {
RSAKey rsaJwk = TestJwks.DEFAULT_RSA_JWK; RSAKey rsaJwk = TestJwks.DEFAULT_RSA_JWK;
given(this.jwkSource.get(any(), any())).willReturn(Arrays.asList(rsaJwk, rsaJwk)); this.jwkList.add(rsaJwk);
this.jwkList.add(rsaJwk);
JoseHeader joseHeader = TestJoseHeaders.joseHeader().build(); JoseHeader joseHeader = TestJoseHeaders.joseHeader().build();
JwtClaimsSet jwtClaimsSet = TestJwtClaimsSets.jwtClaimsSet().build(); JwtClaimsSet jwtClaimsSet = TestJwtClaimsSets.jwtClaimsSet().build();
@ -154,7 +161,7 @@ public class NimbusJwsEncoderTests {
.build(); .build();
// @formatter:on // @formatter:on
given(this.jwkSource.get(any(), any())).willReturn(Collections.singletonList(rsaJwk)); this.jwkList.add(rsaJwk);
JoseHeader joseHeader = TestJoseHeaders.joseHeader().build(); JoseHeader joseHeader = TestJoseHeaders.joseHeader().build();
JwtClaimsSet jwtClaimsSet = TestJwtClaimsSets.jwtClaimsSet().build(); JwtClaimsSet jwtClaimsSet = TestJwtClaimsSets.jwtClaimsSet().build();
@ -172,6 +179,8 @@ public class NimbusJwsEncoderTests {
.build(); .build();
// @formatter:on // @formatter:on
this.jwkSource = mock(JWKSource.class);
this.jwsEncoder = new NimbusJwsEncoder(this.jwkSource);
given(this.jwkSource.get(any(), any())).willReturn(Collections.singletonList(rsaJwk)); given(this.jwkSource.get(any(), any())).willReturn(Collections.singletonList(rsaJwk));
JoseHeader joseHeader = TestJoseHeaders.joseHeader().build(); JoseHeader joseHeader = TestJoseHeaders.joseHeader().build();
@ -185,7 +194,7 @@ public class NimbusJwsEncoderTests {
@Test @Test
public void encodeWhenSuccessThenDecodes() throws Exception { public void encodeWhenSuccessThenDecodes() throws Exception {
RSAKey rsaJwk = TestJwks.DEFAULT_RSA_JWK; RSAKey rsaJwk = TestJwks.DEFAULT_RSA_JWK;
given(this.jwkSource.get(any(), any())).willReturn(Collections.singletonList(rsaJwk)); this.jwkList.add(rsaJwk);
JoseHeader joseHeader = TestJoseHeaders.joseHeader().build(); JoseHeader joseHeader = TestJoseHeaders.joseHeader().build();
JwtClaimsSet jwtClaimsSet = TestJwtClaimsSets.jwtClaimsSet().build(); JwtClaimsSet jwtClaimsSet = TestJwtClaimsSets.jwtClaimsSet().build();