Skip to content

Commit 10880ed

Browse files
duwenxin99duwenxin99
authored
test: add logging and improve ID token retrieval for CI/CD (#2898)
Add logging and remove unnecessary client ID input.
1 parent 7ebfdf1 commit 10880ed

13 files changed

Lines changed: 46 additions & 43 deletions

‎tests/alloydbainl/alloydb_ai_nl_integration_test.go‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -159,7 +159,7 @@ func runAINLToolGetTest(t *testing.T) {
159159

160160
func runAINLToolInvokeTest(t *testing.T) {
161161
// Get ID token
162-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
162+
idToken, err := tests.GetGoogleIdToken(t)
163163
if err != nil {
164164
t.Fatalf("error getting Google ID token: %s", err)
165165
}

‎tests/auth.go‎

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@ import (
1919
"os"
2020
"os/exec"
2121
"strings"
22+
"testing"
2223

2324
"google.golang.org/api/idtoken"
2425
)
@@ -27,13 +28,15 @@ var ServiceAccountEmail = os.Getenv("SERVICE_ACCOUNT_EMAIL")
2728
var ClientId = os.Getenv("CLIENT_ID")
2829

2930
// GetGoogleIdToken retrieve and return the Google ID token
30-
func GetGoogleIdToken(audience string) (string, error) {
31+
func GetGoogleIdToken(t *testing.T) (string, error) {
3132
// For local testing - use gcloud command to print personal ID token
3233
cmd := exec.Command("gcloud", "auth", "print-identity-token")
3334
output, err := cmd.Output()
3435
if err == nil {
3536
return strings.TrimSpace(string(output)), nil
3637
}
38+
t.Logf("gcloud token lookup failed (%v). Falling back to Metadata Server...", err)
39+
3740
// For Cloud Build testing - retrieve ID token from GCE metadata server
3841
ts, err := idtoken.NewTokenSource(context.Background(), ClientId)
3942
if err != nil {

‎tests/bigquery/bigquery_integration_test.go‎

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -952,7 +952,7 @@ func addBigQuerySqlToolConfig(t *testing.T, config map[string]any, toolStatement
952952

953953
func runBigQueryExecuteSqlToolInvokeTest(t *testing.T, select1Want, invokeParamWant, tableNameParam, ddlWant string) {
954954
// Get ID token
955-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
955+
idToken, err := tests.GetGoogleIdToken(t)
956956
if err != nil {
957957
t.Fatalf("error getting Google ID token: %s", err)
958958
}
@@ -1336,7 +1336,7 @@ func runBigQueryWriteModeProtectedTest(t *testing.T, permanentDatasetName string
13361336

13371337
func runBigQueryExecuteSqlToolInvokeDryRunTest(t *testing.T, datasetName string) {
13381338
// Get ID token
1339-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
1339+
idToken, err := tests.GetGoogleIdToken(t)
13401340
if err != nil {
13411341
t.Fatalf("error getting Google ID token: %s", err)
13421342
}
@@ -1444,7 +1444,7 @@ func runBigQueryExecuteSqlToolInvokeDryRunTest(t *testing.T, datasetName string)
14441444
}
14451445

14461446
func runBigQueryForecastToolInvokeTest(t *testing.T, tableName string) {
1447-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
1447+
idToken, err := tests.GetGoogleIdToken(t)
14481448
if err != nil {
14491449
t.Fatalf("error getting Google ID token: %s", err)
14501450
}
@@ -1582,7 +1582,7 @@ func runBigQueryForecastToolInvokeTest(t *testing.T, tableName string) {
15821582
}
15831583

15841584
func runBigQueryAnalyzeContributionToolInvokeTest(t *testing.T, tableName string) {
1585-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
1585+
idToken, err := tests.GetGoogleIdToken(t)
15861586
if err != nil {
15871587
t.Fatalf("error getting Google ID token: %s", err)
15881588
}
@@ -1783,7 +1783,7 @@ func runBigQueryDataTypeTests(t *testing.T) {
17831783

17841784
func runBigQueryListDatasetToolInvokeTest(t *testing.T, datasetWant string) {
17851785
// Get ID token
1786-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
1786+
idToken, err := tests.GetGoogleIdToken(t)
17871787
if err != nil {
17881788
t.Fatalf("error getting Google ID token: %s", err)
17891789
}
@@ -1905,7 +1905,7 @@ func runBigQueryListDatasetToolInvokeTest(t *testing.T, datasetWant string) {
19051905

19061906
func runBigQueryGetDatasetInfoToolInvokeTest(t *testing.T, datasetName, datasetInfoWant string) {
19071907
// Get ID token
1908-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
1908+
idToken, err := tests.GetGoogleIdToken(t)
19091909
if err != nil {
19101910
t.Fatalf("error getting Google ID token: %s", err)
19111911
}
@@ -2055,7 +2055,7 @@ func runBigQueryGetDatasetInfoToolInvokeTest(t *testing.T, datasetName, datasetI
20552055

20562056
func runBigQueryListTableIdsToolInvokeTest(t *testing.T, datasetName, tablename_want string) {
20572057
// Get ID token
2058-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
2058+
idToken, err := tests.GetGoogleIdToken(t)
20592059
if err != nil {
20602060
t.Fatalf("error getting Google ID token: %s", err)
20612061
}
@@ -2205,7 +2205,7 @@ func runBigQueryListTableIdsToolInvokeTest(t *testing.T, datasetName, tablename_
22052205

22062206
func runBigQueryGetTableInfoToolInvokeTest(t *testing.T, datasetName, tableName, tableInfoWant string) {
22072207
// Get ID token
2208-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
2208+
idToken, err := tests.GetGoogleIdToken(t)
22092209
if err != nil {
22102210
t.Fatalf("error getting Google ID token: %s", err)
22112211
}
@@ -2358,7 +2358,7 @@ func runBigQueryConversationalAnalyticsInvokeTest(t *testing.T, datasetName, tab
23582358
const maxRetries = 3
23592359
const requestTimeout = 340 * time.Second
23602360
// Get ID token
2361-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
2361+
idToken, err := tests.GetGoogleIdToken(t)
23622362
if err != nil {
23632363
t.Fatalf("error getting Google ID token: %s", err)
23642364
}
@@ -2921,7 +2921,7 @@ func runConversationalAnalyticsWithRestriction(t *testing.T, allowedDatasetName,
29212921

29222922
func runBigQuerySearchCatalogToolInvokeTest(t *testing.T, datasetName string, tableName string) {
29232923
// Get ID token
2924-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
2924+
idToken, err := tests.GetGoogleIdToken(t)
29252925
if err != nil {
29262926
t.Fatalf("error getting Google ID token: %s", err)
29272927
}

‎tests/cloudgda/cloud_gda_integration_test.go‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -557,7 +557,7 @@ func TestCloudGDAConservationalAnalyticsTools(t *testing.T) {
557557
}
558558

559559
func runListAccessibleDataAgentsInvokeTest(t *testing.T, dataAgentDisplayName string) {
560-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
560+
idToken, err := tests.GetGoogleIdToken(t)
561561
if err != nil {
562562
t.Fatalf("error getting Google ID token: %s", err)
563563
}
@@ -663,7 +663,7 @@ func runListAccessibleDataAgentsInvokeTest(t *testing.T, dataAgentDisplayName st
663663
}
664664

665665
func runGetDataAgentInfoInvokeTest(t *testing.T, dataAgentName, dataAgentDisplayName string) {
666-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
666+
idToken, err := tests.GetGoogleIdToken(t)
667667
if err != nil {
668668
t.Fatalf("error getting Google ID token: %s", err)
669669
}
@@ -767,7 +767,7 @@ func runAskDataAgentInvokeTest(t *testing.T, dataAgentID string) {
767767
const maxRetries = 3
768768
const requestTimeout = 340 * time.Second
769769

770-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
770+
idToken, err := tests.GetGoogleIdToken(t)
771771
if err != nil {
772772
t.Fatalf("error getting Google ID token: %s", err)
773773
}

‎tests/cloudhealthcare/cloud_healthcare_integration_test.go‎

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -736,7 +736,7 @@ func addClientAuthSourceConfig(t *testing.T, config map[string]any) map[string]a
736736
}
737737

738738
func runGetDatasetToolInvokeTest(t *testing.T, want string) {
739-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
739+
idToken, err := tests.GetGoogleIdToken(t)
740740
if err != nil {
741741
t.Fatalf("error getting Google ID token: %s", err)
742742
}
@@ -835,7 +835,7 @@ func runGetDatasetToolInvokeTest(t *testing.T, want string) {
835835
}
836836

837837
func runListFHIRStoresToolInvokeTest(t *testing.T, want string) {
838-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
838+
idToken, err := tests.GetGoogleIdToken(t)
839839
if err != nil {
840840
t.Fatalf("error getting Google ID token: %s", err)
841841
}
@@ -934,7 +934,7 @@ func runListFHIRStoresToolInvokeTest(t *testing.T, want string) {
934934
}
935935

936936
func runListDICOMStoresToolInvokeTest(t *testing.T) {
937-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
937+
idToken, err := tests.GetGoogleIdToken(t)
938938
if err != nil {
939939
t.Fatalf("error getting Google ID token: %s", err)
940940
}
@@ -1029,7 +1029,7 @@ func runListDICOMStoresToolInvokeTest(t *testing.T) {
10291029
}
10301030

10311031
func runGetFHIRStoreToolInvokeTest(t *testing.T, fhirStoreID, want string) {
1032-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
1032+
idToken, err := tests.GetGoogleIdToken(t)
10331033
if err != nil {
10341034
t.Fatalf("error getting Google ID token: %s", err)
10351035
}
@@ -1135,7 +1135,7 @@ func runGetFHIRStoreToolInvokeTest(t *testing.T, fhirStoreID, want string) {
11351135
}
11361136

11371137
func runGetFHIRStoreMetricsToolInvokeTest(t *testing.T, fhirStoreID, want string) {
1138-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
1138+
idToken, err := tests.GetGoogleIdToken(t)
11391139
if err != nil {
11401140
t.Fatalf("error getting Google ID token: %s", err)
11411141
}
@@ -1241,7 +1241,7 @@ func runGetFHIRStoreMetricsToolInvokeTest(t *testing.T, fhirStoreID, want string
12411241
}
12421242

12431243
func runGetFHIRResourceToolInvokeTest(t *testing.T, storeID, resType, resID, want string) {
1244-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
1244+
idToken, err := tests.GetGoogleIdToken(t)
12451245
if err != nil {
12461246
t.Fatalf("error getting Google ID token: %s", err)
12471247
}
@@ -1354,7 +1354,7 @@ func runGetFHIRResourceToolInvokeTest(t *testing.T, storeID, resType, resID, wan
13541354
}
13551355

13561356
func runFHIRPatientSearchToolInvokeTest(t *testing.T, fhirStoreID string, patientIDs ...string) {
1357-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
1357+
idToken, err := tests.GetGoogleIdToken(t)
13581358
if err != nil {
13591359
t.Fatalf("error getting Google ID token: %s", err)
13601360
}
@@ -1508,7 +1508,7 @@ func runFHIRPatientSearchToolInvokeTest(t *testing.T, fhirStoreID string, patien
15081508
}
15091509

15101510
func runFHIRPatientEverythingToolInvokeTest(t *testing.T, fhirStoreID, patientID, want string) {
1511-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
1511+
idToken, err := tests.GetGoogleIdToken(t)
15121512
if err != nil {
15131513
t.Fatalf("error getting Google ID token: %s", err)
15141514
}
@@ -1645,7 +1645,7 @@ func runFHIRPatientEverythingToolInvokeTest(t *testing.T, fhirStoreID, patientID
16451645
}
16461646

16471647
func runFHIRFetchPageToolInvokeTest(t *testing.T, pageURL, want string) {
1648-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
1648+
idToken, err := tests.GetGoogleIdToken(t)
16491649
if err != nil {
16501650
t.Fatalf("error getting Google ID token: %s", err)
16511651
}
@@ -1848,7 +1848,7 @@ func runListDICOMStoresWithRestriction(t *testing.T, allowedDICOMStore, disallow
18481848
}
18491849

18501850
func runGetDICOMStoreToolInvokeTest(t *testing.T, dicomStoreID, want string) {
1851-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
1851+
idToken, err := tests.GetGoogleIdToken(t)
18521852
if err != nil {
18531853
t.Fatalf("error getting Google ID token: %s", err)
18541854
}
@@ -1954,7 +1954,7 @@ func runGetDICOMStoreToolInvokeTest(t *testing.T, dicomStoreID, want string) {
19541954
}
19551955

19561956
func runGetDICOMStoreMetricsToolInvokeTest(t *testing.T, dicomStoreID, want string) {
1957-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
1957+
idToken, err := tests.GetGoogleIdToken(t)
19581958
if err != nil {
19591959
t.Fatalf("error getting Google ID token: %s", err)
19601960
}
@@ -2060,7 +2060,7 @@ func runGetDICOMStoreMetricsToolInvokeTest(t *testing.T, dicomStoreID, want stri
20602060
}
20612061

20622062
func runSearchDICOMStudiesToolInvokeTest(t *testing.T, dicomStoreID string) {
2063-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
2063+
idToken, err := tests.GetGoogleIdToken(t)
20642064
if err != nil {
20652065
t.Fatalf("error getting Google ID token: %s", err)
20662066
}
@@ -2182,7 +2182,7 @@ func runSearchDICOMStudiesToolInvokeTest(t *testing.T, dicomStoreID string) {
21822182
}
21832183

21842184
func runSearchDICOMSeriesToolInvokeTest(t *testing.T, dicomStoreID string) {
2185-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
2185+
idToken, err := tests.GetGoogleIdToken(t)
21862186
if err != nil {
21872187
t.Fatalf("error getting Google ID token: %s", err)
21882188
}
@@ -2304,7 +2304,7 @@ func runSearchDICOMSeriesToolInvokeTest(t *testing.T, dicomStoreID string) {
23042304
}
23052305

23062306
func runSearchDICOMInstancesToolInvokeTest(t *testing.T, dicomStoreID string) {
2307-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
2307+
idToken, err := tests.GetGoogleIdToken(t)
23082308
if err != nil {
23092309
t.Fatalf("error getting Google ID token: %s", err)
23102310
}
@@ -2426,7 +2426,7 @@ func runSearchDICOMInstancesToolInvokeTest(t *testing.T, dicomStoreID string) {
24262426
}
24272427

24282428
func runRetrieveRenderedDICOMInstanceToolInvokeTest(t *testing.T, dicomStoreID string) {
2429-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
2429+
idToken, err := tests.GetGoogleIdToken(t)
24302430
if err != nil {
24312431
t.Fatalf("error getting Google ID token: %s", err)
24322432
}

‎tests/cloudloggingadmin/cloud_logging_admin_integration_test.go‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -235,7 +235,7 @@ func runListLogNamesTest(t *testing.T) {
235235
}
236236

237237
func runAuthListLogNamesTest(t *testing.T) {
238-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
238+
idToken, err := tests.GetGoogleIdToken(t)
239239
if err != nil {
240240
t.Fatalf("error getting Google ID token: %s", err)
241241
}

‎tests/dataplex/dataplex_integration_test.go‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -418,7 +418,7 @@ func runDataplexToolGetTest(t *testing.T) {
418418
}
419419

420420
func runDataplexSearchEntriesToolInvokeTest(t *testing.T, tableName string, datasetName string) {
421-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
421+
idToken, err := tests.GetGoogleIdToken(t)
422422
if err != nil {
423423
t.Fatalf("error getting Google ID token: %s", err)
424424
}
@@ -551,7 +551,7 @@ func runDataplexSearchEntriesToolInvokeTest(t *testing.T, tableName string, data
551551
}
552552

553553
func runDataplexLookupEntryToolInvokeTest(t *testing.T, tableName string, datasetName string) {
554-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
554+
idToken, err := tests.GetGoogleIdToken(t)
555555
if err != nil {
556556
t.Fatalf("error getting Google ID token: %s", err)
557557
}
@@ -721,7 +721,7 @@ func runDataplexLookupEntryToolInvokeTest(t *testing.T, tableName string, datase
721721
}
722722

723723
func runDataplexSearchAspectTypesToolInvokeTest(t *testing.T, aspectTypeId string) {
724-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
724+
idToken, err := tests.GetGoogleIdToken(t)
725725
if err != nil {
726726
t.Fatalf("error getting Google ID token: %s", err)
727727
}
@@ -837,7 +837,7 @@ func runDataplexSearchAspectTypesToolInvokeTest(t *testing.T, aspectTypeId strin
837837
}
838838

839839
func runDataplexLookupContextToolInvokeTest(t *testing.T, tableName string, datasetName string) {
840-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
840+
idToken, err := tests.GetGoogleIdToken(t)
841841
if err != nil {
842842
t.Fatalf("error getting Google ID token: %s", err)
843843
}

‎tests/dataproc/dataproc_integration_test.go‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -593,7 +593,7 @@ func listClustersRpc(t *testing.T, client *dataproc.ClusterControllerClient, ctx
593593
}
594594

595595
func runAuthTest(t *testing.T, toolName string, request map[string]any, wantStatus int) {
596-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
596+
idToken, err := tests.GetGoogleIdToken(t)
597597
if err != nil {
598598
t.Fatalf("error getting Google ID token: %s", err)
599599
}

‎tests/http/http_integration_test.go‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ var (
4343
)
4444

4545
func getHTTPSourceConfig(t *testing.T) map[string]any {
46-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
46+
idToken, err := tests.GetGoogleIdToken(t)
4747
if err != nil {
4848
t.Fatalf("error getting ID token: %s", err)
4949
}

‎tests/http/http_mcp_test.go‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ import (
3232
)
3333

3434
func getMCPHTTPSourceConfig(t *testing.T) map[string]any {
35-
idToken, err := tests.GetGoogleIdToken(tests.ClientId)
35+
idToken, err := tests.GetGoogleIdToken(t)
3636
if err != nil {
3737
t.Logf("Warning: error getting ID token: %s. Using dummy token.", err)
3838
idToken = "dummy-token"

0 commit comments

Comments
 (0)