unit tests of exceptions cases for refreshing access tokens.

pull/419/head
William Kim 2013-07-05 12:32:45 -04:00
parent 3353b92426
commit ed2223cae3
1 changed files with 39 additions and 2 deletions

View File

@ -38,6 +38,7 @@ import org.mockito.Mockito;
import org.mockito.runners.MockitoJUnitRunner; import org.mockito.runners.MockitoJUnitRunner;
import org.springframework.security.authentication.AuthenticationCredentialsNotFoundException; import org.springframework.security.authentication.AuthenticationCredentialsNotFoundException;
import org.springframework.security.oauth2.common.exceptions.InvalidClientException; import org.springframework.security.oauth2.common.exceptions.InvalidClientException;
import org.springframework.security.oauth2.common.exceptions.InvalidTokenException;
import org.springframework.security.oauth2.provider.AuthorizationRequest; import org.springframework.security.oauth2.provider.AuthorizationRequest;
import org.springframework.security.oauth2.provider.OAuth2Authentication; import org.springframework.security.oauth2.provider.OAuth2Authentication;
import org.springframework.security.oauth2.provider.token.TokenEnhancer; import org.springframework.security.oauth2.provider.token.TokenEnhancer;
@ -57,6 +58,11 @@ public class TestDefaultOAuth2ProviderTokenService {
private String clientId = "test_client"; private String clientId = "test_client";
private Set<String> scope = Sets.newHashSet("openid", "profile", "email", "offline_access"); private Set<String> scope = Sets.newHashSet("openid", "profile", "email", "offline_access");
private OAuth2RefreshTokenEntity refreshToken;
private String refreshTokenValue = "refresh_token_value";
private AuthorizationRequest authRequest;
@Mock @Mock
private OAuth2TokenRepository tokenRepository; private OAuth2TokenRepository tokenRepository;
@ -92,6 +98,13 @@ public class TestDefaultOAuth2ProviderTokenService {
// by default in tests, allow refresh tokens // by default in tests, allow refresh tokens
Mockito.when(client.isAllowRefresh()).thenReturn(true); Mockito.when(client.isAllowRefresh()).thenReturn(true);
refreshToken = Mockito.mock(OAuth2RefreshTokenEntity.class);
Mockito.when(tokenRepository.getRefreshTokenByValue(refreshTokenValue)).thenReturn(refreshToken);
Mockito.when(refreshToken.getClient()).thenReturn(client);
Mockito.when(refreshToken.isExpired()).thenReturn(false);
authRequest = Mockito.mock(AuthorizationRequest.class);
} }
/** /**
@ -223,4 +236,28 @@ public class TestDefaultOAuth2ProviderTokenService {
Mockito.verify(authenticationHolderRepository).save(Mockito.any(AuthenticationHolderEntity.class)); Mockito.verify(authenticationHolderRepository).save(Mockito.any(AuthenticationHolderEntity.class));
} }
@Test(expected = InvalidTokenException.class)
public void refreshAccessToken_noRefreshToken() {
Mockito.when(tokenRepository.getRefreshTokenByValue(Mockito.anyString())).thenReturn(null);
service.refreshAccessToken(refreshTokenValue, authRequest);
}
@Test(expected = InvalidClientException.class)
public void refreshAccessToken_notAllowRefresh() {
Mockito.when(client.isAllowRefresh()).thenReturn(false);
service.refreshAccessToken(refreshTokenValue, authRequest);
}
@Test(expected = InvalidTokenException.class)
public void refreshAccessToken_expired() {
Mockito.when(refreshToken.isExpired()).thenReturn(true);
service.refreshAccessToken(refreshTokenValue, authRequest);
}
} }