113 lines
5.1 KiB
Java
113 lines
5.1 KiB
Java
/*
|
|
* Copyright 2020 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.server.authorization.authentication;
|
|
|
|
import org.junit.Test;
|
|
import org.springframework.security.core.Authentication;
|
|
import org.springframework.security.oauth2.server.authorization.client.TestRegisteredClients;
|
|
|
|
import java.util.Collections;
|
|
import java.util.Map;
|
|
|
|
import static org.assertj.core.api.Assertions.assertThat;
|
|
import static org.assertj.core.api.Assertions.assertThatThrownBy;
|
|
|
|
/**
|
|
* Tests for {@link OAuth2AuthorizationCodeAuthenticationToken}.
|
|
*
|
|
* @author Joe Grandja
|
|
*/
|
|
public class OAuth2AuthorizationCodeAuthenticationTokenTests {
|
|
private String code = "code";
|
|
private String clientPrincipalClientId = "clientPrincipal.clientId";
|
|
private OAuth2ClientAuthenticationToken clientPrincipal =
|
|
new OAuth2ClientAuthenticationToken(TestRegisteredClients.registeredClient().clientId(clientPrincipalClientId).build());
|
|
private String clientId = "clientId";
|
|
private String redirectUri = "redirectUri";
|
|
private Map<String, Object> additonalParams = Collections.singletonMap("some_key", "some_value");
|
|
|
|
@Test
|
|
public void constructorWhenCodeNullThenThrowIllegalArgumentException() {
|
|
assertThatThrownBy(() -> new OAuth2AuthorizationCodeAuthenticationToken(null, this.clientPrincipal, this.redirectUri, null))
|
|
.isInstanceOf(IllegalArgumentException.class)
|
|
.hasMessage("code cannot be empty");
|
|
}
|
|
|
|
@Test
|
|
public void constructorWhenClientPrincipalNullThenThrowIllegalArgumentException() {
|
|
assertThatThrownBy(() -> new OAuth2AuthorizationCodeAuthenticationToken(this.code, (Authentication) null, this.redirectUri, null))
|
|
.isInstanceOf(IllegalArgumentException.class)
|
|
.hasMessage("clientPrincipal cannot be null");
|
|
}
|
|
|
|
@Test
|
|
public void constructorWhenClientIdNullThenThrowIllegalArgumentException() {
|
|
assertThatThrownBy(() -> new OAuth2AuthorizationCodeAuthenticationToken(this.code, (String) null, this.redirectUri, null))
|
|
.isInstanceOf(IllegalArgumentException.class)
|
|
.hasMessage("clientId cannot be empty");
|
|
}
|
|
|
|
@Test
|
|
public void constructorWhenClientPrincipalProvidedThenCreated() {
|
|
OAuth2AuthorizationCodeAuthenticationToken authentication = new OAuth2AuthorizationCodeAuthenticationToken(
|
|
this.code, this.clientPrincipal, this.redirectUri, this.additonalParams);
|
|
assertThat(authentication.getPrincipal()).isEqualTo(this.clientPrincipal);
|
|
assertThat(authentication.getCredentials().toString()).isEmpty();
|
|
assertThat(authentication.getCode()).isEqualTo(this.code);
|
|
assertThat(authentication.getRedirectUri()).isEqualTo(this.redirectUri);
|
|
assertThat(authentication.getAdditionalParameters()).isEqualTo(this.additonalParams);
|
|
}
|
|
|
|
@Test
|
|
public void constructorWhenClientIdProvidedThenCreated() {
|
|
OAuth2AuthorizationCodeAuthenticationToken authentication = new OAuth2AuthorizationCodeAuthenticationToken(
|
|
this.code, this.clientId, this.redirectUri, this.additonalParams);
|
|
assertThat(authentication.getPrincipal()).isEqualTo(this.clientId);
|
|
assertThat(authentication.getCredentials().toString()).isEmpty();
|
|
assertThat(authentication.getCode()).isEqualTo(this.code);
|
|
assertThat(authentication.getRedirectUri()).isEqualTo(this.redirectUri);
|
|
assertThat(authentication.getAdditionalParameters()).isEqualTo(this.additonalParams);
|
|
}
|
|
|
|
@Test
|
|
public void getAdditionalParamsIsImmutableMap() {
|
|
OAuth2AuthorizationCodeAuthenticationToken authentication = new OAuth2AuthorizationCodeAuthenticationToken(
|
|
this.code, this.clientId, this.redirectUri, this.additonalParams);
|
|
assertThatThrownBy(() -> authentication.getAdditionalParameters().put("another_key", 1))
|
|
.isInstanceOf(UnsupportedOperationException.class);
|
|
assertThatThrownBy(() -> authentication.getAdditionalParameters().remove("some_key"))
|
|
.isInstanceOf(UnsupportedOperationException.class);
|
|
assertThatThrownBy(() -> authentication.getAdditionalParameters().clear())
|
|
.isInstanceOf(UnsupportedOperationException.class);
|
|
}
|
|
|
|
@Test
|
|
public void getClientIdFromClientId() {
|
|
OAuth2AuthorizationCodeAuthenticationToken authentication = new OAuth2AuthorizationCodeAuthenticationToken(
|
|
this.code, this.clientId, this.redirectUri, this.additonalParams);
|
|
|
|
assertThat(authentication.getClientId()).isEqualTo(this.clientId);
|
|
}
|
|
|
|
@Test
|
|
public void getClientIdFromOAuth2ClientAuthenticationTokenPrincipal() {
|
|
OAuth2AuthorizationCodeAuthenticationToken authentication = new OAuth2AuthorizationCodeAuthenticationToken(
|
|
this.code, this.clientPrincipal, this.redirectUri, this.additonalParams);
|
|
|
|
assertThat(authentication.getClientId()).isEqualTo(this.clientPrincipalClientId);
|
|
}
|
|
}
|