Merge pull request #16 from bronek89/feature/fetch-column-mode

Add support for FETCH_COLUMN in fetch
vimeo · Mar 13, 2021 · 5241021 · 5241021
2 parents 24cee9c + 5aa910d
commit 5241021
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/src/FakePdoStatementTrait.php b/src/FakePdoStatementTrait.php
@@ -336,6 +336,12 @@ public function fetch(
             return \array_values($row);
         }
 
+        if ($fetch_style === \PDO::FETCH_COLUMN) {
+            $this->resultCursor++;
+
+            return \array_values($row)[0] ?? null;
+        }
+
         if ($fetch_style === \PDO::FETCH_BOTH) {
             $this->resultCursor++;
 

diff --git a/tests/EndToEndTest.php b/tests/EndToEndTest.php
@@ -901,6 +901,17 @@ public function testFetchCountForMissingColumn(int $columnIndex)
         $query->fetchColumn($columnIndex);
     }
 
+    public function testFetchWithColumnMode()
+    {
+        $pdo = self::getConnectionToFullDB(false);
+
+        $query = $pdo->prepare('SELECT `id` FROM `video_game_characters` WHERE id=2 LIMIT 1');
+
+        $query->execute();
+
+        self::assertEquals(2, $query->fetch(\PDO::FETCH_COLUMN));
+    }
+
     public function dataProviderTruncateForms(): \Generator
     {
         foreach ([