testutil

func ForAllTransports ¶ added in v0.0.19

func ForAllTransports(t *testing.T, srv *server.Server, fn func(t *testing.T, c *client.Client))

ForAllTransports runs fn as a subtest against all 4 MCP transport types: Streamable HTTP, SSE, in-process memory, and stdio. Each subtest creates its own server instance, connected client, and cleanup handlers.

Use this for any test that should be transport-agnostic — the test logic runs identically regardless of the underlying transport.

The server parameter is used as a template: ForAllTransports creates a fresh server for each transport by calling NewTestServer(). If you need the exact server instance passed in (e.g., with custom tools), note that Streamable HTTP and SSE subtests create httptest.Servers from it directly, while memory and stdio create fresh copies via NewTestServer() to avoid shared state.

Example:

func TestEcho(t *testing.T) {
    testutil.ForAllTransports(t, testutil.NewTestServer(), func(t *testing.T, c *client.Client) {
        text, err := c.ToolCall("echo", map[string]any{"message": "hi"})
        if err != nil { t.Fatal(err) }
        assert.Equal(t, "echo: hi", text)
    })
}

func InitHandshake ¶ added in v0.0.19

func InitHandshake(d interface {
	Dispatch(context.Context, *core.Request) *core.Response
})

InitHandshake performs the MCP initialize + notifications/initialized handshake on any type that implements the Dispatch method (both server.Server and server.Dispatcher satisfy this).

This is required before the server will accept tool/resource requests. For client-based tests, prefer NewTestClient which handles the handshake automatically via client.Connect().

Example:

srv := server.NewServer(info)
srv.RegisterTool(...)
testutil.InitHandshake(srv)
resp := srv.Dispatch(ctx, toolCallRequest)

func NewTestServer ¶ added in v0.0.19

func NewTestServer() *server.Server

NewTestServer creates an MCP server pre-registered with standard test fixtures for use across unit and integration tests.

Standard fixtures:

• Tool "echo": echoes input message → "echo: <message>"
• Tool "fail": always returns an error result ("intentional failure")
• Resource "test://info": static text → "hello from test"
• ResourceTemplate "test://items/{id}": parameterized → "item <id>"

Additional tools, resources, or prompts can be registered on the returned server before passing it to NewTestClient or ForAllTransports.

Example:

srv := testutil.NewTestServer()
srv.RegisterTool(myCustomTool, myHandler) // add test-specific tools
tc := testutil.NewTestClient(t, srv)
result := tc.ToolCall("echo", map[string]any{"message": "hi"})