feat: support auto increment on a composite pk

Author: JesusTheHun

src/Codeception/Module/Db.php

@@ -759,8 +759,8 @@ protected function loadDumpUsingDriver(string $databaseKey): void
     }
 
     /**
-     * Inserts an SQL record into a database. This record will be erased after the test, 
-     * unless you've configured "skip_cleanup_if_failed", and the test fails. 
+     * Inserts an SQL record into a database. This record will be erased after the test,
+     * unless you've configured "skip_cleanup_if_failed", and the test fails.
      *
      * ```php
      * <?php
@@ -802,12 +802,14 @@ private function addInsertedRow(string $table, array $row, $id): void
         $primary = [];
         if ($primaryKey !== []) {
             $filledKeys = array_intersect($primaryKey, array_keys($row));
-            $primaryKeyIsFilled = count($filledKeys) === count($primaryKey);
+            $missingPrimaryKeyColumns = array_diff_key($primaryKey, $filledKeys);
 
-            if ($primaryKeyIsFilled) {
+            if (count($missingPrimaryKeyColumns) === 0) {
                 $primary = array_intersect_key($row, array_flip($primaryKey));
-            } elseif ($id && count($primaryKey) === 1) {
-                $primary[$primaryKey[0]] = $id;
+            } elseif (count($missingPrimaryKeyColumns) === 1) {
+                $primary = array_intersect_key($row, array_flip($primaryKey));
+                $missingColumn = reset($missingPrimaryKeyColumns);
+                $primary[$missingColumn] = $id;
             } else {
                 foreach ($primaryKey as $column) {
                     if (isset($row[$column])) {

tests/data/dumps/mysql.sql

@@ -104,6 +104,13 @@ CREATE INDEX counter ON `auto_increment_not_on_pk` (counter);
 ALTER TABLE `auto_increment_not_on_pk`
     MODIFY counter int AUTO_INCREMENT;
 
+
+CREATE TABLE `auto_increment_on_composite_pk` (
+    `id` int(11) NOT NULL,
+    `counter` int(11) AUTO_INCREMENT NOT NULL,
+PRIMARY KEY (`id`, `counter`)
+) ENGINE=MyISAM DEFAULT CHARSET=utf8;
+
 CREATE TABLE `empty_table` (
   `id` int(11) NOT NULL AUTO_INCREMENT,
   `field` varchar(255),

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

@@ -51,7 +51,7 @@ public function testConnectionIsResetOnEveryTestWhenReconnectIsTrue()
 
         // Simulate a test that runs
         $this->module->_before($testCase1);
-        
+
         $connection1 = $this->module->dbh->query('SELECT CONNECTION_ID()')->fetch(PDO::FETCH_COLUMN);
         $this->module->_after($testCase1);
 
@@ -83,7 +83,7 @@ public function testInitialQueriesAreExecuted()
         ];
         $this->module->_reconfigure($config);
         $this->module->_before(Stub::makeEmpty(TestInterface::class));
-        
+
         $usedDatabaseName = $this->module->dbh->query('SELECT DATABASE();')->fetch(PDO::FETCH_COLUMN);
 
         $this->assertSame($dbName, $usedDatabaseName);
@@ -113,4 +113,16 @@ public function testHaveInDatabaseAutoIncrementOnANonPrimaryKey()
 
         $this->module->dontSeeInDatabase('auto_increment_not_on_pk', $testData);
     }
+
+    public function testHaveInDatabaseAutoIncrementOnCompositePrimaryKey()
+    {
+        $testData = [
+            'id' => 777,
+        ];
+        $this->module->haveInDatabase('auto_increment_on_composite_pk', $testData);
+        $this->module->seeInDatabase('auto_increment_on_composite_pk', $testData);
+        $this->module->_after(Stub::makeEmpty(TestInterface::class));
+
+        $this->module->dontSeeInDatabase('auto_increment_on_composite_pk', $testData);
+    }
 }