From dfe0dbf3353c1324a1dc9b1a19a4e93060fa90c5 Mon Sep 17 00:00:00 2001
From: Alex Renoki <rennokki@gmail.com>
Date: Fri, 11 Sep 2020 15:19:15 +0300
Subject: [PATCH] Updated mode

---
 tests/TestCase.php | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tests/TestCase.php b/tests/TestCase.php
index 668e92b..b361931 100644
--- a/tests/TestCase.php
+++ b/tests/TestCase.php
@@ -230,7 +230,9 @@ abstract class TestCase extends Orchestra
     protected function registerStatisticsCollectors()
     {
         $this->app->singleton(StatisticsCollector::class, function () {
-            $class = config("websockets.replication.modes.{$this->replicationMode}.collector");
+            $mode = config('websockets.replication.mode', $this->replicationMode);
+
+            $class = config("websockets.replication.modes.{$mode}.collector");
 
             return new $class;
         });