Updated the mock init logic in strategy tests

manticoresoftware · Jan 12, 2024 · 8229f5e · 8229f5e
1 parent c8c82b0
commit 8229f5e
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/test/Manticoresearch/Connection/Strategy/RandomTest.php b/test/Manticoresearch/Connection/Strategy/RandomTest.php
@@ -45,8 +45,11 @@ public function testSequenceGood()
 
         $mConns = [];
         for ($i=0; $i<10; $i++) {
-            $mConns[] = mock::mock(\Manticoresearch\Connection::class)
-                ->shouldReceive('isAlive')->andReturn(true)->getMock();
+        	$mock = mock::mock(\Manticoresearch\Connection::class);
+        	$mock->shouldReceive('isAlive')->andReturn(true);
+        	$mock->shouldReceive('getHost')->andReturn($_SERVER['MS_HOST']);
+        	$mock->shouldReceive('getPort')->andReturn((int)($_SERVER['MS_PORT']));
+        	$mConns[] = $mock->getMock();
         }
         $connectionPool = new \Manticoresearch\Connection\ConnectionPool(
             $mConns,