fix(bigquery): bq connection auth scopes (googleapis#6752)

Resolves googleapis#6744
kimihrr · Sep 27, 2022 · 8e09288 · 8e09288
1 parent 7163486
commit 8e09288
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/bigquery/integration_test.go b/bigquery/integration_test.go
@@ -157,7 +157,8 @@ func initIntegrationTest() func() {
 		}
 		bqOpts := []option.ClientOption{option.WithTokenSource(ts)}
 		sOpts := []option.ClientOption{option.WithTokenSource(testutil.TokenSource(ctx, storage.ScopeFullControl))}
-		ptmOpts := []option.ClientOption{option.WithTokenSource(testutil.TokenSource(ctx, "https://www.googleapis.com/auth/cloud-platform"))}
+		ptmOpts := []option.ClientOption{option.WithTokenSource(testutil.TokenSource(ctx, datacatalog.DefaultAuthScopes()...))}
+		connOpts := []option.ClientOption{option.WithTokenSource(testutil.TokenSource(ctx, connection.DefaultAuthScopes()...))}
 		cleanup := func() {}
 		now := time.Now().UTC()
 		if *record {
@@ -196,6 +197,7 @@ func initIntegrationTest() func() {
 			bqOpts = append(bqOpts, grpcHeadersChecker.CallOptions()...)
 			sOpts = append(sOpts, grpcHeadersChecker.CallOptions()...)
 			ptmOpts = append(ptmOpts, grpcHeadersChecker.CallOptions()...)
+			connOpts = append(sOpts, grpcHeadersChecker.CallOptions()...)
 		}
 		var err error
 		client, err = NewClient(ctx, projID, bqOpts...)
@@ -210,7 +212,7 @@ func initIntegrationTest() func() {
 		if err != nil {
 			log.Fatalf("datacatalog.NewPolicyTagManagerClient: %v", err)
 		}
-		connectionsClient, err = connection.NewClient(ctx, sOpts...)
+		connectionsClient, err = connection.NewClient(ctx, connOpts...)
 		if err != nil {
 			log.Fatalf("connection.NewService: %v", err)
 		}