protonemedia / inertiajs-events-laravel-dusk Goto Github PK

use Facebook\WebDriver\Exception\ScriptTimeoutException;
use Facebook\WebDriver\Exception\TimeoutException;
use Laravel\Dusk\Browser;

function (string $name, ?int $seconds = null): Browser {
    /** @var \Laravel\Dusk\Browser $this */
    $seconds = is_null($seconds) ? $this::$waitSeconds : $seconds;

    try {
        $this->driver->manage()
            ->timeouts()
            ->setScriptTimeout($seconds);

        $this->driver->executeAsyncScript("
            const done = () => arguments[0]();
            document.addEventListener('inertia:{$name}', done, { once: true });
        ");
    } catch (ScriptTimeoutException $e) {
        throw new TimeoutException("Waited {$seconds} seconds for an Inertia '{$name}' event.");
    }

    return $this;
}

<?php

use Closure;
use Facebook\WebDriver\Exception\ScriptTimeoutException;
use Facebook\WebDriver\Exception\TimeoutException;
use Laravel\Dusk\Browser;

/** @mixin \Laravel\Dusk\Browser */
class InertiaEventsMixin
{
    public function waitForInertia(?int $seconds = null): Closure
    {
        return fn (?int $seconds = null) => $this->waitForInertiaEvent('navigate', $seconds);
    }

    public function waitForInertiaError(?int $seconds = null): Closure
    {
        return fn (?int $seconds = null) => $this->waitForInertiaEvent('error', $seconds);
    }

    public function waitForInertiaFinish(?int $seconds = null): Closure
    {
        return fn (?int $seconds = null) => $this->waitForInertiaEvent('finish', $seconds);
    }

    public function waitForInertiaInvalid(?int $seconds = null): Closure
    {
        return fn (?int $seconds = null) => $this->waitForInertiaEvent('invalid', $seconds);
    }

    public function waitForInertiaSuccess(): Closure
    {
        return fn (?int $seconds = null) => $this->waitForInertiaEvent('success', $seconds);
    }

    public function waitForInertiaEvent(): Closure
    {
        return function (string $name, ?int $seconds = null): Browser {
            /** @var \Laravel\Dusk\Browser $this */
            $seconds = is_null($seconds) ? $this::$waitSeconds : $seconds;

            try {
                $this->driver->manage()
                    ->timeouts()
                    ->setScriptTimeout($seconds);

                $this->driver->executeAsyncScript("
                    const done = () => arguments[0]();
                    document.addEventListener('inertia:{$name}', done, { once: true });
                ");
            } catch (ScriptTimeoutException $e) {
                throw new TimeoutException("Waited {$seconds} seconds for an Inertia '{$name}' event.");
            }

            return $this;
        };
    }
}

\Laravel\Dusk\Browser::mixin(new InertiaEventsMixin);