Support localhost:port notation for tests

staabm · Nov 27, 2024 · 92a5085 · 92a5085
1 parent a2d990f
commit 92a5085
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/tests/ReflectorFactory.php b/tests/ReflectorFactory.php
@@ -74,12 +74,17 @@ public static function create(string $cacheDir): QueryReflector
         if (self::MODE_RECORDING === $mode || self::MODE_REPLAY_AND_RECORDING === $mode) {
             $schemaHasher = null;
 
+            $port = null;
+            if (str_contains($host, ':')) {
+                [$host, $port] = explode(':', $host, 2);
+            }
+
             if ('mysqli' === $reflector) {
                 $mysqli = mysqli_init();
                 if (! $mysqli) {
                     throw new \RuntimeException('Unable to init mysqli');
                 }
-                $mysqli->real_connect($host, $user, $password, $dbname, null, null, $ssl ? MYSQLI_CLIENT_SSL : 0);
+                $mysqli->real_connect($host, $user, $password, $dbname, $port, null, $ssl ? MYSQLI_CLIENT_SSL : 0);
                 $reflector = new MysqliQueryReflector($mysqli);
                 $schemaHasher = new SchemaHasherMysql($mysqli);
             } elseif ('pdo-mysql' === $reflector) {
@@ -88,11 +93,11 @@ public static function create(string $cacheDir): QueryReflector
                     $options[PDO::MYSQL_ATTR_SSL_CA] = $ssl;
                     $options[PDO::MYSQL_ATTR_SSL_VERIFY_SERVER_CERT] = false;
                 }
-                $pdo = new PDO(sprintf('mysql:dbname=%s;host=%s', $dbname, $host), $user, $password, $options);
+                $pdo = new PDO(sprintf('mysql:dbname=%s;host=%s;port=%s', $dbname, $host, $port), $user, $password, $options);
                 $reflector = new PdoMysqlQueryReflector($pdo);
                 $schemaHasher = new SchemaHasherMysql($pdo);
             } elseif ('pdo-pgsql' === $reflector) {
-                $pdo = new PDO(sprintf('pgsql:dbname=%s;host=%s', $dbname, $host), $user, $password);
+                $pdo = new PDO(sprintf('pgsql:dbname=%s;host=%s;port=%s', $dbname, $host, $port), $user, $password);
                 $reflector = new PdoPgSqlQueryReflector($pdo);
             } else {
                 throw new \RuntimeException('Unknown reflector: ' . $reflector);