Merge pull request #15 from Javakky-pxv/feature/default-fetch-mode

Set default fetch style
vimeo · Mar 13, 2021 · 9afb99f · 9afb99f
2 parents 5241021 + 09f533e
commit 9afb99f
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 2 deletions.
diff --git a/src/FakePdoStatementTrait.php b/src/FakePdoStatementTrait.php
@@ -26,7 +26,7 @@ trait FakePdoStatementTrait
     /**
      * @var int
      */
-    private $fetchMode = \PDO::ATTR_DEFAULT_FETCH_MODE;
+    private $fetchMode = \PDO::FETCH_BOTH;
 
     private $fetchArgument;
 

diff --git a/src/FakePdoTrait.php b/src/FakePdoTrait.php
@@ -28,6 +28,9 @@ trait FakePdoTrait
      */
     public $lowercaseResultKeys = false;
 
+    /** @var ?int */
+    private $defaultFetchMode = null;
+
     /**
      * @var bool
      */
@@ -70,6 +73,13 @@ public function setAttribute($key, $value)
             $this->lowercaseResultKeys = true;
         }
 
+        if ($key === \PDO::ATTR_DEFAULT_FETCH_MODE) {
+            if (!is_int($value)) {
+                throw new \PDOException("SQLSTATE[HY000]: General error: invalid fetch mode type");
+            }
+            $this->defaultFetchMode = $value;
+        }
+
         if ($this->real && $key !== \PDO::ATTR_STATEMENT_CLASS) {
             return $this->real->setAttribute($key, $value);
         }

diff --git a/src/Php7/FakePdo.php b/src/Php7/FakePdo.php
@@ -16,7 +16,11 @@ class FakePdo extends PDO implements FakePdoInterface
      */
     public function prepare($statement, $options = [])
     {
-        return new FakePdoStatement($this, $statement, $this->real);
+        $stmt = new FakePdoStatement($this, $statement, $this->real);
+        if ($this->defaultFetchMode) {
+            $stmt->setFetchMode($this->defaultFetchMode);
+        }
+        return $stmt;
     }
 
     /**