ruby-sdk/test/mcp/annotations_test.rb
Koichi ITO a85fa028f7 Extract MCP::Annotations into a dedicated file
This PR moves `MCP::Annotations` from lib/mcp.rb to lib/mcp/annotations.rb for better organization,
following the pattern used for other classes such as `Tool::Annotations`.

This PR also adds the missing `last_modified` attribute to `MCP::Annotations`:
https://github.com/modelcontextprotocol/modelcontextprotocol/blob/main/schema/2025-11-25/schema.ts#L1701-L1736

It also adds a `to_h` method for serialization and includes comprehensive test coverage.

NOTE: This code is unused within the ruby-sdk repository, but it is kept for compatibility, as it may be used by users.
2026-02-03 15:38:04 +09:00

59 lines
1.9 KiB
Ruby

# frozen_string_literal: true
require "test_helper"
module MCP
class AnnotationsTest < ActiveSupport::TestCase
def test_initialization
annotations = Annotations.new(audience: ["developers"], priority: 0.8)
assert_equal(["developers"], annotations.audience)
assert_equal(0.8, annotations.priority)
assert_nil(annotations.last_modified)
assert_equal({ audience: ["developers"], priority: 0.8 }, annotations.to_h)
end
def test_initialization_with_all_attributes
timestamp = Time.utc(2025, 1, 12, 15, 0, 58).iso8601
annotations = Annotations.new(audience: ["developers"], priority: 0.8, last_modified: timestamp)
assert_equal(["developers"], annotations.audience)
assert_equal(0.8, annotations.priority)
assert_equal(timestamp, annotations.last_modified)
assert_equal({ audience: ["developers"], priority: 0.8, lastModified: timestamp }, annotations.to_h)
end
def test_initialization_by_default
annotations = Annotations.new
assert_nil(annotations.audience)
assert_nil(annotations.priority)
assert_nil(annotations.last_modified)
assert_empty(annotations.to_h)
end
def test_initialization_with_partial_attributes
annotations = Annotations.new(audience: ["developers"])
assert_equal(["developers"], annotations.audience)
assert_nil(annotations.priority)
assert_nil(annotations.last_modified)
assert_equal({ audience: ["developers"] }, annotations.to_h)
end
def test_initialization_with_last_modified_only
timestamp = Time.utc(2025, 1, 12, 15, 0, 58).iso8601
annotations = Annotations.new(last_modified: timestamp)
assert_nil(annotations.audience)
assert_nil(annotations.priority)
assert_equal(timestamp, annotations.last_modified)
assert_equal({ lastModified: timestamp }, annotations.to_h)
end
end
end