Implementing support for getAccessToken() method for AppIdentityCredentialWrapper

Hiranya Jayathilaka · mattwhisenhunt · commit 819e9c3c3e96 · 2017-10-17T12:01:37.000-07:00
diff --git a/google-api-client-appengine/src/main/java/com/google/api/client/googleapis/extensions/appengine/auth/oauth2/AppIdentityCredential.java b/google-api-client-appengine/src/main/java/com/google/api/client/googleapis/extensions/appengine/auth/oauth2/AppIdentityCredential.java
@@ -15,6 +15,7 @@
 package com.google.api.client.googleapis.extensions.appengine.auth.oauth2;
 
 import com.google.api.client.auth.oauth2.BearerToken;
+import com.google.api.client.auth.oauth2.TokenResponse;
 import com.google.api.client.googleapis.auth.oauth2.GoogleCredential;
 import com.google.api.client.http.HttpExecuteInterceptor;
 import com.google.api.client.http.HttpRequest;
@@ -24,6 +25,7 @@
 import com.google.api.client.util.Beta;
 import com.google.api.client.util.Preconditions;
 import com.google.appengine.api.appidentity.AppIdentityService;
+import com.google.appengine.api.appidentity.AppIdentityService.GetAccessTokenResult;
 import com.google.appengine.api.appidentity.AppIdentityServiceFactory;
 
 import java.io.IOException;
@@ -255,5 +257,17 @@ public GoogleCredential createScoped(Collection<String> scopes) {
           getTransport(),
           getJsonFactory());
     }
+
+    @Override
+    protected TokenResponse executeRefreshToken() throws IOException {
+      GetAccessTokenResult tokenResult = appIdentity.getAppIdentityService()
+          .getAccessToken(appIdentity.getScopes());
+      TokenResponse response = new TokenResponse();
+      response.setAccessToken(tokenResult.getAccessToken());
+      long expiresInSeconds =
+          (tokenResult.getExpirationTime().getTime() - System.currentTimeMillis()) / 1000;
+      response.setExpiresInSeconds(expiresInSeconds);
+      return response;
+    }
   }
 }
diff --git a/google-api-client-appengine/src/test/java/com/google/api/client/googleapis/extensions/appengine/auth/oauth2/AppIdentityCredentialTest.java b/google-api-client-appengine/src/test/java/com/google/api/client/googleapis/extensions/appengine/auth/oauth2/AppIdentityCredentialTest.java
@@ -105,6 +105,25 @@ public void testAppEngineCredentialWrapper() throws IOException {
     assertTrue(authHeader.contains(expectedAccessToken));
   }
 
+  public void testAppEngineCredentialWrapperGetAccessToken() {
+    final String expectedAccessToken = "ExpectedAccessToken";
+    final Collection<String> emptyScopes = Collections.emptyList();
+
+    HttpTransport transport = new MockHttpTransport();
+    JsonFactory jsonFactory = new JacksonFactory();
+
+    MockAppIdentityService appIdentity = new MockAppIdentityService();
+    appIdentity.setAccessTokenText(expectedAccessToken);
+
+    AppIdentityCredential.Builder builder = new AppIdentityCredential.Builder(emptyScopes);
+    builder.setAppIdentityService(appIdentity);
+    AppIdentityCredential appCredential = builder.build();
+
+    GoogleCredential wrapper = new
+        AppIdentityCredential.AppEngineCredentialWrapper(appCredential, transport, jsonFactory);
+    assertEquals(expectedAccessToken, wrapper.getAccessToken());
+  }
+
   public void testAppEngineCredentialWrapperNullTransportThrows() throws IOException {
     JsonFactory jsonFactory = new JacksonFactory();
     try {
