From 0871c573350b4977ca8bce1af905525e97d2fccc Mon Sep 17 00:00:00 2001 From: Yuan Teoh Date: Wed, 29 Jan 2025 13:58:07 -0800 Subject: [PATCH] update auth test --- tests/auth_test.go | 6 +++--- tests/common_test.go | 4 ++-- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/tests/auth_test.go b/tests/auth_test.go index 2c4ce7a3a..9c26a8a7a 100644 --- 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)]" diff --git a/tests/common_test.go b/tests/common_test.go index ed32264fc..f768481ce 100644 --- a/tests/common_test.go +++ b/tests/common_test.go @@ -215,7 +215,7 @@ func RunToolInvocationWithParamsTest(t *testing.T, sourceConfig map[string]any, switch toolKind { case "postgres-sql": statement = fmt.Sprintf("SELECT * FROM %s WHERE id = $1 OR name = $2;", tableName) - case "mssql": + case "mssql-sql": statement = fmt.Sprintf("SELECT * FROM %s WHERE id = @id OR name = @p2;", tableName) default: t.Fatalf("invalid tool kind: %s", toolKind) @@ -224,7 +224,7 @@ func RunToolInvocationWithParamsTest(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-tool\"! Parameters parsed: [{\"id\" '\\x03'} {\"name\" \"Alice\"}] \n Output: [%!s(int64=1) Alice][%!s(int64=3) Sid]" default: wantString = "Stub tool call for \"my-tool\"! Parameters parsed: [{\"id\" '\\x03'} {\"name\" \"Alice\"}] \n Output: [%!s(int32=1) Alice][%!s(int32=3) Sid]"