fixed the tests

Signed-off-by: pmahindrakar-oss <[email protected]>
flyteorg · Jan 12, 2023 · 968544d · 968544d
1 parent d1a3323
commit 968544d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/clients/go/admin/auth_interceptor_test.go b/clients/go/admin/auth_interceptor_test.go
@@ -278,6 +278,6 @@ func TestMaterializeCredentials(t *testing.T) {
 			TokenURL:              fmt.Sprintf("http://localhost:%d/api/v1/token", port),
 			Scopes:                []string{"all"},
 		}, &mocks.TokenCache{}, f)
-		assert.EqualError(t, err, "failed to initialized token source provider. Err: failed to fetch client metadata. Error: rpc error: code = Unknown desc = expected err")
+		assert.EqualError(t, err, "failed to fetch client metadata. Error: rpc error: code = Unknown desc = expected err")
 	})
 }
diff --git a/clients/go/admin/token_source_provider.go b/clients/go/admin/token_source_provider.go
@@ -63,7 +63,9 @@ func NewTokenSourceProvider(ctx context.Context, cfg *Config, tokenCache cache.T
 				scopes = publicClientConfig.Scopes
 			}
 			// Update audience from publicClientConfig
-			audienceValue = publicClientConfig.Audience
+			if cfg.UseAudienceFromAdmin {
+				audienceValue = publicClientConfig.Audience
+			}
 		}
 
 		tokenProvider, err = NewClientCredentialsTokenSourceProvider(ctx, cfg, scopes, tokenURL, audienceValue)