Test README prompt definition examples

Author: sambostock

README.md

@@ -349,12 +349,13 @@ The `MCP::Prompt` class provides two ways to create prompts:
 
 1. As a class definition with metadata:
 
+<!-- SNIPPET ID: prompt_class_definition -->
 ```ruby
 class MyPrompt < MCP::Prompt
   prompt_name "my_prompt"  # Optional - defaults to underscored class name
   description "This prompt performs specific functionality..."
   arguments [
-    Prompt::Argument.new(
+    MCP::Prompt::Argument.new(
       name: "message",
       description: "Input message",
       required: true
@@ -363,16 +364,16 @@ class MyPrompt < MCP::Prompt
 
   class << self
     def template(args, server_context:)
-      Prompt::Result.new(
+      MCP::Prompt::Result.new(
         description: "Response description",
         messages: [
-          Prompt::Message.new(
+          MCP::Prompt::Message.new(
             role: "user",
-            content: Content::Text.new("User message")
+            content: MCP::Content::Text.new("User message")
           ),
-          Prompt::Message.new(
+          MCP::Prompt::Message.new(
             role: "assistant",
-            content: Content::Text.new(args["message"])
+            content: MCP::Content::Text.new(args[:message])
           )
         ]
       )
@@ -385,28 +386,29 @@ prompt = MyPrompt
 
 2. Using the `MCP::Prompt.define` method:
 
+<!-- SNIPPET ID: prompt_definition_with_block -->
 ```ruby
 prompt = MCP::Prompt.define(
   name: "my_prompt",
   description: "This prompt performs specific functionality...",
   arguments: [
-    Prompt::Argument.new(
+    MCP::Prompt::Argument.new(
       name: "message",
       description: "Input message",
       required: true
     )
   ]
 ) do |args, server_context:|
-  Prompt::Result.new(
+  MCP::Prompt::Result.new(
     description: "Response description",
     messages: [
-      Prompt::Message.new(
+      MCP::Prompt::Message.new(
         role: "user",
-        content: Content::Text.new("User message")
+        content: MCP::Content::Text.new("User message")
       ),
-      Prompt::Message.new(
+      MCP::Prompt::Message.new(
         role: "assistant",
-        content: Content::Text.new(args["message"])
+        content: MCP::Content::Text.new(args[:message])
       )
     ]
   )

test/fixtures/files/code_snippet_wrappers/readme/prompt_class_definition.rb

@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+
+require "mcp"
+
+require_relative "code_snippet"
+
+b = binding
+eval(File.read("code_snippet.rb"), b) # rubocop:disable Security/Eval -- We need to run the snippet in this context to extract the prompt
+prompt = b.local_variable_get(:prompt)
+
+server = MCP::Server.new(prompts: [prompt])
+
+[
+  { jsonrpc: "2.0", id: "1", method: "prompts/list" },
+  { jsonrpc: "2.0", id: "2", method: "prompts/get", params: { name: "my_prompt", arguments: { message: "Test message" } } },
+].each { |request| puts server.handle_json(request.to_json) }

test/fixtures/files/code_snippet_wrappers/readme/prompt_definition_with_block.rb

@@ -0,0 +1 @@
+prompt_class_definition.rb

test/integration/readme_code_snippets_test.rb

@@ -155,6 +155,72 @@ class ReadmeCodeSnippetsTest < ActiveSupport::TestCase
     )
   end
 
+  test "Prompts examples work exactly as documented in README" do
+    assert_json_lines(
+      [
+        {
+          jsonrpc: "2.0",
+          id: "1",
+          result: {
+            prompts: [
+              {
+                name: "my_prompt",
+                description: "This prompt performs specific functionality...",
+                arguments: [
+                  { name: "message", description: "Input message", required: true },
+                ],
+              },
+            ],
+          },
+        },
+        {
+          jsonrpc: "2.0",
+          id: "2",
+          result: {
+            description: "Response description",
+            messages: [
+              { role: "user",      content: { type: "text", text: "User message" } },
+              { role: "assistant", content: { type: "text", text: "Test message" } },
+            ],
+          },
+        },
+      ],
+      run_code_snippet("prompt_class_definition"),
+    )
+
+    assert_json_lines(
+      [
+        {
+          jsonrpc: "2.0",
+          id: "1",
+          result: {
+            prompts: [
+              {
+                name: "my_prompt",
+                description: "This prompt performs specific functionality...",
+                arguments: [
+                  { name: "message", description: "Input message", required: true },
+                ],
+              },
+            ],
+          },
+        },
+        {
+          jsonrpc: "2.0",
+          id: "2",
+          result: {
+            description: "Response description",
+            messages: [
+              { role: "user",      content: { type: "text", text: "User message" } },
+              { role: "assistant", content: { type: "text", text: "Test message" } },
+            ],
+          },
+        },
+      ],
+      run_code_snippet("prompt_definition_with_block"),
+    )
+  end
+
   private
 
   def assert_json_lines(expected, actual, message = "Expected the given JSON lines")