Rajout de doctrine/orm
[zf2.biz/application_blanche.git] / vendor / doctrine / dbal / tests / Doctrine / Tests / Mocks / ConnectionMock.php
diff --git a/vendor/doctrine/dbal/tests/Doctrine/Tests/Mocks/ConnectionMock.php b/vendor/doctrine/dbal/tests/Doctrine/Tests/Mocks/ConnectionMock.php
new file mode 100644 (file)
index 0000000..c1c84d1
--- /dev/null
@@ -0,0 +1,92 @@
+<?php
+
+namespace Doctrine\Tests\Mocks;
+
+class ConnectionMock extends \Doctrine\DBAL\Connection
+{
+    private $_fetchOneResult;
+    private $_platformMock;
+    private $_lastInsertId = 0;
+    private $_inserts = array();
+
+    public function __construct(array $params, $driver, $config = null, $eventManager = null)
+    {
+        $this->_platformMock = new DatabasePlatformMock();
+
+        parent::__construct($params, $driver, $config, $eventManager);
+
+        // Override possible assignment of platform to database platform mock
+        $this->_platform = $this->_platformMock;
+    }
+
+    /**
+     * @override
+     */
+    public function getDatabasePlatform()
+    {
+        return $this->_platformMock;
+    }
+
+    /**
+     * @override
+     */
+    public function insert($tableName, array $data, array $types = array())
+    {
+        $this->_inserts[$tableName][] = $data;
+    }
+
+    /**
+     * @override
+     */
+    public function lastInsertId($seqName = null)
+    {
+        return $this->_lastInsertId;
+    }
+
+    /**
+     * @override
+     */
+    public function fetchColumn($statement, array $params = array(), $colnum = 0)
+    {
+        return $this->_fetchOneResult;
+    }
+
+    /**
+     * @override
+     */
+    public function quote($input, $type = null)
+    {
+        if (is_string($input)) {
+            return "'" . $input . "'";
+        }
+        return $input;
+    }
+
+    /* Mock API */
+
+    public function setFetchOneResult($fetchOneResult)
+    {
+        $this->_fetchOneResult = $fetchOneResult;
+    }
+
+    public function setDatabasePlatform($platform)
+    {
+        $this->_platformMock = $platform;
+    }
+
+    public function setLastInsertId($id)
+    {
+        $this->_lastInsertId = $id;
+    }
+
+    public function getInserts()
+    {
+        return $this->_inserts;
+    }
+
+    public function reset()
+    {
+        $this->_inserts = array();
+        $this->_lastInsertId = 0;
+    }
+}
\ No newline at end of file