Skip to content

Commit e24ed83

Browse files
committed
Minor updates
1 parent 97c5133 commit e24ed83

File tree

1 file changed

+12
-11
lines changed

1 file changed

+12
-11
lines changed

src/test/java/mertz/security/oauth2/provider/token/store/cassandra/CassandraTokenStoreTests.java

Lines changed: 12 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -62,32 +62,32 @@ public static class SpringConfig {
6262

6363
@Test
6464
public void testExpiringRefreshToken() throws InterruptedException {
65-
String refreshToken = UUID.randomUUID().toString();
66-
DefaultOAuth2RefreshToken expectedExpiringRefreshToken = new DefaultExpiringOAuth2RefreshToken(refreshToken, new Date(System.currentTimeMillis() + 1500));
65+
String refreshToken = "refreshToken-" + UUID.randomUUID();
66+
DefaultOAuth2RefreshToken expectedExpiringRefreshToken = new DefaultExpiringOAuth2RefreshToken(refreshToken, new Date(System.currentTimeMillis() + 1000));
6767
OAuth2Authentication expectedAuthentication = new OAuth2Authentication(RequestTokenFactory.createOAuth2Request("id", false), new TestAuthentication("test2", false));
6868
getTokenStore().storeRefreshToken(expectedExpiringRefreshToken, expectedAuthentication);
6969
OAuth2RefreshToken actualExpiringRefreshToken = getTokenStore().readRefreshToken(refreshToken);
7070
assertEquals(expectedExpiringRefreshToken, actualExpiringRefreshToken);
7171
assertEquals(expectedAuthentication, getTokenStore().readAuthenticationForRefreshToken(expectedExpiringRefreshToken));
7272
// let the token expire
73-
Thread.sleep(1500);
73+
Thread.sleep(2000);
7474
// now it should be gone
7575
assertNull(getTokenStore().readRefreshToken(refreshToken));
7676
assertNull(getTokenStore().readAuthenticationForRefreshToken(expectedExpiringRefreshToken));
7777
}
7878

7979
@Test
8080
public void testExpiringAccessToken() throws InterruptedException {
81-
String accessToken = UUID.randomUUID().toString();
81+
String accessToken = "accessToken-" + UUID.randomUUID();
8282
OAuth2Authentication expectedAuthentication = new OAuth2Authentication(RequestTokenFactory.createOAuth2Request("id", false), new TestAuthentication("test2", false));
8383
DefaultOAuth2AccessToken expectedOAuth2AccessToken = new DefaultOAuth2AccessToken(accessToken);
84-
expectedOAuth2AccessToken.setExpiration(new Date(System.currentTimeMillis() + 1500));
84+
expectedOAuth2AccessToken.setExpiration(new Date(System.currentTimeMillis() + 1000));
8585
getTokenStore().storeAccessToken(expectedOAuth2AccessToken, expectedAuthentication);
8686
OAuth2AccessToken actualOAuth2AccessToken = getTokenStore().readAccessToken(accessToken);
8787
assertEquals(expectedOAuth2AccessToken, actualOAuth2AccessToken);
8888
assertEquals(expectedAuthentication, getTokenStore().readAuthentication(expectedOAuth2AccessToken));
8989
// let the token expire
90-
Thread.sleep(1500);
90+
Thread.sleep(2000);
9191
// now it should be gone
9292
assertNull(getTokenStore().readAccessToken(accessToken));
9393
assertNull(getTokenStore().readAuthentication(expectedOAuth2AccessToken));
@@ -97,7 +97,8 @@ public void testExpiringAccessToken() throws InterruptedException {
9797
public void storeAccessTokenWithoutRefreshTokenRemoveAccessTokenVerifyTokenRemoved() {
9898
OAuth2Request request = RequestTokenFactory.createOAuth2Request("clientId", false);
9999
TestingAuthenticationToken authentication = new TestingAuthenticationToken("user", "password");
100-
OAuth2AccessToken oauth2AccessToken = new DefaultOAuth2AccessToken("access-token-" + UUID.randomUUID());
100+
String accessToken = "accessToken-" + UUID.randomUUID();
101+
OAuth2AccessToken oauth2AccessToken = new DefaultOAuth2AccessToken(accessToken);
101102
OAuth2Authentication oauth2Authentication = new OAuth2Authentication(request, authentication);
102103
getTokenStore().storeAccessToken(oauth2AccessToken, oauth2Authentication);
103104
getTokenStore().removeAccessToken(oauth2AccessToken);
@@ -110,13 +111,13 @@ public void storeExpiringAccessTokenWithRefreshToken_RemoveExpiredAccessTokenUsi
110111
String accessToken = "accessToken-" + UUID.randomUUID();
111112
OAuth2Authentication expectedAuthentication = new OAuth2Authentication(RequestTokenFactory.createOAuth2Request("id", false), new TestAuthentication("test2", false));
112113
DefaultOAuth2AccessToken expectedOAuth2AccessToken = new DefaultOAuth2AccessToken(accessToken);
113-
expectedOAuth2AccessToken.setExpiration(new Date(System.currentTimeMillis() + 1500));
114-
String refreshToken = "testToken-" + UUID.randomUUID();
114+
expectedOAuth2AccessToken.setExpiration(new Date(System.currentTimeMillis() + 1000));
115+
String refreshToken = "refreshToken-" + UUID.randomUUID();
115116
DefaultOAuth2RefreshToken expectedRefreshToken = new DefaultOAuth2RefreshToken(refreshToken);
116117
expectedOAuth2AccessToken.setRefreshToken(expectedRefreshToken);
117118
getTokenStore().storeAccessToken(expectedOAuth2AccessToken, expectedAuthentication);
118119
// let the access token expire
119-
Thread.sleep(1500);
120+
Thread.sleep(2000);
120121
// now it should be gone
121122
assertNull(getTokenStore().readAccessToken(accessToken));
122123
// use refresh token to remove already expired access token, expect no issues since access token has already been removed.
@@ -128,7 +129,7 @@ public void storeAccessTokenWithRefreshToken_RemoveAccessTokenUsingRefreshToken(
128129
String accessToken = "accessToken-" + UUID.randomUUID();
129130
OAuth2Authentication expectedAuthentication = new OAuth2Authentication(RequestTokenFactory.createOAuth2Request("id", false), new TestAuthentication("test2", false));
130131
DefaultOAuth2AccessToken expectedOAuth2AccessToken = new DefaultOAuth2AccessToken(accessToken);
131-
String refreshToken = "testToken-" + UUID.randomUUID();
132+
String refreshToken = "refreshToken-" + UUID.randomUUID();
132133
DefaultOAuth2RefreshToken expectedRefreshToken = new DefaultOAuth2RefreshToken(refreshToken);
133134
expectedOAuth2AccessToken.setRefreshToken(expectedRefreshToken);
134135
getTokenStore().storeAccessToken(expectedOAuth2AccessToken, expectedAuthentication);

0 commit comments

Comments
 (0)