fix: allow uppercase table names by quoting the table name when fetching the primary key of a table in PostgreSQL (#82)

Author: jandrusku

src/Codeception/Lib/Driver/PostgreSql.php

@@ -164,7 +164,7 @@ public function getPrimaryKey(string $tableName): array
                 FROM   pg_index i
                 JOIN   pg_attribute a ON a.attrelid = i.indrelid
                                      AND a.attnum = ANY(i.indkey)
-                WHERE  i.indrelid = '{$tableName}'::regclass
+                WHERE  i.indrelid = '\"{$tableName}\"'::regclass
                 AND    i.indisprimary";
             $stmt = $this->executeQuery($query, []);
             $columns = $stmt->fetchAll(PDO::FETCH_ASSOC);

tests/data/dumps/postgres.sql

@@ -448,6 +448,10 @@ CREATE TABLE "no_pk" (
     "status" VARCHAR NOT NULL
 );
 
+CREATE TABLE "NoPk" (
+    "Status" VARCHAR NOT NULL
+);
+
 CREATE TABLE "order" (
     "id" INTEGER NOT NULL PRIMARY KEY,
     "name" VARCHAR NOT NULL,

tests/unit/Codeception/Module/Db/PostgreSqlDbTest.php

@@ -33,4 +33,10 @@ public function getConfig(): array
             'populate' => true
         ];
     }
+
+    public function testHaveInDatabaseWithUppercaseTableName()
+    {
+        $testData = ['Status' => 'test'];
+        $this->module->haveInDatabase('NoPk', $testData);
+    }
 }