laravel-websockets/tests/TestCase.php

146 lines
4.3 KiB
PHP

<?php
namespace BeyondCode\LaravelWebSockets\Tests;
use BeyondCode\LaravelWebSockets\Facades\StatisticsLogger;
use BeyondCode\LaravelWebSockets\Tests\Mocks\Connection;
use BeyondCode\LaravelWebSockets\Tests\Mocks\Message;
use BeyondCode\LaravelWebSockets\Tests\Statistics\Logger\FakeStatisticsLogger;
use BeyondCode\LaravelWebSockets\WebSockets\Channels\ChannelManager;
use BeyondCode\LaravelWebSockets\WebSockets\WebSocketHandler;
use Clue\React\Buzz\Browser;
use GuzzleHttp\Psr7\Request;
use Mockery;
use Ratchet\ConnectionInterface;
abstract class TestCase extends \Orchestra\Testbench\TestCase
{
/** @var \BeyondCode\LaravelWebSockets\WebSockets\WebSocketHandler */
protected $pusherServer;
/** @var \BeyondCode\LaravelWebSockets\WebSockets\Channels\ChannelManager */
protected $channelManager;
public function setUp(): void
{
parent::setUp();
$this->pusherServer = app(WebSocketHandler::class);
$this->channelManager = app(ChannelManager::class);
StatisticsLogger::swap(new FakeStatisticsLogger(
$this->channelManager,
Mockery::mock(Browser::class)
));
$this->loadMigrationsFrom(__DIR__.'/../database/migrations');
}
protected function getPackageProviders($app)
{
return [
\Illuminate\Broadcasting\BroadcastServiceProvider::class,
\BeyondCode\LaravelWebSockets\WebSocketsServiceProvider::class,
];
}
protected function getEnvironmentSetUp($app)
{
$app['config']->set('websockets.apps', [
[
'name' => 'Test App',
'id' => '1234',
'key' => 'TestKey',
'secret' => 'TestSecret',
'host' => 'localhost',
'capacity' => null,
'enable_client_messages' => false,
'enable_statistics' => true,
],
]);
$app['config']->set('database.redis.default', [
'host' => env('REDIS_HOST', '127.0.0.1'),
'password' => env('REDIS_PASSWORD', null),
'port' => env('REDIS_PORT', '6379'),
'database' => env('REDIS_DB', '0'),
]);
$app['config']->set(
'websockets.replication.driver',
getenv('REPLICATION_DRIVER') ?: 'local'
);
}
protected function getWebSocketConnection(string $url = '/?appKey=TestKey'): Connection
{
$connection = new Connection();
$connection->httpRequest = new Request('GET', $url);
return $connection;
}
protected function getConnectedWebSocketConnection(array $channelsToJoin = [], string $url = '/?appKey=TestKey'): Connection
{
$connection = new Connection();
$connection->httpRequest = new Request('GET', $url);
$this->pusherServer->onOpen($connection);
foreach ($channelsToJoin as $channel) {
$message = new Message(json_encode([
'event' => 'pusher:subscribe',
'data' => [
'channel' => $channel,
],
]));
$this->pusherServer->onMessage($connection, $message);
}
return $connection;
}
protected function joinPresenceChannel($channel): Connection
{
$connection = $this->getWebSocketConnection();
$this->pusherServer->onOpen($connection);
$channelData = [
'user_id' => 1,
'user_info' => [
'name' => 'Marcel',
],
];
$signature = "{$connection->socketId}:{$channel}:".json_encode($channelData);
$message = new Message(json_encode([
'event' => 'pusher:subscribe',
'data' => [
'auth' => $connection->app->key.':'.hash_hmac('sha256', $signature, $connection->app->secret),
'channel' => $channel,
'channel_data' => json_encode($channelData),
],
]));
$this->pusherServer->onMessage($connection, $message);
return $connection;
}
protected function getChannel(ConnectionInterface $connection, string $channelName)
{
return $this->channelManager->findOrCreate($connection->app->id, $channelName);
}
protected function markTestAsPassed()
{
$this->assertTrue(true);
}
}