ba7a08aa04
The MCP spec defines an embedded resource content type, but the SDK had no corresponding class. Add `Content::EmbeddedResource` that wraps a resource object and serializes it via `to_h`. https://modelcontextprotocol.io/specification/2025-11-25/server/tools#embedded-resources
97 lines
2.8 KiB
Ruby
97 lines
2.8 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require "test_helper"
|
|
|
|
module MCP
|
|
module Content
|
|
class ImageTest < ActiveSupport::TestCase
|
|
test "#to_h returns mimeType (camelCase) per MCP spec" do
|
|
image = Image.new("base64data", "image/png")
|
|
result = image.to_h
|
|
|
|
assert_equal "image/png", result[:mimeType]
|
|
refute result.key?(:mime_type), "Expected camelCase mimeType, got snake_case mime_type"
|
|
assert_equal "image", result[:type]
|
|
assert_equal "base64data", result[:data]
|
|
end
|
|
|
|
test "#to_h with annotations" do
|
|
image = Image.new("base64data", "image/png", annotations: { role: "thumbnail" })
|
|
result = image.to_h
|
|
|
|
assert_equal({ role: "thumbnail" }, result[:annotations])
|
|
end
|
|
|
|
test "#to_h without annotations omits the key" do
|
|
image = Image.new("base64data", "image/png")
|
|
result = image.to_h
|
|
|
|
refute result.key?(:annotations)
|
|
end
|
|
end
|
|
|
|
class AudioTest < ActiveSupport::TestCase
|
|
test "#to_h returns correct format per MCP spec" do
|
|
audio = Audio.new("base64data", "audio/wav")
|
|
result = audio.to_h
|
|
|
|
assert_equal "audio", result[:type]
|
|
assert_equal "base64data", result[:data]
|
|
assert_equal "audio/wav", result[:mimeType]
|
|
end
|
|
|
|
test "#to_h with annotations" do
|
|
audio = Audio.new("base64data", "audio/wav", annotations: { role: "recording" })
|
|
result = audio.to_h
|
|
|
|
assert_equal({ role: "recording" }, result[:annotations])
|
|
end
|
|
|
|
test "#to_h without annotations omits the key" do
|
|
audio = Audio.new("base64data", "audio/wav")
|
|
result = audio.to_h
|
|
|
|
refute result.key?(:annotations)
|
|
end
|
|
end
|
|
|
|
class EmbeddedResourceTest < ActiveSupport::TestCase
|
|
test "#to_h returns correct format per MCP spec" do
|
|
resource = Object.new
|
|
def resource.to_h
|
|
{ uri: "test://example", mimeType: "text/plain", text: "content" }
|
|
end
|
|
|
|
embedded = EmbeddedResource.new(resource)
|
|
result = embedded.to_h
|
|
|
|
assert_equal "resource", result[:type]
|
|
assert_equal({ uri: "test://example", mimeType: "text/plain", text: "content" }, result[:resource])
|
|
end
|
|
|
|
test "#to_h with annotations" do
|
|
resource = Object.new
|
|
def resource.to_h
|
|
{ uri: "test://x" }
|
|
end
|
|
|
|
embedded = EmbeddedResource.new(resource, annotations: { role: "data" })
|
|
result = embedded.to_h
|
|
|
|
assert_equal({ role: "data" }, result[:annotations])
|
|
end
|
|
|
|
test "#to_h without annotations omits the key" do
|
|
resource = Object.new
|
|
def resource.to_h
|
|
{ uri: "test://x" }
|
|
end
|
|
|
|
embedded = EmbeddedResource.new(resource)
|
|
result = embedded.to_h
|
|
|
|
refute result.key?(:annotations)
|
|
end
|
|
end
|
|
end
|
|
end
|