|
| 1 | +# frozen_string_literal: true |
| 2 | + |
| 3 | +require "test_helper" |
| 4 | + |
| 5 | +module MCP |
| 6 | + class ServerSessionPingTest < ActiveSupport::TestCase |
| 7 | + class MockTransport < Transport |
| 8 | + attr_reader :requests |
| 9 | + attr_accessor :response, :error_to_raise |
| 10 | + |
| 11 | + def initialize(server) |
| 12 | + super |
| 13 | + @requests = [] |
| 14 | + @response = {} |
| 15 | + @error_to_raise = nil |
| 16 | + end |
| 17 | + |
| 18 | + # Matches real transports: returns the `result` body directly (not the full envelope). |
| 19 | + def send_request(method, params = nil, session_id: nil, related_request_id: nil) |
| 20 | + @requests << { method: method, params: params, session_id: session_id, related_request_id: related_request_id } |
| 21 | + raise @error_to_raise if @error_to_raise |
| 22 | + |
| 23 | + @response |
| 24 | + end |
| 25 | + |
| 26 | + def send_response(response); end |
| 27 | + def send_notification(method, params = nil); end |
| 28 | + def open; end |
| 29 | + def close; end |
| 30 | + def handle_request(request); end |
| 31 | + end |
| 32 | + |
| 33 | + setup do |
| 34 | + @server = Server.new(name: "test_server", version: "1.0.0") |
| 35 | + @mock_transport = MockTransport.new(@server) |
| 36 | + @server.transport = @mock_transport |
| 37 | + end |
| 38 | + |
| 39 | + test "#ping sends request through transport and returns the result hash" do |
| 40 | + session = ServerSession.new(server: @server, transport: @mock_transport) |
| 41 | + |
| 42 | + result = session.ping |
| 43 | + |
| 44 | + assert_equal({}, result) |
| 45 | + assert_equal(1, @mock_transport.requests.size) |
| 46 | + |
| 47 | + request = @mock_transport.requests.first |
| 48 | + assert_equal(Methods::PING, request[:method]) |
| 49 | + assert_nil(request[:params]) |
| 50 | + end |
| 51 | + |
| 52 | + test "#ping passes related_request_id through when session_id is set" do |
| 53 | + session = ServerSession.new(server: @server, transport: @mock_transport, session_id: "session-1") |
| 54 | + |
| 55 | + session.ping(related_request_id: "req-abc") |
| 56 | + |
| 57 | + request = @mock_transport.requests.first |
| 58 | + assert_equal("session-1", request[:session_id]) |
| 59 | + assert_equal("req-abc", request[:related_request_id]) |
| 60 | + end |
| 61 | + |
| 62 | + test "#ping raises ValidationError when result is nil" do |
| 63 | + @mock_transport.response = nil |
| 64 | + session = ServerSession.new(server: @server, transport: @mock_transport) |
| 65 | + |
| 66 | + error = assert_raises(Server::ValidationError) { session.ping } |
| 67 | + assert_equal("Response validation failed: invalid `result`", error.message) |
| 68 | + end |
| 69 | + |
| 70 | + test "#ping raises ValidationError when result is the wrong type" do |
| 71 | + @mock_transport.response = "ok" |
| 72 | + session = ServerSession.new(server: @server, transport: @mock_transport) |
| 73 | + |
| 74 | + error = assert_raises(Server::ValidationError) { session.ping } |
| 75 | + assert_equal("Response validation failed: invalid `result`", error.message) |
| 76 | + end |
| 77 | + |
| 78 | + test "#ping propagates transport-level errors" do |
| 79 | + @mock_transport.error_to_raise = StandardError.new("read timeout") |
| 80 | + session = ServerSession.new(server: @server, transport: @mock_transport) |
| 81 | + |
| 82 | + error = assert_raises(StandardError) { session.ping } |
| 83 | + assert_equal("read timeout", error.message) |
| 84 | + end |
| 85 | + |
| 86 | + test "#ping succeeds without a client capability declaration" do |
| 87 | + session = ServerSession.new(server: @server, transport: @mock_transport) |
| 88 | + assert_nil(session.client_capabilities) |
| 89 | + |
| 90 | + assert_equal({}, session.ping) |
| 91 | + end |
| 92 | + end |
| 93 | +end |
0 commit comments