Add tests

googleapis · Jan 17, 2024 · e9e49e2 · e9e49e2
1 parent cabc7af
commit e9e49e2
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/oauth2_http/java/com/google/auth/oauth2/CloudShellCredentials.java b/oauth2_http/java/com/google/auth/oauth2/CloudShellCredentials.java
@@ -140,6 +140,12 @@ public Builder setAuthPort(int authPort) {
       return this;
     }
 
+    @CanIgnoreReturnValue
+    public Builder setQuotaProjectId(String quotaProjectId) {
+      super.quotaProjectId = quotaProjectId;
+      return this;
+    }
+
     public int getAuthPort() {
       return authPort;
     }

diff --git a/oauth2_http/javatests/com/google/auth/oauth2/CloudShellCredentialsTest.java b/oauth2_http/javatests/com/google/auth/oauth2/CloudShellCredentialsTest.java
@@ -124,4 +124,20 @@ public void serialize() throws IOException, ClassNotFoundException {
     assertEquals(credentials.toString(), deserializedCredentials.toString());
     assertSame(deserializedCredentials.clock, Clock.SYSTEM);
   }
+
+  @Test
+  public void toBuilder() {
+    CloudShellCredentials credentials =
+      CloudShellCredentials.newBuilder()
+            .setAuthPort(42)
+            .setQuotaProjectId("quota-project")
+            .build();
+
+    assertEquals(42, credentials.getAuthPort());
+    assertEquals("quota-project", credentials.getQuotaProjectId());
+
+    CloudShellCredentials secondCredentials = credentials.toBuilder().build();
+
+    assertEquals(credentials, secondCredentials);
+  }
 }