Adopt Schema component in server handlers
This commit is contained in:
parent
6025f890f9
commit
09707b4798
3 changed files with 1516 additions and 2424 deletions
|
|
@ -13,43 +13,50 @@ namespace Mcp\Tests\Server\RequestHandler;
|
|||
|
||||
use Mcp\Capability\Prompt\MetadataInterface;
|
||||
use Mcp\Capability\PromptChain;
|
||||
use Mcp\Message\Request;
|
||||
use Mcp\Server\RequestHandler\PromptListHandler;
|
||||
use PHPUnit\Framework\Attributes\CoversClass;
|
||||
use PHPUnit\Framework\Attributes\Small;
|
||||
use Mcp\Schema\Request\ListPromptsRequest;
|
||||
use Mcp\Schema\Result\ListPromptsResult;
|
||||
use Mcp\Server\RequestHandler\ListPromptsHandler;
|
||||
use Nyholm\NSA;
|
||||
use PHPUnit\Framework\TestCase;
|
||||
|
||||
#[Small]
|
||||
#[CoversClass(PromptListHandler::class)]
|
||||
class PromptListHandlerTest extends TestCase
|
||||
{
|
||||
public function testHandleEmpty(): void
|
||||
public function testHandleEmpty()
|
||||
{
|
||||
$handler = new PromptListHandler(new PromptChain([]));
|
||||
$message = new Request(1, 'prompts/list', []);
|
||||
$response = $handler->createResponse($message);
|
||||
$this->assertEquals(1, $response->id);
|
||||
$this->assertEquals(['prompts' => []], $response->result);
|
||||
$handler = new ListPromptsHandler(new PromptChain([]));
|
||||
$request = new ListPromptsRequest();
|
||||
NSA::setProperty($request, 'id', 1);
|
||||
$response = $handler->handle($request);
|
||||
|
||||
$this->assertInstanceOf(ListPromptsResult::class, $response->result);
|
||||
$this->assertEquals(1, $response->getId());
|
||||
$this->assertEquals([], $response->result->prompts);
|
||||
}
|
||||
|
||||
public function testHandleReturnAll(): void
|
||||
public function testHandleReturnAll()
|
||||
{
|
||||
$item = self::createMetadataItem();
|
||||
$handler = new PromptListHandler(new PromptChain([$item]));
|
||||
$message = new Request(1, 'prompts/list', []);
|
||||
$response = $handler->createResponse($message);
|
||||
$this->assertCount(1, $response->result['prompts']);
|
||||
$this->assertArrayNotHasKey('nextCursor', $response->result);
|
||||
$handler = new ListPromptsHandler(new PromptChain([$item]));
|
||||
$request = new ListPromptsRequest();
|
||||
NSA::setProperty($request, 'id', 1);
|
||||
$response = $handler->handle($request);
|
||||
|
||||
$this->assertInstanceOf(ListPromptsResult::class, $response->result);
|
||||
$this->assertCount(1, $response->result->prompts);
|
||||
$this->assertNull($response->result->nextCursor);
|
||||
}
|
||||
|
||||
public function testHandlePagination(): void
|
||||
public function testHandlePagination()
|
||||
{
|
||||
$item = self::createMetadataItem();
|
||||
$handler = new PromptListHandler(new PromptChain([$item, $item]), 2);
|
||||
$message = new Request(1, 'prompts/list', []);
|
||||
$response = $handler->createResponse($message);
|
||||
$this->assertCount(2, $response->result['prompts']);
|
||||
$this->assertArrayHasKey('nextCursor', $response->result);
|
||||
$handler = new ListPromptsHandler(new PromptChain([$item, $item]), 2);
|
||||
$request = new ListPromptsRequest();
|
||||
NSA::setProperty($request, 'id', 1);
|
||||
$response = $handler->handle($request);
|
||||
|
||||
$this->assertInstanceOf(ListPromptsResult::class, $response->result);
|
||||
$this->assertCount(2, $response->result->prompts);
|
||||
$this->assertNotNull($response->result->nextCursor);
|
||||
}
|
||||
|
||||
private static function createMetadataItem(): MetadataInterface
|
||||
|
|
|
|||
|
|
@ -13,18 +13,16 @@ namespace Mcp\Tests\Server\RequestHandler;
|
|||
|
||||
use Mcp\Capability\Resource\CollectionInterface;
|
||||
use Mcp\Capability\Resource\MetadataInterface;
|
||||
use Mcp\Message\Request;
|
||||
use Mcp\Server\RequestHandler\ResourceListHandler;
|
||||
use PHPUnit\Framework\Attributes\CoversClass;
|
||||
use Mcp\Schema\Request\ListResourcesRequest;
|
||||
use Mcp\Schema\Result\ListResourcesResult;
|
||||
use Mcp\Server\RequestHandler\ListResourcesHandler;
|
||||
use Nyholm\NSA;
|
||||
use PHPUnit\Framework\Attributes\DataProvider;
|
||||
use PHPUnit\Framework\Attributes\Small;
|
||||
use PHPUnit\Framework\TestCase;
|
||||
|
||||
#[Small]
|
||||
#[CoversClass(ResourceListHandler::class)]
|
||||
class ResourceListHandlerTest extends TestCase
|
||||
{
|
||||
public function testHandleEmpty(): void
|
||||
public function testHandleEmpty()
|
||||
{
|
||||
$collection = $this->getMockBuilder(CollectionInterface::class)
|
||||
->disableOriginalConstructor()
|
||||
|
|
@ -32,29 +30,36 @@ class ResourceListHandlerTest extends TestCase
|
|||
->getMock();
|
||||
$collection->expects($this->once())->method('getMetadata')->willReturn([]);
|
||||
|
||||
$handler = new ResourceListHandler($collection);
|
||||
$message = new Request(1, 'resources/list', []);
|
||||
$response = $handler->createResponse($message);
|
||||
$handler = new ListResourcesHandler($collection);
|
||||
$request = new ListResourcesRequest();
|
||||
NSA::setProperty($request, 'id', 1);
|
||||
$response = $handler->handle($request);
|
||||
|
||||
$this->assertInstanceOf(ListResourcesResult::class, $response->result);
|
||||
$this->assertEquals(1, $response->id);
|
||||
$this->assertEquals(['resources' => []], $response->result);
|
||||
$this->assertEquals([], $response->result->resources);
|
||||
}
|
||||
|
||||
/**
|
||||
* @param iterable<MetadataInterface> $metadataList
|
||||
*/
|
||||
#[DataProvider('metadataProvider')]
|
||||
public function testHandleReturnAll(iterable $metadataList): void
|
||||
public function testHandleReturnAll(iterable $metadataList)
|
||||
{
|
||||
$collection = $this->getMockBuilder(CollectionInterface::class)
|
||||
->disableOriginalConstructor()
|
||||
->onlyMethods(['getMetadata'])
|
||||
->getMock();
|
||||
$collection->expects($this->once())->method('getMetadata')->willReturn($metadataList);
|
||||
$handler = new ResourceListHandler($collection);
|
||||
$message = new Request(1, 'resources/list', []);
|
||||
$response = $handler->createResponse($message);
|
||||
$this->assertCount(1, $response->result['resources']);
|
||||
$this->assertArrayNotHasKey('nextCursor', $response->result);
|
||||
|
||||
$handler = new ListResourcesHandler($collection);
|
||||
$request = new ListResourcesRequest();
|
||||
NSA::setProperty($request, 'id', 1);
|
||||
$response = $handler->handle($request);
|
||||
|
||||
$this->assertInstanceOf(ListResourcesResult::class, $response->result);
|
||||
$this->assertCount(1, $response->result->resources);
|
||||
$this->assertNull($response->result->nextCursor);
|
||||
}
|
||||
|
||||
/**
|
||||
|
|
@ -70,7 +75,7 @@ class ResourceListHandlerTest extends TestCase
|
|||
];
|
||||
}
|
||||
|
||||
public function testHandlePagination(): void
|
||||
public function testHandlePagination()
|
||||
{
|
||||
$item = self::createMetadataItem();
|
||||
$collection = $this->getMockBuilder(CollectionInterface::class)
|
||||
|
|
@ -78,11 +83,15 @@ class ResourceListHandlerTest extends TestCase
|
|||
->onlyMethods(['getMetadata'])
|
||||
->getMock();
|
||||
$collection->expects($this->once())->method('getMetadata')->willReturn([$item, $item]);
|
||||
$handler = new ResourceListHandler($collection, 2);
|
||||
$message = new Request(1, 'resources/list', []);
|
||||
$response = $handler->createResponse($message);
|
||||
$this->assertCount(2, $response->result['resources']);
|
||||
$this->assertArrayHasKey('nextCursor', $response->result);
|
||||
|
||||
$handler = new ListResourcesHandler($collection, 2);
|
||||
$request = new ListResourcesRequest();
|
||||
NSA::setProperty($request, 'id', 1);
|
||||
$response = $handler->handle($request);
|
||||
|
||||
$this->assertInstanceOf(ListResourcesResult::class, $response->result);
|
||||
$this->assertCount(2, $response->result->resources);
|
||||
$this->assertNotNull($response->result->nextCursor);
|
||||
}
|
||||
|
||||
private static function createMetadataItem(): MetadataInterface
|
||||
|
|
@ -95,7 +104,7 @@ class ResourceListHandlerTest extends TestCase
|
|||
|
||||
public function getName(): string
|
||||
{
|
||||
return 'src/SomeFile.php';
|
||||
return 'SomeFile';
|
||||
}
|
||||
|
||||
public function getDescription(): string
|
||||
|
|
|
|||
|
|
@ -13,19 +13,16 @@ namespace Mcp\Tests\Server\RequestHandler;
|
|||
|
||||
use Mcp\Capability\Tool\CollectionInterface;
|
||||
use Mcp\Capability\Tool\MetadataInterface;
|
||||
use Mcp\Message\Request;
|
||||
use Mcp\Server\RequestHandler\ToolListHandler;
|
||||
use PHPUnit\Framework\Attributes\CoversClass;
|
||||
use Mcp\Schema\Request\ListToolsRequest;
|
||||
use Mcp\Schema\Result\ListToolsResult;
|
||||
use Mcp\Server\RequestHandler\ListToolsHandler;
|
||||
use Nyholm\NSA;
|
||||
use PHPUnit\Framework\Attributes\DataProvider;
|
||||
use PHPUnit\Framework\Attributes\Small;
|
||||
use PHPUnit\Framework\Attributes\Test;
|
||||
use PHPUnit\Framework\TestCase;
|
||||
|
||||
#[Small]
|
||||
#[CoversClass(ToolListHandler::class)]
|
||||
class ToolListHandlerTest extends TestCase
|
||||
{
|
||||
public function testHandleEmpty(): void
|
||||
public function testHandleEmpty()
|
||||
{
|
||||
$collection = $this->getMockBuilder(CollectionInterface::class)
|
||||
->disableOriginalConstructor()
|
||||
|
|
@ -33,29 +30,35 @@ class ToolListHandlerTest extends TestCase
|
|||
->getMock();
|
||||
$collection->expects($this->once())->method('getMetadata')->willReturn([]);
|
||||
|
||||
$handler = new ToolListHandler($collection);
|
||||
$message = new Request(1, 'tools/list', []);
|
||||
$response = $handler->createResponse($message);
|
||||
$this->assertEquals(1, $response->id);
|
||||
$this->assertEquals(['tools' => []], $response->result);
|
||||
$handler = new ListToolsHandler($collection);
|
||||
$request = new ListToolsRequest();
|
||||
NSA::setProperty($request, 'id', 1);
|
||||
$response = $handler->handle($request);
|
||||
|
||||
$this->assertInstanceOf(ListToolsResult::class, $response->result);
|
||||
$this->assertSame(1, $response->id);
|
||||
$this->assertSame([], $response->result->tools);
|
||||
}
|
||||
|
||||
/**
|
||||
* @param iterable<MetadataInterface> $metadataList
|
||||
*/
|
||||
#[DataProvider('metadataProvider')]
|
||||
public function testHandleReturnAll(iterable $metadataList): void
|
||||
public function testHandleReturnAll(iterable $metadataList)
|
||||
{
|
||||
$collection = $this->getMockBuilder(CollectionInterface::class)
|
||||
->disableOriginalConstructor()
|
||||
->onlyMethods(['getMetadata'])
|
||||
->getMock();
|
||||
$collection->expects($this->once())->method('getMetadata')->willReturn($metadataList);
|
||||
$handler = new ToolListHandler($collection);
|
||||
$message = new Request(1, 'tools/list', []);
|
||||
$response = $handler->createResponse($message);
|
||||
$this->assertCount(1, $response->result['tools']);
|
||||
$this->assertArrayNotHasKey('nextCursor', $response->result);
|
||||
$handler = new ListToolsHandler($collection);
|
||||
$request = new ListToolsRequest();
|
||||
NSA::setProperty($request, 'id', 1);
|
||||
$response = $handler->handle($request);
|
||||
|
||||
$this->assertInstanceOf(ListToolsResult::class, $response->result);
|
||||
$this->assertCount(1, $response->result->tools);
|
||||
$this->assertNull($response->result->nextCursor);
|
||||
}
|
||||
|
||||
/**
|
||||
|
|
@ -71,8 +74,7 @@ class ToolListHandlerTest extends TestCase
|
|||
];
|
||||
}
|
||||
|
||||
#[Test]
|
||||
public function handlePagination(): void
|
||||
public function testHandlePagination()
|
||||
{
|
||||
$item = self::createMetadataItem();
|
||||
$collection = $this->getMockBuilder(CollectionInterface::class)
|
||||
|
|
@ -80,11 +82,14 @@ class ToolListHandlerTest extends TestCase
|
|||
->onlyMethods(['getMetadata'])
|
||||
->getMock();
|
||||
$collection->expects($this->once())->method('getMetadata')->willReturn([$item, $item]);
|
||||
$handler = new ToolListHandler($collection, 2);
|
||||
$message = new Request(1, 'tools/list', []);
|
||||
$response = $handler->createResponse($message);
|
||||
$this->assertCount(2, $response->result['tools']);
|
||||
$this->assertArrayHasKey('nextCursor', $response->result);
|
||||
$handler = new ListToolsHandler($collection, 2);
|
||||
$request = new ListToolsRequest();
|
||||
NSA::setProperty($request, 'id', 1);
|
||||
$response = $handler->handle($request);
|
||||
|
||||
$this->assertInstanceOf(ListToolsResult::class, $response->result);
|
||||
$this->assertCount(2, $response->result->tools);
|
||||
$this->assertNotNull($response->result->nextCursor);
|
||||
}
|
||||
|
||||
private static function createMetadataItem(): MetadataInterface
|
||||
|
|
|
|||
Loading…
Reference in a new issue