Fix build error
Follow-up to #27. This PR fixes the following build error: ```console $ bundle exec rake /Users/koic/.rbenv/versions/3.5-dev/lib/ruby/3.5.0+1/bundled_gems.rb:59:in 'Kernel.require': cannot load such file -- mcp/transports/stdio (LoadError) Did you mean? mcp/transports/stdio_transport_test from /Users/koic/.rbenv/versions/3.5-dev/lib/ruby/3.5.0+1/bundled_gems.rb:59:in 'block (2 levels) in Kernel#replace_require' from /Users/koic/src/github.com/modelcontextprotocol/ruby-sdk/test/mcp/transports/stdio_transport_test.rb:4:in '<top (required)>' from /Users/koic/.rbenv/versions/3.5-dev/lib/ruby/3.5.0+1/bundled_gems.rb:59:in 'Kernel.require' from /Users/koic/.rbenv/versions/3.5-dev/lib/ruby/3.5.0+1/bundled_gems.rb:59:in 'block (2 levels) in Kernel#replace_require' from /Users/koic/.rbenv/versions/3.5-dev/lib/ruby/gems/3.5.0+1/gems/rake-13.3.0/lib/rake/rake_test_loader.rb:21:in 'block in <main>' from /Users/koic/.rbenv/versions/3.5-dev/lib/ruby/gems/3.5.0+1/gems/rake-13.3.0/lib/rake/rake_test_loader.rb:6:in 'Array#select' from /Users/koic/.rbenv/versions/3.5-dev/lib/ruby/gems/3.5.0+1/gems/rake-13.3.0/lib/rake/rake_test_loader.rb:6:in '<main>' rake aborted! Command failed with status (1) /Users/koic/.rbenv/versions/3.5-dev/bin/bundle:25:in 'Kernel#load' /Users/koic/.rbenv/versions/3.5-dev/bin/bundle:25:in '<main>' Tasks: TOP => default => test (See full trace by running task with --trace) ``` https://github.com/modelcontextprotocol/ruby-sdk/actions/runs/15428236624/job/43420346011 The following is stated in #27. > Breaking Changes > `MCP::Transports::StdioTransport` is now `MCP::Server::Transports::StdioTransport` What is shown here is the new name `MCP::Server::Transports::StdioTransport`, not `MCP::Server::Transports::Stdio`. Therefore, the file path has been updated from `lib/mcp/server/transports/stdio.rb` to `lib/mcp/server/transports/stdio_transport.rb` to match the module name. If `MCP::Server::Transports::Stdio` is preferred instead, please let me know and I will open a separate PR.
This commit is contained in:
parent
e901f94a3d
commit
70680fce88
6 changed files with 5 additions and 130 deletions
|
|
@ -86,7 +86,7 @@ If you want to build a local command-line application, you can use the stdio tra
|
|||
```ruby
|
||||
#!/usr/bin/env ruby
|
||||
require "mcp"
|
||||
require "mcp/server/transports/stdio"
|
||||
require "mcp/server/transports/stdio_transport"
|
||||
|
||||
# Create a simple tool
|
||||
class ExampleTool < MCP::Tool
|
||||
|
|
|
|||
|
|
@ -3,7 +3,7 @@
|
|||
|
||||
$LOAD_PATH.unshift(File.expand_path("../lib", __dir__))
|
||||
require "mcp"
|
||||
require "mcp/server/transports/stdio"
|
||||
require "mcp/server/transports/stdio_transport"
|
||||
|
||||
# Create a simple tool
|
||||
class ExampleTool < MCP::Tool
|
||||
|
|
|
|||
|
|
@ -13,7 +13,7 @@ require_relative "mcp/resource/contents"
|
|||
require_relative "mcp/resource/embedded"
|
||||
require_relative "mcp/resource_template"
|
||||
require_relative "mcp/server"
|
||||
require_relative "mcp/server/transports/stdio"
|
||||
require_relative "mcp/server/transports/stdio_transport"
|
||||
require_relative "mcp/string_utils"
|
||||
require_relative "mcp/tool"
|
||||
require_relative "mcp/tool/input_schema"
|
||||
|
|
|
|||
37
lib/mcp/server/transports/stdio_transport.rb
Normal file
37
lib/mcp/server/transports/stdio_transport.rb
Normal file
|
|
@ -0,0 +1,37 @@
|
|||
# frozen_string_literal: true
|
||||
|
||||
require_relative "../../transport"
|
||||
require "json"
|
||||
|
||||
module MCP
|
||||
class Server
|
||||
module Transports
|
||||
class StdioTransport < Transport
|
||||
def initialize(server)
|
||||
@server = server
|
||||
@open = false
|
||||
$stdin.set_encoding("UTF-8")
|
||||
$stdout.set_encoding("UTF-8")
|
||||
super
|
||||
end
|
||||
|
||||
def open
|
||||
@open = true
|
||||
while @open && (line = $stdin.gets)
|
||||
handle_json_request(line.strip)
|
||||
end
|
||||
end
|
||||
|
||||
def close
|
||||
@open = false
|
||||
end
|
||||
|
||||
def send_response(message)
|
||||
json_message = message.is_a?(String) ? message : JSON.generate(message)
|
||||
$stdout.puts(json_message)
|
||||
$stdout.flush
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
127
test/mcp/server/transports/stdio_transport_test.rb
Normal file
127
test/mcp/server/transports/stdio_transport_test.rb
Normal file
|
|
@ -0,0 +1,127 @@
|
|||
# frozen_string_literal: true
|
||||
|
||||
require "test_helper"
|
||||
require "mcp/server/transports/stdio_transport"
|
||||
require "json"
|
||||
|
||||
module MCP
|
||||
class Server
|
||||
module Transports
|
||||
class StdioTransportTest < ActiveSupport::TestCase
|
||||
include InstrumentationTestHelper
|
||||
|
||||
setup do
|
||||
configuration = MCP::Configuration.new
|
||||
configuration.instrumentation_callback = instrumentation_helper.callback
|
||||
@server = Server.new(name: "test_server", configuration: configuration)
|
||||
@transport = StdioTransport.new(@server)
|
||||
end
|
||||
|
||||
test "initializes with server and closed state" do
|
||||
server = @transport.instance_variable_get(:@server)
|
||||
assert_equal @server.object_id, server.object_id
|
||||
refute @transport.instance_variable_get(:@open)
|
||||
end
|
||||
|
||||
test "processes JSON-RPC requests from stdin and sends responses to stdout" do
|
||||
request = {
|
||||
jsonrpc: "2.0",
|
||||
method: "ping",
|
||||
id: "123",
|
||||
}
|
||||
input = StringIO.new(JSON.generate(request) + "\n")
|
||||
output = StringIO.new
|
||||
|
||||
original_stdin = $stdin
|
||||
original_stdout = $stdout
|
||||
|
||||
begin
|
||||
$stdin = input
|
||||
$stdout = output
|
||||
|
||||
thread = Thread.new { @transport.open }
|
||||
sleep(0.1)
|
||||
@transport.close
|
||||
thread.join
|
||||
|
||||
response = JSON.parse(output.string, symbolize_names: true)
|
||||
assert_equal("2.0", response[:jsonrpc])
|
||||
assert_equal("123", response[:id])
|
||||
assert_empty(response[:result])
|
||||
refute(@transport.instance_variable_get(:@open))
|
||||
ensure
|
||||
$stdin = original_stdin
|
||||
$stdout = original_stdout
|
||||
end
|
||||
end
|
||||
|
||||
test "sends string responses to stdout" do
|
||||
output = StringIO.new
|
||||
original_stdout = $stdout
|
||||
|
||||
begin
|
||||
$stdout = output
|
||||
@transport.send_response("test response")
|
||||
assert_equal("test response\n", output.string)
|
||||
ensure
|
||||
$stdout = original_stdout
|
||||
end
|
||||
end
|
||||
|
||||
test "sends JSON responses to stdout" do
|
||||
output = StringIO.new
|
||||
original_stdout = $stdout
|
||||
|
||||
begin
|
||||
$stdout = output
|
||||
response = { key: "value" }
|
||||
@transport.send_response(response)
|
||||
assert_equal(JSON.generate(response) + "\n", output.string)
|
||||
ensure
|
||||
$stdout = original_stdout
|
||||
end
|
||||
end
|
||||
|
||||
test "handles valid JSON-RPC requests" do
|
||||
request = {
|
||||
jsonrpc: "2.0",
|
||||
method: "ping",
|
||||
id: "123",
|
||||
}
|
||||
output = StringIO.new
|
||||
original_stdout = $stdout
|
||||
|
||||
begin
|
||||
$stdout = output
|
||||
@transport.send(:handle_request, JSON.generate(request))
|
||||
response = JSON.parse(output.string, symbolize_names: true)
|
||||
assert_equal("2.0", response[:jsonrpc])
|
||||
assert_nil(response[:id])
|
||||
assert_nil(response[:result])
|
||||
ensure
|
||||
$stdout = original_stdout
|
||||
end
|
||||
end
|
||||
|
||||
test "handles invalid JSON requests" do
|
||||
invalid_json = "invalid json"
|
||||
output = StringIO.new
|
||||
original_stdout = $stdout
|
||||
|
||||
begin
|
||||
$stdout = output
|
||||
@transport.send(:handle_request, invalid_json)
|
||||
response = JSON.parse(output.string, symbolize_names: true)
|
||||
assert_equal("2.0", response[:jsonrpc])
|
||||
assert_nil(response[:id])
|
||||
assert_equal(-32600, response[:error][:code])
|
||||
assert_equal("Invalid Request", response[:error][:message])
|
||||
assert_equal("Request must be an array or a hash", response[:error][:data])
|
||||
ensure
|
||||
$stdout = original_stdout
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
|
|
@ -1,125 +0,0 @@
|
|||
# frozen_string_literal: true
|
||||
|
||||
require "test_helper"
|
||||
require "mcp/transports/stdio"
|
||||
require "json"
|
||||
|
||||
module MCP
|
||||
module Transports
|
||||
class StdioTransportTest < ActiveSupport::TestCase
|
||||
include InstrumentationTestHelper
|
||||
|
||||
setup do
|
||||
configuration = MCP::Configuration.new
|
||||
configuration.instrumentation_callback = instrumentation_helper.callback
|
||||
@server = Server.new(name: "test_server", configuration: configuration)
|
||||
@transport = StdioTransport.new(@server)
|
||||
end
|
||||
|
||||
test "initializes with server and closed state" do
|
||||
server = @transport.instance_variable_get(:@server)
|
||||
assert_equal @server.object_id, server.object_id
|
||||
refute @transport.instance_variable_get(:@open)
|
||||
end
|
||||
|
||||
test "processes JSON-RPC requests from stdin and sends responses to stdout" do
|
||||
request = {
|
||||
jsonrpc: "2.0",
|
||||
method: "ping",
|
||||
id: "123",
|
||||
}
|
||||
input = StringIO.new(JSON.generate(request) + "\n")
|
||||
output = StringIO.new
|
||||
|
||||
original_stdin = $stdin
|
||||
original_stdout = $stdout
|
||||
|
||||
begin
|
||||
$stdin = input
|
||||
$stdout = output
|
||||
|
||||
thread = Thread.new { @transport.open }
|
||||
sleep(0.1)
|
||||
@transport.close
|
||||
thread.join
|
||||
|
||||
response = JSON.parse(output.string, symbolize_names: true)
|
||||
assert_equal("2.0", response[:jsonrpc])
|
||||
assert_equal("123", response[:id])
|
||||
assert_empty(response[:result])
|
||||
refute(@transport.instance_variable_get(:@open))
|
||||
ensure
|
||||
$stdin = original_stdin
|
||||
$stdout = original_stdout
|
||||
end
|
||||
end
|
||||
|
||||
test "sends string responses to stdout" do
|
||||
output = StringIO.new
|
||||
original_stdout = $stdout
|
||||
|
||||
begin
|
||||
$stdout = output
|
||||
@transport.send_response("test response")
|
||||
assert_equal("test response\n", output.string)
|
||||
ensure
|
||||
$stdout = original_stdout
|
||||
end
|
||||
end
|
||||
|
||||
test "sends JSON responses to stdout" do
|
||||
output = StringIO.new
|
||||
original_stdout = $stdout
|
||||
|
||||
begin
|
||||
$stdout = output
|
||||
response = { key: "value" }
|
||||
@transport.send_response(response)
|
||||
assert_equal(JSON.generate(response) + "\n", output.string)
|
||||
ensure
|
||||
$stdout = original_stdout
|
||||
end
|
||||
end
|
||||
|
||||
test "handles valid JSON-RPC requests" do
|
||||
request = {
|
||||
jsonrpc: "2.0",
|
||||
method: "ping",
|
||||
id: "123",
|
||||
}
|
||||
output = StringIO.new
|
||||
original_stdout = $stdout
|
||||
|
||||
begin
|
||||
$stdout = output
|
||||
@transport.send(:handle_request, JSON.generate(request))
|
||||
response = JSON.parse(output.string, symbolize_names: true)
|
||||
assert_equal("2.0", response[:jsonrpc])
|
||||
assert_nil(response[:id])
|
||||
assert_nil(response[:result])
|
||||
ensure
|
||||
$stdout = original_stdout
|
||||
end
|
||||
end
|
||||
|
||||
test "handles invalid JSON requests" do
|
||||
invalid_json = "invalid json"
|
||||
output = StringIO.new
|
||||
original_stdout = $stdout
|
||||
|
||||
begin
|
||||
$stdout = output
|
||||
@transport.send(:handle_request, invalid_json)
|
||||
response = JSON.parse(output.string, symbolize_names: true)
|
||||
assert_equal("2.0", response[:jsonrpc])
|
||||
assert_nil(response[:id])
|
||||
assert_equal(-32600, response[:error][:code])
|
||||
assert_equal("Invalid Request", response[:error][:message])
|
||||
assert_equal("Request must be an array or a hash", response[:error][:data])
|
||||
ensure
|
||||
$stdout = original_stdout
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
Loading…
Reference in a new issue