<?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\Server;

use Mcp\Message\Factory;
use Mcp\Message\Response;
use Mcp\Server\JsonRpcHandler;
use Mcp\Server\NotificationHandlerInterface;
use Mcp\Server\RequestHandlerInterface;
use PHPUnit\Framework\Attributes\CoversClass;
use PHPUnit\Framework\Attributes\Small;
use PHPUnit\Framework\Attributes\TestDox;
use PHPUnit\Framework\TestCase;
use Psr\Log\NullLogger;

#[Small]
#[CoversClass(JsonRpcHandler::class)]
class JsonRpcHandlerTest extends TestCase
{
    #[TestDox('Make sure a single notification can be handled by multiple handlers.')]
    public function testHandleMultipleNotifications(): void
    {
        $handlerA = $this->getMockBuilder(NotificationHandlerInterface::class)
            ->disableOriginalConstructor()
            ->onlyMethods(['supports', 'handle'])
            ->getMock();
        $handlerA->method('supports')->willReturn(true);
        $handlerA->expects($this->once())->method('handle');

        $handlerB = $this->getMockBuilder(NotificationHandlerInterface::class)
            ->disableOriginalConstructor()
            ->onlyMethods(['supports', 'handle'])
            ->getMock();
        $handlerB->method('supports')->willReturn(false);
        $handlerB->expects($this->never())->method('handle');

        $handlerC = $this->getMockBuilder(NotificationHandlerInterface::class)
            ->disableOriginalConstructor()
            ->onlyMethods(['supports', 'handle'])
            ->getMock();
        $handlerC->method('supports')->willReturn(true);
        $handlerC->expects($this->once())->method('handle');

        $jsonRpc = new JsonRpcHandler(new Factory(), [], [$handlerA, $handlerB, $handlerC], new NullLogger());
        $result = $jsonRpc->process(
            '{"jsonrpc": "2.0", "id": 1, "method": "notifications/foobar"}'
        );
        iterator_to_array($result);
    }

    #[TestDox('Make sure a single request can NOT be handled by multiple handlers.')]
    public function testHandleMultipleRequests(): void
    {
        $handlerA = $this->getMockBuilder(RequestHandlerInterface::class)
            ->disableOriginalConstructor()
            ->onlyMethods(['supports', 'createResponse'])
            ->getMock();
        $handlerA->method('supports')->willReturn(true);
        $handlerA->expects($this->once())->method('createResponse')->willReturn(new Response(1));

        $handlerB = $this->getMockBuilder(RequestHandlerInterface::class)
            ->disableOriginalConstructor()
            ->onlyMethods(['supports', 'createResponse'])
            ->getMock();
        $handlerB->method('supports')->willReturn(false);
        $handlerB->expects($this->never())->method('createResponse');

        $handlerC = $this->getMockBuilder(RequestHandlerInterface::class)
            ->disableOriginalConstructor()
            ->onlyMethods(['supports', 'createResponse'])
            ->getMock();
        $handlerC->method('supports')->willReturn(true);
        $handlerC->expects($this->never())->method('createResponse');

        $jsonRpc = new JsonRpcHandler(new Factory(), [$handlerA, $handlerB, $handlerC], [], new NullLogger());
        $result = $jsonRpc->process(
            '{"jsonrpc": "2.0", "id": 1, "method": "request/foobar"}'
        );
        iterator_to_array($result);
    }
}