protected String getTokenKey(final OAuth2Token token) { return this.getTokenKey(token.getGadgetUri(), token.getServiceName(), token.getUser(), token.getScope(), token.getType()); } }
@Test public void testStoreTokens_1() throws Exception { this.cache.clearTokens(); final Collection<OAuth2Token> tokens = new HashSet<OAuth2Token>(2); final OAuth2Token accessToken = MockUtils.getAccessToken(); final OAuth2Token refreshToken = MockUtils.getRefreshToken(); tokens.add(accessToken); tokens.add(refreshToken); this.cache.storeTokens(tokens); Assert.assertNotNull(this.cache.getToken(accessToken.getGadgetUri(), accessToken.getServiceName(), accessToken.getUser(), accessToken.getScope(), accessToken.getType())); Assert.assertNotNull(this.cache.getToken(refreshToken.getGadgetUri(), refreshToken.getServiceName(), refreshToken.getUser(), refreshToken.getScope(), refreshToken.getType())); } }
@Test public void testStoreTokens_1() throws Exception { this.cache.clearTokens(); final Collection<OAuth2Token> tokens = new HashSet<OAuth2Token>(2); final OAuth2Token accessToken = MockUtils.getAccessToken(); final OAuth2Token refreshToken = MockUtils.getRefreshToken(); tokens.add(accessToken); tokens.add(refreshToken); this.cache.storeTokens(tokens); Assert.assertNotNull(this.cache.getToken(accessToken.getGadgetUri(), accessToken.getServiceName(), accessToken.getUser(), accessToken.getScope(), accessToken.getType())); Assert.assertNotNull(this.cache.getToken(refreshToken.getGadgetUri(), refreshToken.getServiceName(), refreshToken.getUser(), refreshToken.getScope(), refreshToken.getType())); } }
@Test public void testSetTokenForSharedClient() throws Exception { final OAuth2Cache cache = new InMemoryCache(); final OAuth2Persister persister = MockUtils.getDummyPersister(); final OAuth2Encrypter encrypter = MockUtils.getDummyEncrypter(); final BlobCrypter stateCrypter = MockUtils.getDummyStateCrypter(); OAuth2Token token = MockUtils.getAccessToken(); OAuth2Client client = MockUtils.getClient_Code_Confidential(); client.setSharedToken( true ); BasicOAuth2Store mockStore = createMockBuilder( BasicOAuth2Store.class ) .withConstructor( OAuth2Cache.class, OAuth2Persister.class, OAuth2Encrypter.class, String.class, Authority.class, String.class, BlobCrypter.class ) .withArgs( cache, persister, encrypter, MockUtils.REDIRECT_URI, (Authority)null, (String)null, stateCrypter ) .addMockedMethod( "getClient" ) .addMockedMethod( "getToken" ) .createMock(); expect( mockStore.getClient( eq(MockUtils.GADGET_URI1), eq(MockUtils.SERVICE_NAME) ) ).andReturn( client ); expect( mockStore.getToken( eq(token.getGadgetUri()), eq(token.getServiceName()), eq(token.getUser()), eq(token.getScope()), eq(token.getType() ) )).andReturn( token ); replay( mockStore ); mockStore.setToken( token ); verify( mockStore ); } }
@Test public void testSetTokenForSharedClient() throws Exception { final OAuth2Cache cache = new InMemoryCache(); final OAuth2Persister persister = MockUtils.getDummyPersister(); final OAuth2Encrypter encrypter = MockUtils.getDummyEncrypter(); final BlobCrypter stateCrypter = MockUtils.getDummyStateCrypter(); OAuth2Token token = MockUtils.getAccessToken(); OAuth2Client client = MockUtils.getClient_Code_Confidential(); client.setSharedToken( true ); BasicOAuth2Store mockStore = createMockBuilder( BasicOAuth2Store.class ) .withConstructor( OAuth2Cache.class, OAuth2Persister.class, OAuth2Encrypter.class, String.class, Authority.class, String.class, BlobCrypter.class ) .withArgs( cache, persister, encrypter, MockUtils.REDIRECT_URI, (Authority)null, (String)null, stateCrypter ) .addMockedMethod( "getClient" ) .addMockedMethod( "getToken" ) .createMock(); expect( mockStore.getClient( eq(MockUtils.GADGET_URI1), eq(MockUtils.SERVICE_NAME) ) ).andReturn( client ); expect( mockStore.getToken( eq(token.getGadgetUri()), eq(token.getServiceName()), eq(token.getUser()), eq(token.getScope()), eq(token.getType() ) )).andReturn( token ); replay( mockStore ); mockStore.setToken( token ); verify( mockStore ); } }
this.persister.removeToken(processedGadgetUri, token.getServiceName(), token.getUser(), token.getScope(), token.getType()); } finally { token.setGadgetUri(origGadgetApi);
@Test public void testStoreToken_1() throws Exception { OAuth2Token token = new OAuth2TokenPersistence(MockUtils.getDummyEncrypter()); token.setGadgetUri("xxx"); token.setServiceName("yyy"); token.setExpiresAt(2); token.setIssuedAt(1); token.setMacAlgorithm(OAuth2Message.HMAC_SHA_1); token.setMacSecret("shh, it's a secret".getBytes("UTF-8")); token.setScope("mac_scope"); token.setSecret("i'll never tell".getBytes("UTF-8")); token.setTokenType(OAuth2Message.MAC_TOKEN_TYPE); token.setType(OAuth2Token.Type.ACCESS); token.setUser("zzz"); this.cache.storeToken(token); token = this.cache.getToken(token.getGadgetUri(), token.getServiceName(), token.getUser(), token.getScope(), token.getType()); Assert.assertNotNull(token); Assert.assertEquals("xxx", token.getGadgetUri()); Assert.assertEquals("yyy", token.getServiceName()); Assert.assertEquals(2, token.getExpiresAt()); Assert.assertEquals(1, token.getIssuedAt()); Assert.assertEquals(OAuth2Message.HMAC_SHA_1, token.getMacAlgorithm()); Assert.assertEquals("shh, it's a secret", new String(token.getMacSecret(), "UTF-8")); Assert.assertEquals("mac_scope", token.getScope()); Assert.assertEquals("i'll never tell", new String(token.getSecret(), "UTF-8")); Assert.assertEquals(OAuth2Message.MAC_TOKEN_TYPE, token.getTokenType()); Assert.assertEquals(OAuth2Token.Type.ACCESS, token.getType()); Assert.assertEquals("zzz", token.getUser()); }
@Test public void testStoreToken_1() throws Exception { OAuth2Token token = new OAuth2TokenPersistence(MockUtils.getDummyEncrypter()); token.setGadgetUri("xxx"); token.setServiceName("yyy"); token.setExpiresAt(2); token.setIssuedAt(1); token.setMacAlgorithm(OAuth2Message.HMAC_SHA_1); token.setMacSecret("shh, it's a secret".getBytes("UTF-8")); token.setScope("mac_scope"); token.setSecret("i'll never tell".getBytes("UTF-8")); token.setTokenType(OAuth2Message.MAC_TOKEN_TYPE); token.setType(OAuth2Token.Type.ACCESS); token.setUser("zzz"); this.cache.storeToken(token); token = this.cache.getToken(token.getGadgetUri(), token.getServiceName(), token.getUser(), token.getScope(), token.getType()); Assert.assertNotNull(token); Assert.assertEquals("xxx", token.getGadgetUri()); Assert.assertEquals("yyy", token.getServiceName()); Assert.assertEquals(2, token.getExpiresAt()); Assert.assertEquals(1, token.getIssuedAt()); Assert.assertEquals(OAuth2Message.HMAC_SHA_1, token.getMacAlgorithm()); Assert.assertEquals("shh, it's a secret", new String(token.getMacSecret(), "UTF-8")); Assert.assertEquals("mac_scope", token.getScope()); Assert.assertEquals("i'll never tell", new String(token.getSecret(), "UTF-8")); Assert.assertEquals(OAuth2Message.MAC_TOKEN_TYPE, token.getTokenType()); Assert.assertEquals(OAuth2Token.Type.ACCESS, token.getType()); Assert.assertEquals("zzz", token.getUser()); }
try { final OAuth2Token existingToken = this.getToken(gadgetUri, token.getServiceName(), token.getUser(), token.getScope(), token.getType()); try { if (existingToken == null) {