use of com.google.auth.TestClock in project google-auth-library-java by google.
the class OAuth2CredentialsTest method addChangeListener_notifiesOnRefresh.
@Test
public void addChangeListener_notifiesOnRefresh() throws IOException {
final String accessToken1 = "1/MkSJoj1xsli0AccessToken_NKPY2";
final String accessToken2 = "2/MkSJoj1xsli0AccessToken_NKPY2";
MockTokenServerTransportFactory transportFactory = new MockTokenServerTransportFactory();
transportFactory.transport.addClient(CLIENT_ID, CLIENT_SECRET);
transportFactory.transport.addRefreshToken(REFRESH_TOKEN, accessToken1);
OAuth2Credentials userCredentials = UserCredentials.newBuilder().setClientId(CLIENT_ID).setClientSecret(CLIENT_SECRET).setRefreshToken(REFRESH_TOKEN).setHttpTransportFactory(transportFactory).build();
// Use a fixed clock so tokens don't expire
userCredentials.clock = new TestClock();
TestChangeListener listener = new TestChangeListener();
userCredentials.addChangeListener(listener);
Map<String, List<String>> metadata;
assertEquals(0, listener.callCount);
// Get a first token
metadata = userCredentials.getRequestMetadata(CALL_URI);
TestUtils.assertContainsBearerToken(metadata, accessToken1);
assertEquals(accessToken1, listener.accessToken.getTokenValue());
assertEquals(1, listener.callCount);
// Change server to a different token and refresh
transportFactory.transport.addRefreshToken(REFRESH_TOKEN, accessToken2);
// Refresh to force getting next token
userCredentials.refresh();
metadata = userCredentials.getRequestMetadata(CALL_URI);
TestUtils.assertContainsBearerToken(metadata, accessToken2);
assertEquals(accessToken2, listener.accessToken.getTokenValue());
assertEquals(2, listener.callCount);
}
use of com.google.auth.TestClock in project google-auth-library-java by google.
the class OAuth2CredentialsTest method getRequestMetadata_async_refreshRace.
@Test
public void getRequestMetadata_async_refreshRace() throws IOException {
final String accessToken1 = "1/MkSJoj1xsli0AccessToken_NKPY2";
MockTokenServerTransportFactory transportFactory = new MockTokenServerTransportFactory();
transportFactory.transport.addClient(CLIENT_ID, CLIENT_SECRET);
transportFactory.transport.addRefreshToken(REFRESH_TOKEN, accessToken1);
TestClock clock = new TestClock();
OAuth2Credentials credentials = UserCredentials.newBuilder().setClientId(CLIENT_ID).setClientSecret(CLIENT_SECRET).setRefreshToken(REFRESH_TOKEN).setHttpTransportFactory(transportFactory).build();
credentials.clock = clock;
MockExecutor executor = new MockExecutor();
MockRequestMetadataCallback callback = new MockRequestMetadataCallback();
// Getting the first token, which uses the transport and calls the callback in the executor.
credentials.getRequestMetadata(CALL_URI, executor, callback);
assertEquals(0, transportFactory.transport.buildRequestCount);
assertNull(callback.metadata);
// Asynchronous task is scheduled, but beaten by another blocking get call.
assertEquals(1, executor.numTasks());
Map<String, List<String>> metadata = credentials.getRequestMetadata(CALL_URI);
assertEquals(1, transportFactory.transport.buildRequestCount--);
TestUtils.assertContainsBearerToken(metadata, accessToken1);
// When the task is run, the cached data is used.
assertEquals(1, executor.runTasksExhaustively());
assertEquals(0, transportFactory.transport.buildRequestCount);
assertSame(metadata, callback.metadata);
}
use of com.google.auth.TestClock in project google-auth-library-java by google.
the class OAuth2CredentialsTest method getRequestMetadata_blocking_cachesExpiringToken.
@Test
public void getRequestMetadata_blocking_cachesExpiringToken() throws IOException {
final String accessToken1 = "1/MkSJoj1xsli0AccessToken_NKPY2";
final String accessToken2 = "2/MkSJoj1xsli0AccessToken_NKPY2";
MockTokenServerTransportFactory transportFactory = new MockTokenServerTransportFactory();
transportFactory.transport.addClient(CLIENT_ID, CLIENT_SECRET);
transportFactory.transport.addRefreshToken(REFRESH_TOKEN, accessToken1);
TestClock clock = new TestClock();
OAuth2Credentials credentials = UserCredentials.newBuilder().setClientId(CLIENT_ID).setClientSecret(CLIENT_SECRET).setRefreshToken(REFRESH_TOKEN).setHttpTransportFactory(transportFactory).build();
credentials.clock = clock;
// Verify getting the first token
assertEquals(0, transportFactory.transport.buildRequestCount);
Map<String, List<String>> metadata = credentials.getRequestMetadata(CALL_URI);
TestUtils.assertContainsBearerToken(metadata, accessToken1);
assertEquals(1, transportFactory.transport.buildRequestCount--);
// Change server to a different token
transportFactory.transport.addRefreshToken(REFRESH_TOKEN, accessToken2);
// Make transport fail when used next time.
IOException error = new IOException("error");
transportFactory.transport.setError(error);
// Advance 5 minutes and verify original token
clock.addToCurrentTime(5 * 60 * 1000);
metadata = credentials.getRequestMetadata(CALL_URI);
TestUtils.assertContainsBearerToken(metadata, accessToken1);
// Advance 60 minutes and verify revised token
clock.addToCurrentTime(60 * 60 * 1000);
assertEquals(0, transportFactory.transport.buildRequestCount);
try {
credentials.getRequestMetadata(CALL_URI);
fail("Should throw");
} catch (IOException e) {
assertSame(error, e);
assertEquals(1, transportFactory.transport.buildRequestCount--);
}
// Reset the error and try again
transportFactory.transport.setError(null);
metadata = credentials.getRequestMetadata(CALL_URI);
TestUtils.assertContainsBearerToken(metadata, accessToken2);
assertEquals(1, transportFactory.transport.buildRequestCount--);
}
use of com.google.auth.TestClock in project google-auth-library-java by google.
the class ServiceAccountJwtAccessCredentialsTest method getRequestMetadata_async_cache_expired.
@Test
public void getRequestMetadata_async_cache_expired() throws IOException {
TestClock testClock = new TestClock();
PrivateKey privateKey = ServiceAccountCredentials.privateKeyFromPkcs8(SA_PRIVATE_KEY_PKCS8);
ServiceAccountJwtAccessCredentials credentials = ServiceAccountJwtAccessCredentials.newBuilder().setClientId(SA_CLIENT_ID).setClientEmail(SA_CLIENT_EMAIL).setPrivateKey(privateKey).setPrivateKeyId(SA_PRIVATE_KEY_ID).build();
credentials.clock = testClock;
MockExecutor executor = new MockExecutor();
MockRequestMetadataCallback callback1 = new MockRequestMetadataCallback();
credentials.getRequestMetadata(CALL_URI, executor, callback1);
// Fast forward time past the expiration
long lifeSpanMs = TimeUnit.SECONDS.toMillis(ServiceAccountJwtAccessCredentials.LIFE_SPAN_SECS);
testClock.setCurrentTime(lifeSpanMs);
MockRequestMetadataCallback callback2 = new MockRequestMetadataCallback();
credentials.getRequestMetadata(CALL_URI, executor, callback2);
assertNotEquals(callback1.metadata, callback2.metadata);
}
use of com.google.auth.TestClock in project google-auth-library-java by google.
the class ServiceAccountJwtAccessCredentialsTest method getRequestMetadata_blocking_cached.
@Test
public void getRequestMetadata_blocking_cached() throws IOException {
TestClock testClock = new TestClock();
PrivateKey privateKey = ServiceAccountCredentials.privateKeyFromPkcs8(SA_PRIVATE_KEY_PKCS8);
ServiceAccountJwtAccessCredentials credentials = ServiceAccountJwtAccessCredentials.newBuilder().setClientId(SA_CLIENT_ID).setClientEmail(SA_CLIENT_EMAIL).setPrivateKey(privateKey).setPrivateKeyId(SA_PRIVATE_KEY_ID).build();
credentials.clock = testClock;
Map<String, List<String>> metadata1 = credentials.getRequestMetadata(CALL_URI);
// Fast forward time a little
long lifeSpanMs = TimeUnit.SECONDS.toMillis(10);
testClock.setCurrentTime(lifeSpanMs);
Map<String, List<String>> metadata2 = credentials.getRequestMetadata(CALL_URI);
assertEquals(metadata1, metadata2);
}
Aggregations