Use 'userName' instead of 'sub' in naming
							parent
							
								
									3f277047e3
								
							
						
					
					
						commit
						51b580aa18
					
				|  | @ -72,7 +72,7 @@ import com.nimbusds.jwt.JWT; | |||
| 	@NamedQuery(name = OAuth2AccessTokenEntity.QUERY_BY_TOKEN_VALUE, query = "select a from OAuth2AccessTokenEntity a where a.jwt = :" + OAuth2AccessTokenEntity.PARAM_TOKEN_VALUE), | ||||
| 	@NamedQuery(name = OAuth2AccessTokenEntity.QUERY_BY_APPROVED_SITE, query = "select a from OAuth2AccessTokenEntity a where a.approvedSite = :" + OAuth2AccessTokenEntity.PARAM_APPROVED_SITE), | ||||
| 	@NamedQuery(name = OAuth2AccessTokenEntity.QUERY_BY_RESOURCE_SET, query = "select a from OAuth2AccessTokenEntity a join a.permissions p where p.resourceSet.id = :" + OAuth2AccessTokenEntity.PARAM_RESOURCE_SET_ID), | ||||
| 	@NamedQuery(name = OAuth2AccessTokenEntity.QUERY_BY_SUB, query = "select r from OAuth2AccessTokenEntity r where r.authenticationHolder.userAuth.name = :" + OAuth2AccessTokenEntity.PARAM_SUB) | ||||
| 	@NamedQuery(name = OAuth2AccessTokenEntity.QUERY_BY_NAME, query = "select r from OAuth2AccessTokenEntity r where r.authenticationHolder.userAuth.name = :" + OAuth2AccessTokenEntity.PARAM_NAME) | ||||
| }) | ||||
| @org.codehaus.jackson.map.annotate.JsonSerialize(using = OAuth2AccessTokenJackson1Serializer.class) | ||||
| @org.codehaus.jackson.map.annotate.JsonDeserialize(using = OAuth2AccessTokenJackson1Deserializer.class) | ||||
|  | @ -87,7 +87,7 @@ public class OAuth2AccessTokenEntity implements OAuth2AccessToken { | |||
| 	public static final String QUERY_EXPIRED_BY_DATE = "OAuth2AccessTokenEntity.getAllExpiredByDate"; | ||||
| 	public static final String QUERY_ALL = "OAuth2AccessTokenEntity.getAll"; | ||||
| 	public static final String QUERY_BY_RESOURCE_SET = "OAuth2AccessTokenEntity.getByResourceSet"; | ||||
| 	public static final String QUERY_BY_SUB = "OAuth2AccessTokenEntity.getBySub"; | ||||
| 	public static final String QUERY_BY_NAME = "OAuth2AccessTokenEntity.getByName"; | ||||
| 
 | ||||
| 	public static final String PARAM_TOKEN_VALUE = "tokenValue"; | ||||
| 	public static final String PARAM_CLIENT = "client"; | ||||
|  | @ -95,7 +95,7 @@ public class OAuth2AccessTokenEntity implements OAuth2AccessToken { | |||
| 	public static final String PARAM_DATE = "date"; | ||||
| 	public static final String PARAM_RESOURCE_SET_ID = "rsid"; | ||||
| 	public static final String PARAM_APPROVED_SITE = "approvedSite"; | ||||
| 	public static final String PARAM_SUB = "sub"; | ||||
| 	public static final String PARAM_NAME = "name"; | ||||
| 
 | ||||
| 	public static final String ID_TOKEN_FIELD_NAME = "id_token"; | ||||
| 
 | ||||
|  |  | |||
|  | @ -54,7 +54,7 @@ import com.nimbusds.jwt.JWT; | |||
| 	@NamedQuery(name = OAuth2RefreshTokenEntity.QUERY_EXPIRED_BY_DATE, query = "select r from OAuth2RefreshTokenEntity r where r.expiration <= :" + OAuth2RefreshTokenEntity.PARAM_DATE), | ||||
| 	@NamedQuery(name = OAuth2RefreshTokenEntity.QUERY_BY_CLIENT, query = "select r from OAuth2RefreshTokenEntity r where r.client = :" + OAuth2RefreshTokenEntity.PARAM_CLIENT), | ||||
| 	@NamedQuery(name = OAuth2RefreshTokenEntity.QUERY_BY_TOKEN_VALUE, query = "select r from OAuth2RefreshTokenEntity r where r.jwt = :" + OAuth2RefreshTokenEntity.PARAM_TOKEN_VALUE), | ||||
| 	@NamedQuery(name = OAuth2RefreshTokenEntity.QUERY_BY_SUB, query = "select r from OAuth2RefreshTokenEntity r where r.authenticationHolder.userAuth.name = :" + OAuth2RefreshTokenEntity.PARAM_SUB) | ||||
| 	@NamedQuery(name = OAuth2RefreshTokenEntity.QUERY_BY_NAME, query = "select r from OAuth2RefreshTokenEntity r where r.authenticationHolder.userAuth.name = :" + OAuth2RefreshTokenEntity.PARAM_NAME) | ||||
| }) | ||||
| public class OAuth2RefreshTokenEntity implements OAuth2RefreshToken { | ||||
| 
 | ||||
|  | @ -62,12 +62,12 @@ public class OAuth2RefreshTokenEntity implements OAuth2RefreshToken { | |||
| 	public static final String QUERY_BY_CLIENT = "OAuth2RefreshTokenEntity.getByClient"; | ||||
| 	public static final String QUERY_EXPIRED_BY_DATE = "OAuth2RefreshTokenEntity.getAllExpiredByDate"; | ||||
| 	public static final String QUERY_ALL = "OAuth2RefreshTokenEntity.getAll"; | ||||
| 	public static final String QUERY_BY_SUB = "OAuth2RefreshTokenEntity.getBySub"; | ||||
| 	public static final String QUERY_BY_NAME = "OAuth2RefreshTokenEntity.getByName"; | ||||
| 
 | ||||
| 	public static final String PARAM_TOKEN_VALUE = "tokenValue"; | ||||
| 	public static final String PARAM_CLIENT = "client"; | ||||
| 	public static final String PARAM_DATE = "date"; | ||||
| 	public static final String PARAM_SUB = "sub"; | ||||
| 	public static final String PARAM_NAME = "name"; | ||||
| 
 | ||||
| 	private Long id; | ||||
| 
 | ||||
|  |  | |||
|  | @ -53,9 +53,9 @@ public interface OAuth2TokenRepository { | |||
| 
 | ||||
| 	public List<OAuth2RefreshTokenEntity> getRefreshTokensForClient(ClientDetailsEntity client); | ||||
| 	 | ||||
| 	public Set<OAuth2AccessTokenEntity> getAccessTokensBySub(String sub); | ||||
| 	public Set<OAuth2AccessTokenEntity> getAccessTokensByUserName(String name); | ||||
| 	 | ||||
| 	public Set<OAuth2RefreshTokenEntity> getRefreshTokensBySub(String sub); | ||||
| 	public Set<OAuth2RefreshTokenEntity> getRefreshTokensByUserName(String name); | ||||
| 
 | ||||
| 	public Set<OAuth2AccessTokenEntity> getAllAccessTokens(); | ||||
| 
 | ||||
|  |  | |||
|  | @ -186,17 +186,17 @@ public class JpaOAuth2TokenRepository implements OAuth2TokenRepository { | |||
| 	} | ||||
| 	 | ||||
| 	@Override | ||||
| 	public Set<OAuth2AccessTokenEntity> getAccessTokensBySub(String sub) { | ||||
| 		TypedQuery<OAuth2AccessTokenEntity> query = manager.createNamedQuery(OAuth2AccessTokenEntity.QUERY_BY_SUB, OAuth2AccessTokenEntity.class); | ||||
| 	    query.setParameter(OAuth2AccessTokenEntity.PARAM_SUB, sub); | ||||
| 	public Set<OAuth2AccessTokenEntity> getAccessTokensByUserName(String name) { | ||||
| 		TypedQuery<OAuth2AccessTokenEntity> query = manager.createNamedQuery(OAuth2AccessTokenEntity.QUERY_BY_NAME, OAuth2AccessTokenEntity.class); | ||||
| 	    query.setParameter(OAuth2AccessTokenEntity.PARAM_NAME, name); | ||||
| 	    List<OAuth2AccessTokenEntity> results = query.getResultList(); | ||||
| 	    return results != null ? new HashSet<>(query.getResultList()) : new HashSet<>(); | ||||
| 	} | ||||
| 	 | ||||
| 	@Override | ||||
| 	public Set<OAuth2RefreshTokenEntity> getRefreshTokensBySub(String sub) { | ||||
| 		TypedQuery<OAuth2RefreshTokenEntity> query = manager.createNamedQuery(OAuth2RefreshTokenEntity.QUERY_BY_SUB, OAuth2RefreshTokenEntity.class); | ||||
| 	    query.setParameter(OAuth2RefreshTokenEntity.PARAM_SUB, sub); | ||||
| 	public Set<OAuth2RefreshTokenEntity> getRefreshTokensByUserName(String name) { | ||||
| 		TypedQuery<OAuth2RefreshTokenEntity> query = manager.createNamedQuery(OAuth2RefreshTokenEntity.QUERY_BY_NAME, OAuth2RefreshTokenEntity.class); | ||||
| 	    query.setParameter(OAuth2RefreshTokenEntity.PARAM_NAME, name); | ||||
| 	    List<OAuth2RefreshTokenEntity> results = query.getResultList(); | ||||
| 	    return results != null ? new HashSet<>(query.getResultList()) : new HashSet<>(); | ||||
| 	} | ||||
|  |  | |||
|  | @ -102,13 +102,13 @@ public class DefaultOAuth2ProviderTokenService implements OAuth2TokenEntityServi | |||
| 	private ApprovedSiteService approvedSiteService; | ||||
| 
 | ||||
| 	@Override | ||||
| 	public Set<OAuth2AccessTokenEntity> getAllAccessTokensForUser(String sub) { | ||||
| 		return tokenRepository.getAccessTokensBySub(sub); | ||||
| 	public Set<OAuth2AccessTokenEntity> getAllAccessTokensForUser(String userName) { | ||||
| 		return tokenRepository.getAccessTokensByUserName(userName); | ||||
| 	} | ||||
| 
 | ||||
| 	@Override | ||||
| 	public Set<OAuth2RefreshTokenEntity> getAllRefreshTokensForUser(String sub) { | ||||
| 		return tokenRepository.getRefreshTokensBySub(sub); | ||||
| 	public Set<OAuth2RefreshTokenEntity> getAllRefreshTokensForUser(String userName) { | ||||
| 		return tokenRepository.getRefreshTokensByUserName(userName); | ||||
| 	} | ||||
| 
 | ||||
| 	@Override | ||||
|  |  | |||
|  | @ -21,7 +21,6 @@ import java.util.Date; | |||
| import java.util.HashSet; | ||||
| import java.util.Set; | ||||
| 
 | ||||
| import org.junit.Assert; | ||||
| import org.junit.Before; | ||||
| import org.junit.Test; | ||||
| import org.junit.runner.RunWith; | ||||
|  | @ -63,7 +62,6 @@ import static org.hamcrest.CoreMatchers.nullValue; | |||
| import static org.mockito.Mockito.never; | ||||
| import static org.mockito.Mockito.when; | ||||
| import static org.junit.Assert.assertEquals; | ||||
| import static org.junit.Assert.assertFalse; | ||||
| import static org.junit.Assert.assertThat; | ||||
| import static org.junit.Assert.assertTrue; | ||||
| import static org.junit.Assert.fail; | ||||
|  | @ -88,7 +86,7 @@ public class TestDefaultOAuth2ProviderTokenService { | |||
| 	private OAuth2RefreshTokenEntity refreshToken; | ||||
| 	private OAuth2AccessTokenEntity accessToken; | ||||
| 	private String refreshTokenValue = "refresh_token_value"; | ||||
| 	private String userSub = "6a50ac11786d402a9591d3e592ac770f"; | ||||
| 	private String userName = "6a50ac11786d402a9591d3e592ac770f"; | ||||
| 	private TokenRequest tokenRequest; | ||||
| 
 | ||||
| 	// for use when refreshing access tokens
 | ||||
|  | @ -552,18 +550,18 @@ public class TestDefaultOAuth2ProviderTokenService { | |||
| 	 | ||||
| 	@Test | ||||
| 	public void getAllAccessTokensForUser(){ | ||||
| 		Mockito.when(tokenRepository.getAccessTokensBySub(userSub)).thenReturn(newHashSet(accessToken)); | ||||
| 		Mockito.when(tokenRepository.getAccessTokensByUserName(userName)).thenReturn(newHashSet(accessToken)); | ||||
| 		 | ||||
| 		Set<OAuth2AccessTokenEntity> tokens = service.getAllAccessTokensForUser(userSub); | ||||
| 		Set<OAuth2AccessTokenEntity> tokens = service.getAllAccessTokensForUser(userName); | ||||
| 		assertEquals(1, tokens.size()); | ||||
| 		assertTrue(tokens.contains(accessToken)); | ||||
| 	} | ||||
| 	 | ||||
| 	@Test | ||||
| 	public void getAllRefreshTokensForUser(){ | ||||
| 		Mockito.when(tokenRepository.getRefreshTokensBySub(userSub)).thenReturn(newHashSet(refreshToken)); | ||||
| 		Mockito.when(tokenRepository.getRefreshTokensByUserName(userName)).thenReturn(newHashSet(refreshToken)); | ||||
| 		 | ||||
| 		Set<OAuth2RefreshTokenEntity> tokens = service.getAllRefreshTokensForUser(userSub); | ||||
| 		Set<OAuth2RefreshTokenEntity> tokens = service.getAllRefreshTokensForUser(userName); | ||||
| 		assertEquals(1, tokens.size()); | ||||
| 		assertTrue(tokens.contains(refreshToken)); | ||||
| 	} | ||||
|  |  | |||
		Loading…
	
		Reference in New Issue
	
	 Sauli Ketola
						Sauli Ketola