initalization_test.go

package mcp_test

import (
	"context"
	"log/slog"
	"net"
	"testing"
	"time"

	"github.com/agent-api/jsonrpc2"
	"github.com/agent-api/mcp"
)

func TestInitialization(t *testing.T) {
	ctx, cancel := context.WithTimeout(context.Background(), 1*time.Second)
	defer cancel()

	// Create in-memory pipe connection
	connA, connB := net.Pipe()

	// Setup server
	serverOpts := mcp.ServerOpts{
		SupportedVersions: []string{"2024-11-05"},
		Info:              &mcp.ServerInfo{Name: "TestServer", Version: "1.0.0"},
		Capabilities: &mcp.ServerCapabilities{
			Logging: &mcp.LoggingCapabilities{},
			Prompts: &mcp.PromptsCapabilities{ListChanged: true},
		},
		Logger: slog.Default(),
	}
	server := mcp.NewServer(&serverOpts)

	// Start server connection
	jsonrpcServer := jsonrpc2.NewConn(ctx, jsonrpc2.NewPlainObjectStream(connA), server.Handler())
	defer jsonrpcServer.Close()

	// Setup client
	jsonrpcClient := jsonrpc2.NewConn(ctx, jsonrpc2.NewPlainObjectStream(connB), nil)
	defer jsonrpcClient.Close()

	clientOpts := mcp.ClientOpts{
		Conn:              jsonrpcClient,
		SupportedVersions: []string{"2024-11-05"},
		Logger:            slog.Default(),
	}
	client := mcp.NewClient(&clientOpts)

	// Perform initialization
	err := client.Initialize(ctx,
		mcp.ClientInfo{Name: "TestClient", Version: "1.0.0"},
		mcp.ClientCapabilities{
			Roots: &mcp.RootsCapabilities{ListChanged: true},
		},
	)
	if err != nil {
		t.Errorf("could not initialize server: %s", err)
	}
}