github/genai-toolbox
1
0
Fork 0
mirror of https://github.com/googleapis/genai-toolbox.git synced 2026-02-13 08:35:15 -05:00
Code Issues Packages Projects Releases 14 Wiki Activity

feat: update Oracle test mapping config names and tool references to use 'oracledb'

Browse Source
This commit is contained in:
RUN
2026-01-29 16:25:00 +01:00
parent 05152f732d
commit 6843b46328
2 changed files with 11 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
cmd/root_test.go
View File

@@ -761,13 +761,13 @@ func TestPrebuiltTools(t *testing.T) {
t.Setenv("SNOWFLAKE_WAREHOUSE", "your_wh")
t.Setenv("SNOWFLAKE_ROLE", "your_role")
t.Setenv("ORACLE_USERNAME", "your_oracle_username")
t.Setenv("ORACLE_PASS", "your_oracle_password")
t.Setenv("ORACLE_HOST", "your_oracle_host")
t.Setenv("ORACLE_PORT", "your_oracle_port")
t.Setenv("ORACLE_USERNAME", "your_oracle_db_username")
t.Setenv("ORACLE_PASS", "your_oracle_db_password")
t.Setenv("ORACLE_HOST", "your_oracle_db_host")
t.Setenv("ORACLE_PORT", "your_oracle_db_port")
t.Setenv("ORACLE_SERVICE_NAME", "your_oracle_service_name")
t.Setenv("ORACLE_USE_OCI", "your_oracle_use_oci")
t.Setenv("ORACLE_WALLET_LOCATION", "your_path_to_wallet")
t.Setenv("ORACLE_USE_OCI", "your_oracle_db_use_oci")
t.Setenv("ORACLE_WALLET_LOCATION", "your_path_to_oracldb_wallet")
t.Setenv("ORACLE_TNS_ADMIN", "your_path_to_tns_admin")
ctx, err := testutils.ContextWithNewLogger()
@@ -1083,7 +1083,7 @@ func TestPrebuiltTools(t *testing.T) {
wantToolset: server.ToolsetConfigs{
"snowflake_tools": tools.ToolsetConfig{
Name: "snowflake_tools",
ToolNames: []string{"execute_sql", "list_tables","list_active_sessions","list_top_sql_by_resource","get_query_plan","list_tablespace_usage"},
ToolNames: []string{"execute_sql", "list_tables"},
},
},
},
@@ -1093,7 +1093,7 @@ func TestPrebuiltTools(t *testing.T) {
wantToolset: server.ToolsetConfigs{
"oracle_tools": tools.ToolsetConfig{
Name: "oracle_tools",
ToolNames: []string{"execute_sql", "list_tables"},
ToolNames: []string{"execute_sql", "list_tables","list_active_sessions","list_top_sql_by_resource","get_query_plan","list_tablespace_usage"},
},
},

4
internal/prebuiltconfigs/prebuiltconfigs_test.go
View File

@@ -47,7 +47,7 @@ var expectedToolSources = []string{
"mysql",
"neo4j",
"oceanbase",
"oracle",
"oracledb",
"postgres",
"serverless-spark",
"singlestore",
@@ -132,6 +132,8 @@ func TestGetPrebuiltTool(t *testing.T) {
neo4jconfig := getOrFatal(t, "neo4j")
healthcare_config := getOrFatal(t, "cloud-healthcare")
snowflake_config := getOrFatal(t, "snowflake")
oracle_config := getOrFatal(t,"oracledb")
if len(alloydb_omni_config) <= 0 {
t.Fatalf("unexpected error: could not fetch alloydb omni prebuilt tools yaml")
}