diff --git a/examples/sse_test_client.rb b/examples/streaming_http_test_client.rb similarity index 100% rename from examples/sse_test_client.rb rename to examples/streaming_http_test_client.rb diff --git a/examples/sse_test_server.rb b/examples/streaming_http_test_server.rb similarity index 99% rename from examples/sse_test_server.rb rename to examples/streaming_http_test_server.rb index 4e49758..2ff9d96 100644 --- a/examples/sse_test_server.rb +++ b/examples/streaming_http_test_server.rb @@ -137,7 +137,7 @@ rack_app = Rack::Builder.new do end # Print usage instructions -puts "=== MCP SSE Test Server ===" +puts "=== MCP Streaming HTTP Test Server ===" puts "" puts "Starting server on http://localhost:9393" puts "" diff --git a/lib/mcp/server/transports/stdio_transport.rb b/lib/mcp/server/transports/stdio_transport.rb index 53b2724..702af08 100644 --- a/lib/mcp/server/transports/stdio_transport.rb +++ b/lib/mcp/server/transports/stdio_transport.rb @@ -8,11 +8,6 @@ module MCP module Transports class StdioTransport < Transport STATUS_INTERRUPTED = Signal.list["INT"] + 128 - class << self - def register - super("stdio", self) - end - end def initialize(server) @server = server diff --git a/lib/mcp/server/transports/streamable_http_transport.rb b/lib/mcp/server/transports/streamable_http_transport.rb index ee3244f..4c91f4c 100644 --- a/lib/mcp/server/transports/streamable_http_transport.rb +++ b/lib/mcp/server/transports/streamable_http_transport.rb @@ -8,12 +8,6 @@ module MCP class Server module Transports class StreamableHTTPTransport < Transport - class << self - def register - super("http", self) - end - end - def initialize(server) super # { session_id => { stream: stream_object } diff --git a/lib/mcp/transport.rb b/lib/mcp/transport.rb index d3226fc..a995c3a 100644 --- a/lib/mcp/transport.rb +++ b/lib/mcp/transport.rb @@ -2,19 +2,6 @@ module MCP class Transport - class << self - def register(name, transport_class) - @transports ||= {} - @transports[name] = transport_class - end - - def get(name) - raise "Transport #{name} not found" unless @transports - - @transports.fetch(name) { raise "Transport #{name} not found" } - end - end - # Initialize the transport with the server instance def initialize(server) @server = server diff --git a/test/mcp/server_notification_test.rb b/test/mcp/server_notification_test.rb index 4750738..af35936 100644 --- a/test/mcp/server_notification_test.rb +++ b/test/mcp/server_notification_test.rb @@ -126,11 +126,5 @@ module MCP assert_equal Methods::NOTIFICATIONS_PROMPTS_LIST_CHANGED, notifications[1][:method] assert_equal Methods::NOTIFICATIONS_RESOURCES_LIST_CHANGED, notifications[2][:method] end - - test "notification method constants are defined correctly" do - assert_equal "notifications/tools/list_changed", Methods::NOTIFICATIONS_TOOLS_LIST_CHANGED - assert_equal "notifications/prompts/list_changed", Methods::NOTIFICATIONS_PROMPTS_LIST_CHANGED - assert_equal "notifications/resources/list_changed", Methods::NOTIFICATIONS_RESOURCES_LIST_CHANGED - end end end diff --git a/test/mcp/transport_test.rb b/test/mcp/transport_test.rb index 297631a..dc94881 100644 --- a/test/mcp/transport_test.rb +++ b/test/mcp/transport_test.rb @@ -28,22 +28,6 @@ module MCP resources: [], ) @transport = TestTransport.new(@server) - # Clear transports registry before each test - Transport.instance_variable_set(:@transports, nil) - end - - teardown do - # Clear transports registry after each test - Transport.instance_variable_set(:@transports, nil) - end - - test "registers transport class" do - Transport.register("test", TestTransport) - assert_equal TestTransport, Transport.get("test") - end - - test "raises error for unknown transport" do - assert_raises(RuntimeError, "Transport unknown not found") { Transport.get("unknown") } end test "initializes with server instance" do