add unit tests

googleapis · Apr 1, 2024 · 4b1a44f · 4b1a44f
1 parent 78e866a
commit 4b1a44f
Show file tree

Hide file tree

Showing 2 changed files with 123 additions and 0 deletions.
diff --git a/auth/credentials/impersonate/impersonate_test.go b/auth/credentials/impersonate/impersonate_test.go
@@ -138,3 +138,43 @@ func TestNewCredentials_serviceAccount(t *testing.T) {
 type RoundTripFn func(req *http.Request) *http.Response
 
 func (f RoundTripFn) RoundTrip(req *http.Request) (*http.Response, error) { return f(req), nil }
+
+func TestCredentialsOptions_UniverseDomain(t *testing.T) {
+	testCases := []struct {
+		name               string
+		opts               *CredentialsOptions
+		wantUniverseDomain string
+		wantIsGDU          bool
+	}{
+		{
+			name:               "empty",
+			opts:               &CredentialsOptions{},
+			wantUniverseDomain: "googleapis.com",
+			wantIsGDU:          true,
+		},
+		{
+			name: "defaults",
+			opts: &CredentialsOptions{
+				UniverseDomain: "googleapis.com",
+			},
+			wantUniverseDomain: "googleapis.com",
+			wantIsGDU:          true,
+		},
+		{
+			name: "non-GDU",
+			opts: &CredentialsOptions{
+				UniverseDomain: "example.com",
+			},
+			wantUniverseDomain: "example.com",
+			wantIsGDU:          false,
+		},
+	}
+	for _, tc := range testCases {
+		if got := tc.opts.getUniverseDomain(); got != tc.wantUniverseDomain {
+			t.Errorf("%s: got %v, want %v", tc.name, got, tc.wantUniverseDomain)
+		}
+		if got := tc.opts.isUniverseDomainGDU(); got != tc.wantIsGDU {
+			t.Errorf("%s: got %v, want %v", tc.name, got, tc.wantIsGDU)
+		}
+	}
+}
diff --git a/auth/internal/transport/cba_test.go b/auth/internal/transport/cba_test.go
@@ -64,6 +64,89 @@ var (
 	fakeClientCertSource = func(info *tls.CertificateRequestInfo) (*tls.Certificate, error) { return nil, nil }
 )
 
+func TestOptions_UniverseDomain(t *testing.T) {
+	testCases := []struct {
+		name                      string
+		opts                      *Options
+		wantDefaultUniverseDomain string
+		wantUniverseDomain        string
+		wantDefaultEndpoint       string
+		wantIsGDU                 bool
+		wantMergedEndpoint        string
+	}{
+		{
+			name:                      "empty",
+			opts:                      &Options{},
+			wantDefaultUniverseDomain: "googleapis.com",
+			wantUniverseDomain:        "googleapis.com",
+			wantDefaultEndpoint:       "",
+			wantIsGDU:                 true,
+			wantMergedEndpoint:        "",
+		},
+		{
+			name: "defaults",
+			opts: &Options{
+				DefaultEndpointTemplate: "https://test.UNIVERSE_DOMAIN/",
+				DefaultUniverseDomain:   "googleapis.com",
+			},
+			wantDefaultUniverseDomain: "googleapis.com",
+			wantUniverseDomain:        "googleapis.com",
+			wantDefaultEndpoint:       "https://test.googleapis.com/",
+			wantIsGDU:                 true,
+			wantMergedEndpoint:        "",
+		},
+		{
+			name: "non-GDU",
+			opts: &Options{
+				DefaultEndpointTemplate: "https://test.UNIVERSE_DOMAIN/",
+				DefaultUniverseDomain:   "googleapis.com",
+				UniverseDomain:          "example.com",
+			},
+			wantDefaultUniverseDomain: "googleapis.com",
+			wantUniverseDomain:        "example.com",
+			wantDefaultEndpoint:       "https://test.example.com/",
+			wantIsGDU:                 false,
+			wantMergedEndpoint:        "",
+		},
+		{
+			name: "merged endpoint",
+			opts: &Options{
+				DefaultEndpointTemplate: "https://test.UNIVERSE_DOMAIN/bar/baz",
+				DefaultUniverseDomain:   "googleapis.com",
+				Endpoint:                "myhost:8000",
+			},
+			wantDefaultUniverseDomain: "googleapis.com",
+			wantUniverseDomain:        "googleapis.com",
+			wantDefaultEndpoint:       "https://test.googleapis.com/bar/baz",
+			wantIsGDU:                 true,
+			wantMergedEndpoint:        "https://myhost:8000/bar/baz",
+		},
+	}
+	for _, tc := range testCases {
+		if got := tc.opts.getDefaultUniverseDomain(); got != tc.wantDefaultUniverseDomain {
+			t.Errorf("%s: got %v, want %v", tc.name, got, tc.wantDefaultUniverseDomain)
+		}
+		if got := tc.opts.getUniverseDomain(); got != tc.wantUniverseDomain {
+			t.Errorf("%s: got %v, want %v", tc.name, got, tc.wantUniverseDomain)
+		}
+		if got := tc.opts.isUniverseDomainGDU(); got != tc.wantIsGDU {
+			t.Errorf("%s: got %v, want %v", tc.name, got, tc.wantIsGDU)
+		}
+		if got := tc.opts.defaultEndpoint(); got != tc.wantDefaultEndpoint {
+			t.Errorf("%s: got %v, want %v", tc.name, got, tc.wantDefaultEndpoint)
+		}
+		if tc.opts.Endpoint != "" {
+			got, err := tc.opts.mergedEndpoint()
+			if err != nil {
+				t.Fatalf("%s: %v", tc.name, err)
+			}
+			if got != tc.wantMergedEndpoint {
+				t.Errorf("%s: got %v, want %v", tc.name, got, tc.wantMergedEndpoint)
+			}
+		}
+	}
+}
+
 func TestGetEndpoint(t *testing.T) {
 	testCases := []struct {
 		endpoint                string