Add RequestContext and logging (#176)
* Implemented MCP logging specification with auto-injection * Add RequestContext * fix tests and style * Better deprecation * fix: minor * Remove ClientAwareInterface * fix cs * fix docs --------- Co-authored-by: Adam Jamiu <jamiuadam120@gmail.com>
This commit is contained in:
parent
369932378f
commit
a1e0f61b6b
4 changed files with 578 additions and 162 deletions
|
|
@ -71,7 +71,7 @@ class CallToolHandlerTest extends TestCase
|
|||
$this->referenceHandler
|
||||
->expects($this->once())
|
||||
->method('handle')
|
||||
->with($toolReference, ['name' => 'John', '_session' => $this->session])
|
||||
->with($toolReference, ['name' => 'John', '_session' => $this->session, '_request' => $request])
|
||||
->willReturn('Hello, John!');
|
||||
|
||||
$toolReference
|
||||
|
|
@ -104,7 +104,7 @@ class CallToolHandlerTest extends TestCase
|
|||
$this->referenceHandler
|
||||
->expects($this->once())
|
||||
->method('handle')
|
||||
->with($toolReference, ['_session' => $this->session])
|
||||
->with($toolReference, ['_session' => $this->session, '_request' => $request])
|
||||
->willReturn('Simple result');
|
||||
|
||||
$toolReference
|
||||
|
|
@ -141,7 +141,7 @@ class CallToolHandlerTest extends TestCase
|
|||
$this->referenceHandler
|
||||
->expects($this->once())
|
||||
->method('handle')
|
||||
->with($toolReference, array_merge($arguments, ['_session' => $this->session]))
|
||||
->with($toolReference, array_merge($arguments, ['_session' => $this->session, '_request' => $request]))
|
||||
->willReturn('Complex result');
|
||||
|
||||
$toolReference
|
||||
|
|
@ -192,7 +192,7 @@ class CallToolHandlerTest extends TestCase
|
|||
$this->referenceHandler
|
||||
->expects($this->once())
|
||||
->method('handle')
|
||||
->with($toolReference, ['param' => 'value', '_session' => $this->session])
|
||||
->with($toolReference, ['param' => 'value', '_session' => $this->session, '_request' => $request])
|
||||
->willThrowException($exception);
|
||||
|
||||
$this->logger
|
||||
|
|
@ -227,7 +227,7 @@ class CallToolHandlerTest extends TestCase
|
|||
$this->referenceHandler
|
||||
->expects($this->once())
|
||||
->method('handle')
|
||||
->with($toolReference, ['_session' => $this->session])
|
||||
->with($toolReference, ['_session' => $this->session, '_request' => $request])
|
||||
->willReturn(null);
|
||||
|
||||
$toolReference
|
||||
|
|
@ -264,7 +264,7 @@ class CallToolHandlerTest extends TestCase
|
|||
$this->referenceHandler
|
||||
->expects($this->once())
|
||||
->method('handle')
|
||||
->with($toolReference, ['key1' => 'value1', 'key2' => 42, '_session' => $this->session])
|
||||
->with($toolReference, ['key1' => 'value1', 'key2' => 42, '_session' => $this->session, '_request' => $request])
|
||||
->willThrowException($exception);
|
||||
|
||||
$this->logger
|
||||
|
|
@ -274,7 +274,7 @@ class CallToolHandlerTest extends TestCase
|
|||
'Error while executing tool "test_tool": "Custom error message".',
|
||||
[
|
||||
'tool' => 'test_tool',
|
||||
'arguments' => ['key1' => 'value1', 'key2' => 42, '_session' => $this->session],
|
||||
'arguments' => ['key1' => 'value1', 'key2' => 42, '_session' => $this->session, '_request' => $request],
|
||||
],
|
||||
);
|
||||
|
||||
|
|
@ -307,7 +307,7 @@ class CallToolHandlerTest extends TestCase
|
|||
$this->referenceHandler
|
||||
->expects($this->once())
|
||||
->method('handle')
|
||||
->with($toolReference, ['param' => 'value', '_session' => $this->session])
|
||||
->with($toolReference, ['param' => 'value', '_session' => $this->session, '_request' => $request])
|
||||
->willThrowException($exception);
|
||||
|
||||
$response = $this->handler->handle($request, $this->session);
|
||||
|
|
@ -334,7 +334,7 @@ class CallToolHandlerTest extends TestCase
|
|||
$this->referenceHandler
|
||||
->expects($this->once())
|
||||
->method('handle')
|
||||
->with($toolReference, ['_session' => $this->session])
|
||||
->with($toolReference, ['_session' => $this->session, '_request' => $request])
|
||||
->willReturn('Special tool result');
|
||||
|
||||
$toolReference
|
||||
|
|
@ -369,7 +369,7 @@ class CallToolHandlerTest extends TestCase
|
|||
$this->referenceHandler
|
||||
->expects($this->once())
|
||||
->method('handle')
|
||||
->with($toolReference, array_merge($arguments, ['_session' => $this->session]))
|
||||
->with($toolReference, array_merge($arguments, ['_session' => $this->session, '_request' => $request]))
|
||||
->willReturn('Unicode handled');
|
||||
|
||||
$toolReference
|
||||
|
|
@ -399,7 +399,7 @@ class CallToolHandlerTest extends TestCase
|
|||
$this->referenceHandler
|
||||
->expects($this->once())
|
||||
->method('handle')
|
||||
->with($toolReference, ['query' => 'php', '_session' => $this->session])
|
||||
->with($toolReference, ['query' => 'php', '_session' => $this->session, '_request' => $request])
|
||||
->willReturn($structuredResult);
|
||||
|
||||
$toolReference
|
||||
|
|
@ -428,7 +428,7 @@ class CallToolHandlerTest extends TestCase
|
|||
$this->referenceHandler
|
||||
->expects($this->once())
|
||||
->method('handle')
|
||||
->with($toolReference, ['query' => 'php', '_session' => $this->session])
|
||||
->with($toolReference, ['query' => 'php', '_session' => $this->session, '_request' => $request])
|
||||
->willReturn($callToolResult);
|
||||
|
||||
$toolReference
|
||||
|
|
|
|||
|
|
@ -70,7 +70,7 @@ class GetPromptHandlerTest extends TestCase
|
|||
$this->referenceHandler
|
||||
->expects($this->once())
|
||||
->method('handle')
|
||||
->with($promptReference, ['_session' => $this->session])
|
||||
->with($promptReference, ['_session' => $this->session, '_request' => $request])
|
||||
->willReturn($expectedMessages);
|
||||
|
||||
$promptReference
|
||||
|
|
@ -112,7 +112,7 @@ class GetPromptHandlerTest extends TestCase
|
|||
$this->referenceHandler
|
||||
->expects($this->once())
|
||||
->method('handle')
|
||||
->with($promptReference, array_merge($arguments, ['_session' => $this->session]))
|
||||
->with($promptReference, array_merge($arguments, ['_session' => $this->session, '_request' => $request]))
|
||||
->willReturn($expectedMessages);
|
||||
|
||||
$promptReference
|
||||
|
|
@ -145,7 +145,7 @@ class GetPromptHandlerTest extends TestCase
|
|||
$this->referenceHandler
|
||||
->expects($this->once())
|
||||
->method('handle')
|
||||
->with($promptReference, ['_session' => $this->session])
|
||||
->with($promptReference, ['_session' => $this->session, '_request' => $request])
|
||||
->willReturn($expectedMessages);
|
||||
|
||||
$promptReference
|
||||
|
|
@ -178,7 +178,7 @@ class GetPromptHandlerTest extends TestCase
|
|||
$this->referenceHandler
|
||||
->expects($this->once())
|
||||
->method('handle')
|
||||
->with($promptReference, ['_session' => $this->session])
|
||||
->with($promptReference, ['_session' => $this->session, '_request' => $request])
|
||||
->willReturn($expectedMessages);
|
||||
|
||||
$promptReference
|
||||
|
|
@ -213,7 +213,7 @@ class GetPromptHandlerTest extends TestCase
|
|||
$this->referenceHandler
|
||||
->expects($this->once())
|
||||
->method('handle')
|
||||
->with($promptReference, ['_session' => $this->session])
|
||||
->with($promptReference, ['_session' => $this->session, '_request' => $request])
|
||||
->willReturn($expectedMessages);
|
||||
|
||||
$promptReference
|
||||
|
|
@ -299,7 +299,7 @@ class GetPromptHandlerTest extends TestCase
|
|||
$this->referenceHandler
|
||||
->expects($this->once())
|
||||
->method('handle')
|
||||
->with($promptReference, array_merge($arguments, ['_session' => $this->session]))
|
||||
->with($promptReference, array_merge($arguments, ['_session' => $this->session, '_request' => $request]))
|
||||
->willReturn($expectedMessages);
|
||||
|
||||
$promptReference
|
||||
|
|
@ -337,7 +337,7 @@ class GetPromptHandlerTest extends TestCase
|
|||
$this->referenceHandler
|
||||
->expects($this->once())
|
||||
->method('handle')
|
||||
->with($promptReference, array_merge($arguments, ['_session' => $this->session]))
|
||||
->with($promptReference, array_merge($arguments, ['_session' => $this->session, '_request' => $request]))
|
||||
->willReturn($expectedMessages);
|
||||
|
||||
$promptReference
|
||||
|
|
@ -367,7 +367,7 @@ class GetPromptHandlerTest extends TestCase
|
|||
$this->referenceHandler
|
||||
->expects($this->once())
|
||||
->method('handle')
|
||||
->with($promptReference, ['_session' => $this->session])
|
||||
->with($promptReference, ['_session' => $this->session, '_request' => $request])
|
||||
->willReturn([]);
|
||||
|
||||
$promptReference
|
||||
|
|
@ -405,7 +405,7 @@ class GetPromptHandlerTest extends TestCase
|
|||
$this->referenceHandler
|
||||
->expects($this->once())
|
||||
->method('handle')
|
||||
->with($promptReference, array_merge($arguments, ['_session' => $this->session]))
|
||||
->with($promptReference, array_merge($arguments, ['_session' => $this->session, '_request' => $request]))
|
||||
->willReturn($expectedMessages);
|
||||
|
||||
$promptReference
|
||||
|
|
|
|||
|
|
@ -75,7 +75,7 @@ class ReadResourceHandlerTest extends TestCase
|
|||
$this->referenceHandler
|
||||
->expects($this->once())
|
||||
->method('handle')
|
||||
->with($resourceReference, ['uri' => $uri, '_session' => $this->session])
|
||||
->with($resourceReference, ['uri' => $uri, '_session' => $this->session, '_request' => $request])
|
||||
->willReturn('test');
|
||||
|
||||
$resourceReference
|
||||
|
|
@ -115,7 +115,7 @@ class ReadResourceHandlerTest extends TestCase
|
|||
$this->referenceHandler
|
||||
->expects($this->once())
|
||||
->method('handle')
|
||||
->with($resourceReference, ['uri' => $uri, '_session' => $this->session])
|
||||
->with($resourceReference, ['uri' => $uri, '_session' => $this->session, '_request' => $request])
|
||||
->willReturn('fake-image-data');
|
||||
|
||||
$resourceReference
|
||||
|
|
@ -159,7 +159,7 @@ class ReadResourceHandlerTest extends TestCase
|
|||
$this->referenceHandler
|
||||
->expects($this->once())
|
||||
->method('handle')
|
||||
->with($resourceReference, ['uri' => $uri, '_session' => $this->session])
|
||||
->with($resourceReference, ['uri' => $uri, '_session' => $this->session, '_request' => $request])
|
||||
->willReturn('binary-data');
|
||||
|
||||
$resourceReference
|
||||
|
|
@ -267,7 +267,7 @@ class ReadResourceHandlerTest extends TestCase
|
|||
$this->referenceHandler
|
||||
->expects($this->once())
|
||||
->method('handle')
|
||||
->with($resourceReference, ['uri' => $uri, '_session' => $this->session])
|
||||
->with($resourceReference, ['uri' => $uri, '_session' => $this->session, '_request' => $request])
|
||||
->willReturn('test');
|
||||
|
||||
$resourceReference
|
||||
|
|
@ -312,7 +312,7 @@ class ReadResourceHandlerTest extends TestCase
|
|||
$this->referenceHandler
|
||||
->expects($this->once())
|
||||
->method('handle')
|
||||
->with($resourceReference, ['uri' => $uri, '_session' => $this->session])
|
||||
->with($resourceReference, ['uri' => $uri, '_session' => $this->session, '_request' => $request])
|
||||
->willReturn('');
|
||||
|
||||
$resourceReference
|
||||
|
|
@ -374,7 +374,7 @@ class ReadResourceHandlerTest extends TestCase
|
|||
$this->referenceHandler
|
||||
->expects($this->once())
|
||||
->method('handle')
|
||||
->with($resourceReference, ['uri' => $uri, '_session' => $this->session])
|
||||
->with($resourceReference, ['uri' => $uri, '_session' => $this->session, '_request' => $request])
|
||||
->willReturn($expectedContent);
|
||||
|
||||
$resourceReference
|
||||
|
|
|
|||
74
tests/Unit/Server/Handler/Request/SetLogLevelHandlerTest.php
Normal file
74
tests/Unit/Server/Handler/Request/SetLogLevelHandlerTest.php
Normal file
|
|
@ -0,0 +1,74 @@
|
|||
<?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\Unit\Server\Handler\Request;
|
||||
|
||||
use Mcp\Schema\Enum\LoggingLevel;
|
||||
use Mcp\Schema\JsonRpc\Request;
|
||||
use Mcp\Schema\Request\SetLogLevelRequest;
|
||||
use Mcp\Schema\Result\EmptyResult;
|
||||
use Mcp\Server\Handler\Request\SetLogLevelHandler;
|
||||
use Mcp\Server\Protocol;
|
||||
use Mcp\Server\Session\Session;
|
||||
use PHPUnit\Framework\TestCase;
|
||||
|
||||
/**
|
||||
* @author Adam Jamiu <jamiuadam120@gmail.com>
|
||||
*/
|
||||
class SetLogLevelHandlerTest extends TestCase
|
||||
{
|
||||
public function testSupports(): void
|
||||
{
|
||||
$request = $this->createSetLogLevelRequest(LoggingLevel::Info);
|
||||
$handler = new SetLogLevelHandler();
|
||||
$this->assertTrue($handler->supports($request));
|
||||
}
|
||||
|
||||
public function testDoesNotSupportOtherRequests(): void
|
||||
{
|
||||
$otherRequest = $this->createMock(Request::class);
|
||||
$handler = new SetLogLevelHandler();
|
||||
$this->assertFalse($handler->supports($otherRequest));
|
||||
}
|
||||
|
||||
public function testHandleAllLogLevelsAndSupport(): void
|
||||
{
|
||||
$handler = new SetLogLevelHandler();
|
||||
|
||||
foreach (LoggingLevel::cases() as $level) {
|
||||
$request = $this->createSetLogLevelRequest($level);
|
||||
|
||||
$session = $this->getMockBuilder(Session::class)
|
||||
->disableOriginalConstructor()
|
||||
->onlyMethods(['set'])
|
||||
->getMock();
|
||||
$session->expects($this->once())
|
||||
->method('set')
|
||||
->with(Protocol::SESSION_LOGGING_LEVEL, $level->value);
|
||||
|
||||
$response = $handler->handle($request, $session);
|
||||
$this->assertEquals($request->getId(), $response->id);
|
||||
$this->assertInstanceOf(EmptyResult::class, $response->result);
|
||||
}
|
||||
}
|
||||
|
||||
private function createSetLogLevelRequest(LoggingLevel $level): SetLogLevelRequest
|
||||
{
|
||||
return SetLogLevelRequest::fromArray([
|
||||
'jsonrpc' => '2.0',
|
||||
'method' => SetLogLevelRequest::getMethod(),
|
||||
'id' => 'test-request-'.uniqid(),
|
||||
'params' => [
|
||||
'level' => $level->value,
|
||||
],
|
||||
]);
|
||||
}
|
||||
}
|
||||
Loading…
Reference in a new issue