Added tests for statistics loggers drivers

Author: rennokki

src/Statistics/Logger/MemoryStatisticsLogger.php

@@ -125,4 +125,14 @@ protected function findOrMakeStatisticForAppId($appId): Statistic
 
         return $this->statistics[$appId];
     }
+
+    /**
+     * Get the saved statistics.
+     *
+     * @return array
+     */
+    public function getStatistics(): array
+    {
+        return $this->statistics;
+    }
 }

tests/Statistics/Logger/StatisticsLoggerTest.php

@@ -3,6 +3,9 @@
 namespace BeyondCode\LaravelWebSockets\Tests\Statistics\Controllers;
 
 use BeyondCode\LaravelWebSockets\Facades\StatisticsLogger;
+use BeyondCode\LaravelWebSockets\Statistics\Logger\MemoryStatisticsLogger;
+use BeyondCode\LaravelWebSockets\Statistics\Logger\NullStatisticsLogger;
+use BeyondCode\LaravelWebSockets\Statistics\Models\WebSocketsStatisticsEntry;
 use BeyondCode\LaravelWebSockets\Tests\TestCase;
 
 class StatisticsLoggerTest extends TestCase
@@ -43,4 +46,50 @@ public function it_counts_unique_connections_no_channel_subscriptions()
 
         $this->assertEquals(1, StatisticsLogger::getForAppId(1234)['peak_connection_count']);
     }
+
+    /** @test */
+    public function it_counts_connections_with_memory_logger()
+    {
+        $connection = $this->getConnectedWebSocketConnection(['channel-1']);
+
+        $logger = new MemoryStatisticsLogger(
+            $this->channelManager,
+            $this->statisticsDriver
+        );
+
+        $logger->webSocketMessage($connection->app->id);
+        $logger->apiMessage($connection->app->id);
+        $logger->connection($connection->app->id);
+        $logger->disconnection($connection->app->id);
+
+        $logger->save();
+
+        $this->assertCount(1, WebSocketsStatisticsEntry::all());
+
+        $entry = WebSocketsStatisticsEntry::first();
+
+        $this->assertEquals(1, $entry->peak_connection_count);
+        $this->assertEquals(1, $entry->websocket_message_count);
+        $this->assertEquals(1, $entry->api_message_count);
+    }
+
+    /** @test */
+    public function it_counts_connections_with_null_logger()
+    {
+        $connection = $this->getConnectedWebSocketConnection(['channel-1']);
+
+        $logger = new NullStatisticsLogger(
+            $this->channelManager,
+            $this->statisticsDriver
+        );
+
+        $logger->webSocketMessage($connection->app->id);
+        $logger->apiMessage($connection->app->id);
+        $logger->connection($connection->app->id);
+        $logger->disconnection($connection->app->id);
+
+        $logger->save();
+
+        $this->assertCount(0, WebSocketsStatisticsEntry::all());
+    }
 }

tests/TestCase.php

@@ -31,16 +31,25 @@ abstract class TestCase extends \Orchestra\Testbench\TestCase
      */
     protected $channelManager;
 
+    /**
+     * The used statistics driver.
+     *
+     * @var \BeyondCode\LaravelWebSockets\Statistics\Drivers\StatisticsDriver
+     */
+    protected $statisticsDriver;
+
     /**
      * {@inheritdoc}
      */
     public function setUp(): void
     {
         parent::setUp();
 
-        $this->pusherServer = app(config('websockets.handlers.websocket'));
+        $this->pusherServer = $this->app->make(config('websockets.handlers.websocket'));
+
+        $this->channelManager = $this->app->make(ChannelManager::class);
 
-        $this->channelManager = app(ChannelManager::class);
+        $this->statisticsDriver = $this->app->make(StatisticsDriver::class);
 
         StatisticsLogger::swap(new FakeStatisticsLogger(
             $this->channelManager,