4f6510b9ae
In accordance with the updated design, merge the protocol package into the mcp package. To do so, rename protocol.Content to WireContent and protocol.ResourceContents to WireResource. Also, remove some stray logging in readResource. Change-Id: I76f61ce8d67dd3ac520ef2e1959ade0a0785a480 Reviewed-on: https://go-review.googlesource.com/c/tools/+/672375 LUCI-TryBot-Result: Go LUCI <golang-scoped@luci-project-accounts.iam.gserviceaccount.com> Reviewed-by: Jonathan Amsterdam <jba@google.com>
90 lines
2.6 KiB
Go
90 lines
2.6 KiB
Go
// Copyright 2025 The Go MCP SDK Authors. All rights reserved.
|
|
// Use of this source code is governed by an MIT-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package mcp_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/google/go-cmp/cmp"
|
|
"github.com/google/go-cmp/cmp/cmpopts"
|
|
"github.com/modelcontextprotocol/go-sdk/mcp"
|
|
"github.com/modelcontextprotocol/go-sdk/jsonschema"
|
|
)
|
|
|
|
// testToolHandler is used for type inference in TestNewTool.
|
|
func testToolHandler[T any](context.Context, *mcp.ServerConnection, T) ([]mcp.Content, error) {
|
|
panic("not implemented")
|
|
}
|
|
|
|
func TestNewTool(t *testing.T) {
|
|
tests := []struct {
|
|
tool *mcp.ServerTool
|
|
want *jsonschema.Schema
|
|
}{
|
|
{
|
|
mcp.NewTool("basic", "", testToolHandler[struct {
|
|
Name string `json:"name"`
|
|
}]),
|
|
&jsonschema.Schema{
|
|
Type: "object",
|
|
Required: []string{"name"},
|
|
Properties: map[string]*jsonschema.Schema{
|
|
"name": {Type: "string"},
|
|
},
|
|
AdditionalProperties: &jsonschema.Schema{Not: new(jsonschema.Schema)},
|
|
},
|
|
},
|
|
{
|
|
mcp.NewTool("enum", "", testToolHandler[struct{ Name string }], mcp.Input(
|
|
mcp.Property("Name", mcp.Enum("x", "y", "z")),
|
|
)),
|
|
&jsonschema.Schema{
|
|
Type: "object",
|
|
Required: []string{"Name"},
|
|
Properties: map[string]*jsonschema.Schema{
|
|
"Name": {Type: "string", Enum: []any{"x", "y", "z"}},
|
|
},
|
|
AdditionalProperties: &jsonschema.Schema{Not: new(jsonschema.Schema)},
|
|
},
|
|
},
|
|
{
|
|
mcp.NewTool("required", "", testToolHandler[struct {
|
|
Name string `json:"name"`
|
|
Language string `json:"language"`
|
|
X int `json:"x,omitempty"`
|
|
Y int `json:"y,omitempty"`
|
|
}], mcp.Input(
|
|
mcp.Property("x", mcp.Required(true)))),
|
|
&jsonschema.Schema{
|
|
Type: "object",
|
|
Required: []string{"name", "language", "x"},
|
|
Properties: map[string]*jsonschema.Schema{
|
|
"language": {Type: "string"},
|
|
"name": {Type: "string"},
|
|
"x": {Type: "integer"},
|
|
"y": {Type: "integer"},
|
|
},
|
|
AdditionalProperties: &jsonschema.Schema{Not: new(jsonschema.Schema)},
|
|
},
|
|
},
|
|
{
|
|
mcp.NewTool("set_schema", "", testToolHandler[struct {
|
|
X int `json:"x,omitempty"`
|
|
Y int `json:"y,omitempty"`
|
|
}], mcp.Input(
|
|
mcp.Schema(&jsonschema.Schema{Type: "object"})),
|
|
),
|
|
&jsonschema.Schema{
|
|
Type: "object",
|
|
},
|
|
},
|
|
}
|
|
for _, test := range tests {
|
|
if diff := cmp.Diff(test.want, test.tool.Definition.InputSchema, cmpopts.IgnoreUnexported(jsonschema.Schema{})); diff != "" {
|
|
t.Errorf("NewTool(%v) mismatch (-want +got):\n%s", test.tool.Definition.Name, diff)
|
|
}
|
|
}
|
|
}
|