Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 9 additions & 9 deletions test/mcp/server_test.rb
Original file line number Diff line number Diff line change
Expand Up @@ -379,7 +379,7 @@ def call(message:, server_context: nil)
test "#tools_call_handler sets the tools/call handler" do
@server.tools_call_handler do |request|
tool_name = request[:name]
return Tool::Response.new("#{tool_name} called successfully").to_h
Tool::Response.new("#{tool_name} called successfully").to_h
end

request = {
Expand All @@ -391,7 +391,7 @@ def call(message:, server_context: nil)

response = @server.handle(request)
assert_equal({ content: "my_tool called successfully", isError: false }, response[:result])
assert_instrumentation_data({ method: "tools/call", tool_name: "my_tool" })
assert_instrumentation_data({ method: "tools/call" })
end

test "#handle prompts/list returns list of prompts" do
Expand Down Expand Up @@ -484,10 +484,10 @@ def call(message:, server_context: nil)
test "#prompts_get_handler sets the prompts/get handler" do
@server.prompts_get_handler do |request|
prompt_name = request[:name]
return Prompt::Result.new(
Prompt::Result.new(
description: prompt_name,
messages: [
Prompt::Message.new(role: "user", content: Content::Text.new(request[:arguments][:foo])),
Prompt::Message.new(role: "user", content: Content::Text.new(request[:arguments]["foo"])),
],
).to_h
end
Expand All @@ -501,10 +501,10 @@ def call(message:, server_context: nil)

response = @server.handle(request)
assert_equal(
{ description: "foo_bar_prompt", messages: [{ role: "user", content: { text: "bar" } }] },
{ description: "foo_bar_prompt", messages: [{ role: "user", content: { type: "text", text: "bar" } }] },
response[:result],
)
assert_instrumentation_data({ method: "prompts/get", prompt_name: "foo_bar_prompt" })
assert_instrumentation_data({ method: "prompts/get" })
end

test "#handle resources/list returns a list of resources" do
Expand Down Expand Up @@ -555,9 +555,9 @@ def call(message:, server_context: nil)

test "#resources_read_handler sets the resources/read handler" do
@server.resources_read_handler do |request|
return {
{
uri: request[:uri],
mime_type: "text/plain",
mimeType: "text/plain",
text: "Lorem ipsum dolor sit amet",
}
end
Expand All @@ -573,7 +573,7 @@ def call(message:, server_context: nil)

response = @server.handle(request)
assert_equal(
{ contents: [{ uri: "example.com/my_resource", mimeType: "text/plain", text: "Lorem ipsum dolor sit amet" }] },
{ contents: { uri: "example.com/my_resource", mimeType: "text/plain", text: "Lorem ipsum dolor sit amet" } },
response[:result],
)
end
Expand Down