refactor(oauth2): update actor token transport check to ContextRebuildableTransportFactory

Author: macastelaz

google-auth-library-java/oauth2_http/java/com/google/auth/oauth2/IdentityPoolCredentials.java

@@ -31,6 +31,7 @@
 
 package com.google.auth.oauth2;
 
+import com.google.auth.http.ContextRebuildableTransportFactory;
 import com.google.auth.http.HttpTransportFactory;
 import com.google.auth.mtls.MtlsHttpTransportFactory;
 import com.google.auth.mtls.MtlsUtils;
@@ -154,7 +155,7 @@ public class IdentityPoolCredentials extends ExternalAccountCredentials {
     }
 
     if (this.actorTokenSupplier != null
-        && !(this.transportFactory instanceof MtlsHttpTransportFactory)) {
+        && !(this.transportFactory instanceof ContextRebuildableTransportFactory)) {
       throw new IllegalArgumentException(
           "Actor tokens are only supported for mTLS token exchanges. Please configure a certificate source or MtlsHttpTransportFactory.");
     }
@@ -198,8 +199,7 @@ IdentityPoolSubjectTokenSupplier getIdentityPoolSubjectTokenSupplier() {
   }
 
   @VisibleForTesting
-  @Nullable
-  IdentityPoolActorTokenSupplier getIdentityPoolActorTokenSupplier() {
+  @Nullable IdentityPoolActorTokenSupplier getIdentityPoolActorTokenSupplier() {
     return this.actorTokenSupplier;
   }
 

google-auth-library-java/oauth2_http/javatests/com/google/auth/oauth2/IdentityPoolCredentialsTest.java

@@ -45,6 +45,7 @@
 import com.google.api.client.json.GenericJson;
 import com.google.api.client.util.Clock;
 import com.google.auth.TestUtils;
+import com.google.auth.http.ContextRebuildableTransportFactory;
 import com.google.auth.http.HttpTransportFactory;
 import com.google.auth.mtls.MtlsHttpTransportFactory;
 import com.google.auth.mtls.X509Provider;
@@ -1515,8 +1516,7 @@ void createScoped_fileSourcedWithActorToken_preservesSharedSupplierInstance() th
     assertEquals(newScopes, scoped.getScopes());
     // Verify scoped clone maintains a single shared supplier instance for its own cache
     assertSame(
-        scoped.getIdentityPoolSubjectTokenSupplier(),
-        scoped.getIdentityPoolActorTokenSupplier());
+        scoped.getIdentityPoolSubjectTokenSupplier(), scoped.getIdentityPoolActorTokenSupplier());
   }
 
   @Test
@@ -1525,8 +1525,11 @@ void refreshAccessToken_withActorToken_injectsActingPartyIntoStsRequest() throws
         new MockExternalAccountCredentialsTransportFactory();
     KeyStore ks = KeyStore.getInstance(KeyStore.getDefaultType());
     ks.load(null, null);
-    MtlsHttpTransportFactory mtlsTransport =
-        new MtlsHttpTransportFactory(ks) {
+    ContextRebuildableTransportFactory mtlsTransport =
+        new ContextRebuildableTransportFactory() {
+          @Override
+          public void rebuildContext() {}
+
           @Override
           public com.google.api.client.http.HttpTransport create() {
             return transportFactory.create();