Skip to main content
Glama

MCP Toolbox for Databases

by googleapis
Apache 2.0
11,039
  • Linux
clickhouselistdatabases_test.go2.91 kB
// 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 clickhouse import ( "testing" yaml "github.com/goccy/go-yaml" "github.com/google/go-cmp/cmp" "github.com/googleapis/genai-toolbox/internal/server" "github.com/googleapis/genai-toolbox/internal/sources" "github.com/googleapis/genai-toolbox/internal/testutils" "github.com/googleapis/genai-toolbox/internal/tools" ) func TestListDatabasesConfigToolConfigKind(t *testing.T) { cfg := Config{} if cfg.ToolConfigKind() != listDatabasesKind { t.Errorf("expected %q, got %q", listDatabasesKind, cfg.ToolConfigKind()) } } func TestListDatabasesConfigInitializeMissingSource(t *testing.T) { cfg := Config{ Name: "test-list-databases", Kind: listDatabasesKind, Source: "missing-source", Description: "Test list databases tool", } srcs := map[string]sources.Source{} _, err := cfg.Initialize(srcs) if err == nil { t.Error("expected error for missing source") } } func TestParseFromYamlClickHouseListDatabases(t *testing.T) { ctx, err := testutils.ContextWithNewLogger() if err != nil { t.Fatalf("unexpected error: %s", err) } tcs := []struct { desc string in string want server.ToolConfigs }{ { desc: "basic example", in: ` tools: example_tool: kind: clickhouse-list-databases source: my-instance description: some description `, want: server.ToolConfigs{ "example_tool": Config{ Name: "example_tool", Kind: "clickhouse-list-databases", Source: "my-instance", Description: "some description", AuthRequired: []string{}, }, }, }, } for _, tc := range tcs { t.Run(tc.desc, func(t *testing.T) { got := struct { Tools server.ToolConfigs `yaml:"tools"` }{} err := yaml.UnmarshalContext(ctx, testutils.FormatYaml(tc.in), &got) if err != nil { t.Fatalf("unable to unmarshal: %s", err) } if diff := cmp.Diff(tc.want, got.Tools); diff != "" { t.Fatalf("incorrect parse: diff %v", diff) } }) } } func TestListDatabasesToolParseParams(t *testing.T) { tool := Tool{ Parameters: tools.Parameters{}, } params, err := tool.ParseParams(map[string]any{}, map[string]map[string]any{}) if err != nil { t.Errorf("unexpected error: %v", err) } if len(params) != 0 { t.Errorf("expected 0 parameters, got %d", len(params)) } }

MCP directory API

We provide all the information about MCP servers via our MCP API.

curl -X GET 'https://glama.ai/api/mcp/v1/servers/googleapis/genai-toolbox'

If you have feedback or need assistance with the MCP directory API, please join our Discord server