fix: pass the column number 0 to the statement

tflori · Jun 13, 2021 · 245bd09 · 245bd09
1 parent e4c1da6
commit 245bd09
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 1 deletion.
diff --git a/src/QueryBuilder/ExecutesQueries.php b/src/QueryBuilder/ExecutesQueries.php
@@ -44,7 +44,9 @@ public function setFetchMode($mode, $classNameObject = null, array $constructorA
             return $this;
         }
 
-        $statement->setFetchMode(...array_filter([$mode, $classNameObject, $constructorArgs]));
+        $statement->setFetchMode(...array_filter([$mode, $classNameObject, $constructorArgs], function ($p) {
+            return $p !== null;
+        }));
         return $this;
     }
 

diff --git a/tests/QueryBuilder/BasicTest.php b/tests/QueryBuilder/BasicTest.php
@@ -300,4 +300,19 @@ public function doesNotPassNull()
 
         $query->setFetchMode(\PDO::FETCH_ASSOC, null);
     }
+
+    /** @test */
+    public function passesColumnNumber0()
+    {
+        if (!extension_loaded('pdo_sqlite')) {
+            $this->markTestSkipped('pdo_sqlite extension required for this test');
+        }
+
+        $em = new EntityManager([
+            EntityManager::OPT_CONNECTION => new DbConfig('sqlite', '/tmp/test.sqlite')
+        ]);
+        $query = $em->query('sqlite_master');
+
+        $query->setFetchMode(\PDO::FETCH_COLUMN, 0);
+    }
 }