tests

Author: sambostock

lib/mcp.rb

@@ -15,7 +15,6 @@
 require_relative "mcp/resource_template"
 require_relative "mcp/server"
 require_relative "mcp/server/transports/stdio_transport"
-require_relative "mcp/string_utils"
 require_relative "mcp/tool"
 require_relative "mcp/tool/input_schema"
 require_relative "mcp/tool/response"

lib/mcp/prompt.rb

@@ -2,7 +2,7 @@
 # frozen_string_literal: true
 
 module MCP
-  module Prompt
+  class Prompt
     attr_reader :name, :description, :arguments, :to_h
 
     class << self

lib/mcp/prompt/argument.rb

@@ -2,7 +2,7 @@
 # frozen_string_literal: true
 
 module MCP
-  module Prompt
+  class Prompt
     class Argument
       attr_reader :name, :description, :required, :to_h
 

lib/mcp/prompt/message.rb

@@ -2,7 +2,7 @@
 # frozen_string_literal: true
 
 module MCP
-  module Prompt
+  class Prompt
     class Message
       attr_reader :role, :content
 

lib/mcp/prompt/result.rb

@@ -2,7 +2,7 @@
 # frozen_string_literal: true
 
 module MCP
-  module Prompt
+  class Prompt
     class Result
       attr_reader :description, :messages
 

lib/mcp/tool/annotations.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
 module MCP
-  module Tool
+  class Tool
     class Annotations
       attr_reader :title, :read_only_hint, :destructive_hint, :idempotent_hint, :open_world_hint, :to_h
 

lib/mcp/tool/input_schema.rb

@@ -1,13 +1,13 @@
 # frozen_string_literal: true
 
 module MCP
-  module Tool
+  class Tool
     class InputSchema
       attr_reader :properties, :required, :to_h
 
       def initialize(properties: {}, required: [])
-        @properties = properties
-        @required = required
+        @properties = properties.transform_keys(&:to_sym)
+        @required = required.map(&:to_sym)
 
         @to_h = {
           type: "object",

lib/mcp/tool/response.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
 module MCP
-  module Tool
+  class Tool
     class Response
       attr_reader :content, :is_error
 

test/mcp/tool/input_schema_test.rb

@@ -3,13 +3,18 @@
 require "test_helper"
 
 module MCP
-  module Tool
+  class Tool
     class InputSchemaTest < ActiveSupport::TestCase
       test "required arguments are converted to symbols" do
         input_schema = InputSchema.new(properties: { message: { type: "string" } }, required: ["message"])
         assert_equal [:message], input_schema.required
       end
 
+      test "properties keys   are converted to symbols" do
+        input_schema = InputSchema.new(properties: { "message" => { type: "string" } }, required: [:message])
+        assert_equal({ message: { type: "string" } }, input_schema.properties)
+      end
+
       test "to_h returns a hash representation of the input schema" do
         input_schema = InputSchema.new(properties: { message: { type: "string" } }, required: [:message])
         assert_equal(

test/mcp/tool_test.rb

@@ -21,7 +21,7 @@ class ToolTest < ActiveSupport::TestCase
     end
 
     test "#to_h returns a hash with name, description, and inputSchema" do
-      tool = Tool.define(name: "mock_tool", description: "a mock tool for testing")
+      tool = Tool.define(name: "mock_tool", description: "a mock tool for testing") {}
       assert_equal tool.to_h, { name: "mock_tool", description: "a mock tool for testing" }
     end