Adopt Schema component in server handlers

2025-08-03 22:06:46 +02:00 · 2025-08-03 22:06:46 +02:00 · 09707b4798
commit 09707b4798
parent 6025f890f9
1 changed files with 1516 additions and 2424 deletions
--- a/tests/JsonRpc/HandlerTest.php
+++ b/tests/JsonRpc/HandlerTest.php
@ -0,0 +1,84 @@
+<?php
+
+/*
+ * This file is part of the official PHP MCP SDK.
+ *
+ * A collaboration between Symfony and the PHP Foundation.
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Mcp\Tests\JsonRpc;
+
+use Mcp\JsonRpc\Handler;
+use Mcp\JsonRpc\MessageFactory;
+use Mcp\Schema\JsonRpc\Response;
+use Mcp\Server\MethodHandlerInterface;
+use PHPUnit\Framework\Attributes\TestDox;
+use PHPUnit\Framework\TestCase;
+
+class HandlerTest extends TestCase
+{
+    #[TestDox('Make sure a single notification can be handled by multiple handlers.')]
+    public function testHandleMultipleNotifications()
+    {
+        $handlerA = $this->getMockBuilder(MethodHandlerInterface::class)
+            ->disableOriginalConstructor()
+            ->onlyMethods(['supports', 'handle'])
+            ->getMock();
+        $handlerA->method('supports')->willReturn(true);
+        $handlerA->expects($this->once())->method('handle');
+
+        $handlerB = $this->getMockBuilder(MethodHandlerInterface::class)
+            ->disableOriginalConstructor()
+            ->onlyMethods(['supports', 'handle'])
+            ->getMock();
+        $handlerB->method('supports')->willReturn(false);
+        $handlerB->expects($this->never())->method('handle');
+
+        $handlerC = $this->getMockBuilder(MethodHandlerInterface::class)
+            ->disableOriginalConstructor()
+            ->onlyMethods(['supports', 'handle'])
+            ->getMock();
+        $handlerC->method('supports')->willReturn(true);
+        $handlerC->expects($this->once())->method('handle');
+
+        $jsonRpc = new Handler(MessageFactory::make(), [$handlerA, $handlerB, $handlerC]);
+        $result = $jsonRpc->process(
+            '{"jsonrpc": "2.0", "method": "notifications/initialized"}'
+        );
+        iterator_to_array($result);
+    }
+
+    #[TestDox('Make sure a single request can NOT be handled by multiple handlers.')]
+    public function testHandleMultipleRequests()
+    {
+        $handlerA = $this->getMockBuilder(MethodHandlerInterface::class)
+            ->disableOriginalConstructor()
+            ->onlyMethods(['supports', 'handle'])
+            ->getMock();
+        $handlerA->method('supports')->willReturn(true);
+        $handlerA->expects($this->once())->method('handle')->willReturn(new Response(1, ['result' => 'success']));
+
+        $handlerB = $this->getMockBuilder(MethodHandlerInterface::class)
+            ->disableOriginalConstructor()
+            ->onlyMethods(['supports', 'handle'])
+            ->getMock();
+        $handlerB->method('supports')->willReturn(false);
+        $handlerB->expects($this->never())->method('handle');
+
+        $handlerC = $this->getMockBuilder(MethodHandlerInterface::class)
+            ->disableOriginalConstructor()
+            ->onlyMethods(['supports', 'handle'])
+            ->getMock();
+        $handlerC->method('supports')->willReturn(true);
+        $handlerC->expects($this->never())->method('handle');
+
+        $jsonRpc = new Handler(MessageFactory::make(), [$handlerA, $handlerB, $handlerC]);
+        $result = $jsonRpc->process(
+            '{"jsonrpc": "2.0", "id": 1, "method": "tools/list"}'
+        );
+        iterator_to_array($result);
+    }
+}