Execution.php

<?php

declare(strict_types=1);

namespace Pest\Browser;

use Amp\ByteStream\ReadableResourceStream;
use Pest\Browser\Playwright\Playwright;
use Pest\Support\Container;
use PHPUnit\Framework\Assert;
use PHPUnit\Framework\ExpectationFailedException;
use PHPUnit\Framework\TestStatus\TestStatus;
use ReflectionClass;
use Symfony\Component\Console\Output\OutputInterface;

use function Amp\async;
use function Amp\delay;

/**
 * @internal
 */
final class Execution
{
    /**
     * Either the current context
     */
    private bool $waiting = false;

    /**
     * The current context instance, or null if not set.
     */
    private static ?Execution $instance = null;

    /**
     * Creates a new context instance.
     */
    public static function instance(): self
    {
        if (! self::$instance instanceof self) {
            self::$instance = new self();
        }

        return self::$instance;
    }

    /**
     * Debugs the test execution by waiting for a key press.
     */
    public function debug(TestStatus $status): void
    {
        $this->waiting = true;

        // @phpstan-ignore-next-line
        $testName = str_replace('__pest_evaluable_', '', test()->name());
        $message = $status->message();

        // @phpstan-ignore-next-line
        Container::getInstance()->get(OutputInterface::class)->writeln(
            "\n  <info>Test [{$testName}] failed with the message:</info> {$message}\n"
        );

        // @phpstan-ignore-next-line
        Container::getInstance()->get(OutputInterface::class)->writeln(
            '  <info>Press any key to continue...</info>'
        );

        $stdin = new ReadableResourceStream(STDIN);

        async(function () use ($stdin): void {
            while ($stdin->read() !== null) {
                $stdin->close();

                delay(0.1);

                break;
            }
        })->await();
    }

    /**
     * Pauses the execution.
     */
    public function wait(int|float $seconds = 1): void
    {
        async(function () use ($seconds): void {
            $this->waiting = true;

            delay($seconds);

            $this->waiting = false;
        })->await();
    }

    /**
     * Checks if the execution is paused.
     */
    public function isWaiting(): bool
    {
        return $this->waiting;
    }

    /**
     * Ticks the execution.
     */
    public function tick(): void
    {
        delay(0);
    }

    /**
     * Waits for a key press.
     */
    public function waitForKey(): void
    {
        $this->waiting = true;

        // @phpstan-ignore-next-line
        Container::getInstance()->get(OutputInterface::class)->writeln(
            '  <info>Press any key to continue...</info>'
        );

        $stdin = new ReadableResourceStream(STDIN);

        async(function () use ($stdin): void {
            while ($stdin->read() !== null) {
                $stdin->close();

                delay(0.1);

                break;
            }
        })->await();
    }

    /**
     * Awaits for a condition to be met, retrying until the timeout is reached.
     */
    public function waitForExpectation(callable $callback): mixed
    {
        $timeout = Playwright::timeout();

        $originalCount = Assert::getCount();

        $start = microtime(true);
        $end = $start + ($timeout / 1_000);

        while (microtime(true) < $end) {
            try {
                return Playwright::usingTimeout(1_000, $callback);
            } catch (ExpectationFailedException) {
                //
            }

            $this->resetAssertions($originalCount);

            self::instance()->tick();
        }

        return $callback();
    }

    /**
     * Resets the assertion count to the original value.
     */
    private function resetAssertions(int $originalCount): void
    {
        if (Assert::getCount() === $originalCount) {
            return;
        }

        $reflector = new ReflectionClass(Assert::class);
        $property = $reflector->getProperty('count');

        $property->setValue(null, $originalCount);
    }
}