Function TestGCPProviderCustomAudience

auth/subjecttokenprovider_test.go:190–214

(t *testing.T)

}

func TestGCPProviderCustomAudience(t *testing.T) {
	mockClient := &http.Client{
		Transport: roundTripFunc(func(req *http.Request) (*http.Response, error) {
			if !strings.Contains(req.URL.RawQuery, "audience=https%3A%2F%2Fcustom.openai.com") {
				t.Fatalf("Expected custom audience in URL, got %q", req.URL.RawQuery)
			}
			return &http.Response{
				StatusCode: http.StatusOK,
				Body:       io.NopCloser(strings.NewReader("gcp-custom-token-jwt")),
				Header:     make(http.Header),
			}, nil
		}),
	}

	provider := auth.GCPIDTokenProvider(&auth.GCPIDTokenProviderConfig{
		Audience: "https://custom.openai.com",
	})
	token, err := provider.GetToken(context.Background(), mockClient)
	if err != nil {
		t.Fatalf("GetToken() error = %v", err)
	}
	if token != "gcp-custom-token-jwt" {
		t.Errorf("GetToken() = %q, want %q", token, "gcp-custom-token-jwt")
	}
}

func TestAzureProviderNilConfigBackwardCompatibility(t *testing.T) {
	mockClient := &http.Client{