MCP Toolbox for Databases

// Copyright 2025 Google LLC // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. // You may obtain a copy of the License at // // http://www.apache.org/licenses/LICENSE-2.0 // // Unless required by applicable law or agreed to in writing, software // distributed under the License is distributed on an "AS IS" BASIS, // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // See the License for the specific language governing permissions and // limitations under the License. package resources_test import ( "testing" "github.com/google/go-cmp/cmp" "github.com/googleapis/genai-toolbox/internal/auth" "github.com/googleapis/genai-toolbox/internal/prompts" "github.com/googleapis/genai-toolbox/internal/server/resources" "github.com/googleapis/genai-toolbox/internal/sources" "github.com/googleapis/genai-toolbox/internal/sources/alloydbpg" "github.com/googleapis/genai-toolbox/internal/tools" ) func TestUpdateServer(t *testing.T) { newSources := map[string]sources.Source{ "example-source": &alloydbpg.Source{ Config: alloydbpg.Config{ Name: "example-alloydb-source", Kind: "alloydb-postgres", }, }, } newAuth := map[string]auth.AuthService{"example-auth": nil} newTools := map[string]tools.Tool{"example-tool": nil} newToolsets := map[string]tools.Toolset{ "example-toolset": { ToolsetConfig: tools.ToolsetConfig{ Name: "example-toolset", }, Tools: []*tools.Tool{}, }, } newPrompts := map[string]prompts.Prompt{"example-prompt": nil} newPromptsets := map[string]prompts.Promptset{ "example-promptset": { PromptsetConfig: prompts.PromptsetConfig{ Name: "example-promptset", }, Prompts: []*prompts.Prompt{}, }, } resMgr := resources.NewResourceManager(newSources, newAuth, newTools, newToolsets, newPrompts, newPromptsets) gotSource, _ := resMgr.GetSource("example-source") if diff := cmp.Diff(gotSource, newSources["example-source"]); diff != "" { t.Errorf("error updating server, sources (-want +got):\n%s", diff) } gotAuthService, _ := resMgr.GetAuthService("example-auth") if diff := cmp.Diff(gotAuthService, newAuth["example-auth"]); diff != "" { t.Errorf("error updating server, authServices (-want +got):\n%s", diff) } gotTool, _ := resMgr.GetTool("example-tool") if diff := cmp.Diff(gotTool, newTools["example-tool"]); diff != "" { t.Errorf("error updating server, tools (-want +got):\n%s", diff) } gotToolset, _ := resMgr.GetToolset("example-toolset") if diff := cmp.Diff(gotToolset, newToolsets["example-toolset"]); diff != "" { t.Errorf("error updating server, toolset (-want +got):\n%s", diff) } gotPrompt, _ := resMgr.GetPrompt("example-prompt") if diff := cmp.Diff(gotPrompt, newPrompts["example-prompt"]); diff != "" { t.Errorf("error updating server, prompts (-want +got):\n%s", diff) } gotPromptset, _ := resMgr.GetPromptset("example-promptset") if diff := cmp.Diff(gotPromptset, newPromptsets["example-promptset"]); diff != "" { t.Errorf("error updating server, promptset (-want +got):\n%s", diff) } updateSource := map[string]sources.Source{ "example-source2": &alloydbpg.Source{ Config: alloydbpg.Config{ Name: "example-alloydb-source2", Kind: "alloydb-postgres", }, }, } resMgr.SetResources(updateSource, newAuth, newTools, newToolsets, newPrompts, newPromptsets) gotSource, _ = resMgr.GetSource("example-source2") if diff := cmp.Diff(gotSource, updateSource["example-source2"]); diff != "" { t.Errorf("error updating server, sources (-want +got):\n%s", diff) } }