monitor/api/tests/Unit/Controller/StreamControllerTest.php

<?php

declare(strict_types=1);

namespace Monitor\Api\Tests\Controller;

use Monitor\Api\Controller\StreamController;
use Monitor\Api\Db\Database;
use Monitor\Api\Http\Request;
use Monitor\Api\Realtime\ChangeFeed;
use Monitor\Api\Repository\AgentRepository;
use Monitor\Api\Repository\EventRepository;
use Monitor\Api\Repository\KpiRepository;
use Monitor\Api\Repository\SystemRepository;
use Monitor\Api\Repository\TicketRepository;
use PHPUnit\Framework\TestCase;

final class StreamControllerTest extends TestCase
{
    private TicketRepository $tickets;
    private ChangeFeed $changeFeed;

    protected function setUp(): void
    {
        $db = Database::inMemory();
        $this->tickets = new TicketRepository($db);
        $this->changeFeed = new ChangeFeed(
            $this->tickets,
            new EventRepository($db),
            new SystemRepository($db),
            new KpiRepository($db),
            new AgentRepository($db)
        );
    }

    public function testStreamSendsHeartbeatsWhenNothingChanges(): void
    {
        $controller = new StreamController(
            $this->changeFeed,
            maxIterations: 2,
            pollIntervalSeconds: 0,
            sleep: function (int $seconds): void {
            },
        );

        $response = $controller->stream(Request::create('GET', '/api/v1/stream'));
        self::assertTrue($response->isStream());

        $output = $this->captureWriterOutput($response->writer());

        self::assertStringContainsString(': heartbeat', $output);
        self::assertStringNotContainsString('event: update', $output);
    }

    public function testStreamEmitsUpdateEventWhenAResourceChangesBetweenPolls(): void
    {
        $tickets = $this->tickets;
        $pollCount = 0;
        $controller = new StreamController(
            $this->changeFeed,
            maxIterations: 2,
            pollIntervalSeconds: 0,
            sleep: function (int $seconds) use (&$pollCount, $tickets): void {
                $pollCount++;
                if ($pollCount === 1) {
                    $tickets->create(['title' => 'Neu waehrend des Wartens']);
                }
            },
        );

        $response = $controller->stream(Request::create('GET', '/api/v1/stream'));
        $output = $this->captureWriterOutput($response->writer());

        self::assertStringContainsString('event: update', $output);
        self::assertStringContainsString('"tickets"', $output);
        self::assertSame(2, $pollCount);
    }

    public function testStreamStopsAfterMaxIterations(): void
    {
        $pollCount = 0;
        $controller = new StreamController(
            $this->changeFeed,
            maxIterations: 3,
            pollIntervalSeconds: 0,
            sleep: function (int $seconds) use (&$pollCount): void {
                $pollCount++;
            },
        );

        $this->captureWriterOutput($controller->stream(Request::create('GET', '/api/v1/stream'))->writer());

        self::assertSame(3, $pollCount);
    }

    private function captureWriterOutput(\Closure $writer): string
    {
        ob_start();
        $writer();

        return (string) ob_get_clean();
    }
}