<?php

declare(strict_types=1);

namespace LIT\MCP\Tests\Unit\Server;

use Brain\Monkey\Functions;
use LIT\MCP\Server\RequestContext;
use LIT\MCP\Tests\Unit\TestCase;
use WP_REST_Request;

final class RequestContextTest extends TestCase
{
    protected function tearDown(): void
    {
        RequestContext::reset();
        parent::tearDown();
    }

    public function testIsOAuthServerRequestIsFalseByDefault(): void
    {
        self::assertFalse(RequestContext::isOAuthServerRequest());
    }

    public function testRecordRouteMarksTheOAuthServerRoute(): void
    {
        $request = new WP_REST_Request('/mcp/mcp-oauth-server');

        RequestContext::recordRoute(null, [], $request);

        self::assertTrue(RequestContext::isOAuthServerRequest());
    }

    public function testRecordRouteIgnoresOtherRoutes(): void
    {
        $request = new WP_REST_Request('/wp/v2/users/me');

        RequestContext::recordRoute(null, [], $request);

        self::assertFalse(RequestContext::isOAuthServerRequest());
    }

    public function testRecordRouteReturnsResponseUnchanged(): void
    {
        $response = new \stdClass();
        $request = new WP_REST_Request('/mcp/mcp-oauth-server');

        $result = RequestContext::recordRoute($response, [], $request);

        self::assertSame($response, $result);
    }

    public function testRegisterHooksRecordRouteOnRestRequestBeforeCallbacks(): void
    {
        Functions\expect('add_filter')
            ->once()
            ->with('rest_request_before_callbacks', [RequestContext::class, 'recordRoute'], 10, 3);

        (new RequestContext())->register();
    }
}