Merge pull request #6 from M1ke/master

Add Exec and Query methods to FakePDO

Author: muglug

.gitignore

@@ -5,3 +5,6 @@ www.pid
 *~
 composer.lock
 .vscode
+/.idea
+composer.phar
+.phpunit.result.cache

src/Php7/FakePdo.php

@@ -1,16 +1,53 @@
 <?php
 namespace Vimeo\MysqlEngine\Php7;
 
-class FakePdo extends \PDO implements \Vimeo\MysqlEngine\FakePdoInterface
+use PDO;
+use Vimeo\MysqlEngine\FakePdoInterface;
+use Vimeo\MysqlEngine\FakePdoTrait;
+
+class FakePdo extends PDO implements FakePdoInterface
 {
-    use \Vimeo\MysqlEngine\FakePdoTrait;
+    use FakePdoTrait;
 
-    /**
-     * @param  string $statement
-     * @param  ?array $options
-     */
-    public function prepare($statement, $options = null)
+	/**
+	 * @param string $statement
+	 * @param array $options
+	 * @return FakePdoStatement
+	 */
+    public function prepare($statement, array $options = [])
     {
         return new FakePdoStatement($this, $statement, $this->real);
     }
+
+	/**
+	 * @param string $statement
+	 * @return int|false
+	 */
+	public function exec($statement)
+	{
+		$statement = trim($statement);
+		if (strpos($statement, 'SET ')===0){
+			return false;
+		}
+
+		$sth = $this->prepare($statement);
+		if ($sth->execute()){
+			return $sth->rowCount();
+		}
+		return false;
+	}
+
+	/**
+	 * @param string $statement
+	 * @param int $mode
+	 * @param null $arg3
+	 * @param array $ctorargs
+	 * @return FakePdoStatement
+	 */
+	public function query($statement, $mode = PDO::ATTR_DEFAULT_FETCH_MODE, $arg3 = null, array $ctorargs = [])
+	{
+		$sth = $this->prepare($statement);
+		$sth->execute();
+		return $sth;
+	}
 }

src/Php8/FakePdo.php

@@ -1,16 +1,52 @@
 <?php
 namespace Vimeo\MysqlEngine\Php8;
 
-class FakePdo extends \PDO implements \Vimeo\MysqlEngine\FakePdoInterface
+use PDO;
+use Vimeo\MysqlEngine\FakePdoInterface;
+use Vimeo\MysqlEngine\FakePdoTrait;
+
+class FakePdo extends \PDO implements FakePdoInterface
 {
-    use \Vimeo\MysqlEngine\FakePdoTrait;
+    use FakePdoTrait;
 
-    /**
-     * @param  string $statement
-     * @param  ?array $options
-     */
-    public function prepare($statement, $options = null)
+	/**
+	 * @param string $statement
+	 * @param array $options
+	 * @return FakePdoStatement
+	 */
+    public function prepare($statement, array $options = [])
     {
         return new FakePdoStatement($this, $statement, $this->real);
     }
+
+	/**
+	 * @param string $statement
+	 * @return int|false
+	 */
+	public function exec($statement)
+	{
+		$statement = trim($statement);
+		if (str_starts_with($statement, 'SET ')){
+			return false;
+		}
+
+		$sth = $this->prepare($statement);
+		if ($sth->execute()){
+			return $sth->rowCount();
+		}
+		return false;
+	}
+
+	/**
+	 * @param string $statement
+	 * @param int|null $mode
+	 * @param mixed ...$fetchModeArgs
+	 * @return FakePdoStatement
+	 */
+	public function query(string $statement, ?int $mode = PDO::ATTR_DEFAULT_FETCH_MODE, mixed ...$fetchModeArgs)
+	{
+		$sth = $this->prepare($statement);
+		$sth->execute();
+		return $sth;
+	}
 }