fix: address incorrect universe domain validation when quota project …

…id is set
googleapis · Sep 30, 2024 · 97b73e7 · 97b73e7
1 parent f1f7e96
commit 97b73e7
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/...va/gax/src/main/java/com/google/api/gax/rpc/internal/QuotaProjectIdHidingCredentials.java b/...va/gax/src/main/java/com/google/api/gax/rpc/internal/QuotaProjectIdHidingCredentials.java
@@ -89,4 +89,9 @@ public boolean hasRequestMetadataOnly() {
   public void refresh() throws IOException {
     this.wrappedCredentials.refresh();
   }
+
+  @Override
+  public String getUniverseDomain() throws IOException {
+    return this.wrappedCredentials.getUniverseDomain();
+  }
 }
diff --git a/...ax/src/test/java/com/google/api/gax/rpc/internal/QuotaProjectIdHidingCredentialsTest.java b/...ax/src/test/java/com/google/api/gax/rpc/internal/QuotaProjectIdHidingCredentialsTest.java
@@ -101,4 +101,15 @@ void quotaProjectIdHidingCredentials_getAuthenticationType() throws IOException
 
     Mockito.verify(credentials, Mockito.atLeastOnce()).refresh();
   }
+
+  @Test
+  void quotaProjectIdHidingCredentials_getUniverseDomain() throws IOException {
+    Credentials credentials = Mockito.mock(GoogleCredentials.class);
+    Mockito.when(credentials.getUniverseDomain()).thenReturn("example.com");
+
+    QuotaProjectIdHidingCredentials quotaProjectIdHidingCredentials =
+        new QuotaProjectIdHidingCredentials(credentials);
+
+    assertEquals(quotaProjectIdHidingCredentials.getUniverseDomain(), "example.com");
+  }
 }