laravel-websockets/tests/Mocks/Connection.php

<?php

namespace BeyondCode\LaravelWebSockets\Test\Mocks;

use GuzzleHttp\Psr7\Request;
use PHPUnit\Framework\Assert as PHPUnit;
use Ratchet\ConnectionInterface;

class Connection implements ConnectionInterface
{
    /**
     * The request instance.
     *
     * @var Request
     */
    public $httpRequest;

    /**
     * The sent data through the connection.
     *
     * @var array
     */
    public $sentData = [];

    /**
     * The raw (unencoded) sent data.
     *
     * @var array
     */
    public $sentRawData = [];

    /**
     * Wether the connection has been closed.
     *
     * @var bool
     */
    public $closed = false;

    /**
     * Send the data through the connection.
     *
     * @param  mixed  $data
     * @return void
     */
    public function send($data)
    {
        $this->sentData[] = json_decode($data, true);
        $this->sentRawData[] = $data;
    }

    /**
     * Mark the connection as closed.
     *
     * @return void
     */
    public function close()
    {
        $this->closed = true;
    }

    /**
     * Reset the events for assertions.
     *
     * @return $this
     */
    public function resetEvents()
    {
        $this->sentData = [];
        $this->sentRawData = [];

        return $this;
    }

    /**
     * Dump & stop execution.
     *
     * @return void
     */
    public function dd()
    {
        dd([
            'sentData' => $this->sentData,
            'sentRawData' => $this->sentRawData,
        ]);
    }

    /**
     * Assert that an event got sent.
     *
     * @param  string  $name
     * @param  array  $additionalParameters
     * @return $this
     */
    public function assertSentEvent(string $name, array $additionalParameters = [])
    {
        $event = collect($this->sentData)->firstWhere('event', '=', $name);

        PHPUnit::assertTrue(
            ! is_null($event)
        );

        foreach ($additionalParameters as $parameter => $value) {
            PHPUnit::assertSame($event[$parameter], $value);
        }

        return $this;
    }

    /**
     * Assert that an event got not sent.
     *
     * @param  string  $name
     * @return $this
     */
    public function assertNotSentEvent(string $name)
    {
        $event = collect($this->sentData)->firstWhere('event', '=', $name);

        PHPUnit::assertTrue(
            is_null($event)
        );

        return $this;
    }

    /**
     * Assert that no events occured within the connection.
     *
     * @return $this
     */
    public function assertNothingSent()
    {
        PHPUnit::assertEquals([], $this->sentData);

        return $this;
    }

    /**
     * Assert the connection is closed.
     *
     * @return $this
     */
    public function assertClosed()
    {
        PHPUnit::assertTrue($this->closed);

        return $this;
    }
}