update auth test

googleapis · Jan 29, 2025 · e6b0aca · e6b0aca
1 parent 093aedb
commit e6b0aca
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/auth_test.go b/tests/auth_test.go
@@ -63,7 +63,7 @@ func RunGoogleAuthenticatedParameterTest(t *testing.T, sourceConfig map[string]a
 	switch {
 	case strings.EqualFold(toolKind, "postgres-sql"):
 		statement = fmt.Sprintf("SELECT * FROM %s WHERE email = $1;", tableName)
-	case strings.EqualFold(toolKind, "mssql"):
+	case strings.EqualFold(toolKind, "mssql-sql"):
 		statement = fmt.Sprintf("SELECT * FROM %s WHERE email = @email;", tableName)
 	default:
 		t.Fatalf("invalid tool kind: %s", toolKind)
@@ -132,7 +132,7 @@ func RunGoogleAuthenticatedParameterTest(t *testing.T, sourceConfig map[string]a
 	// Tools using database/sql interface only outputs `int64` instead of `int32`
 	var wantString string
 	switch toolKind {
-	case "mssql":
+	case "mssql-sql":
 		wantString = fmt.Sprintf("Stub tool call for \"my-auth-tool\"! Parameters parsed: [{\"email\" \"%s\"}] \n Output: [%%!s(int64=1) Alice %s]", SERVICE_ACCOUNT_EMAIL, SERVICE_ACCOUNT_EMAIL)
 	default:
 		wantString = fmt.Sprintf("Stub tool call for \"my-auth-tool\"! Parameters parsed: [{\"email\" \"%s\"}] \n Output: [%%!s(int32=1) Alice %s]", SERVICE_ACCOUNT_EMAIL, SERVICE_ACCOUNT_EMAIL)
@@ -216,7 +216,7 @@ func RunAuthRequiredToolInvocationTest(t *testing.T, sourceConfig map[string]any
 	// Tools using database/sql interface only outputs `int64` instead of `int32`
 	var wantString string
 	switch toolKind {
-	case "mssql":
+	case "mssql-sql":
 		wantString = "Stub tool call for \"my-auth-tool\"! Parameters parsed: [] \n Output: [%!s(int64=1)]"
 	default:
 		wantString = "Stub tool call for \"my-auth-tool\"! Parameters parsed: [] \n Output: [%!s(int32=1)]"