web/Zend/Test/PHPUnit/Db/Operation/Truncate.php
changeset 0 4eba9c11703f
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/web/Zend/Test/PHPUnit/Db/Operation/Truncate.php	Mon Dec 13 18:29:26 2010 +0100
@@ -0,0 +1,130 @@
+<?php
+/**
+ * Zend Framework
+ *
+ * LICENSE
+ *
+ * This source file is subject to the new BSD license that is bundled
+ * with this package in the file LICENSE.txt.
+ * It is also available through the world-wide-web at this URL:
+ * http://framework.zend.com/license/new-bsd
+ * If you did not receive a copy of the license and are unable to
+ * obtain it through the world-wide-web, please send an email
+ * to license@zend.com so we can send you a copy immediately.
+ *
+ * @category   Zend
+ * @package    Zend_Test
+ * @subpackage PHPUnit
+ * @copyright  Copyright (c) 2005-2010 Zend Technologies USA Inc. (http://www.zend.com)
+ * @license    http://framework.zend.com/license/new-bsd     New BSD License
+ * @version    $Id: Truncate.php 20096 2010-01-06 02:05:09Z bkarwin $
+ */
+
+/**
+ * @see PHPUnit_Extensions_Database_Operation_IDatabaseOperation
+ */
+require_once "PHPUnit/Extensions/Database/Operation/IDatabaseOperation.php";
+
+/**
+ * @see PHPUnit_Extensions_Database_DB_IDatabaseConnection
+ */
+require_once "PHPUnit/Extensions/Database/DB/IDatabaseConnection.php";
+
+/**
+ * @see PHPUnit_Extensions_Database_DataSet_IDataSet
+ */
+require_once "PHPUnit/Extensions/Database/DataSet/IDataSet.php";
+
+/**
+ * @see PHPUnit_Extensions_Database_Operation_Exception
+ */
+require_once "PHPUnit/Extensions/Database/Operation/Exception.php";
+
+/**
+ * @see Zend_Test_PHPUnit_Db_Connection
+ */
+require_once "Zend/Test/PHPUnit/Db/Connection.php";
+
+/**
+ * Operation for Truncating on setup or teardown of a database tester.
+ *
+ * @uses       PHPUnit_Extensions_Database_Operation_IDatabaseOperation
+ * @category   Zend
+ * @package    Zend_Test
+ * @subpackage PHPUnit
+ * @copyright  Copyright (c) 2005-2010 Zend Technologies USA Inc. (http://www.zend.com)
+ * @license    http://framework.zend.com/license/new-bsd     New BSD License
+ */
+class Zend_Test_PHPUnit_Db_Operation_Truncate implements PHPUnit_Extensions_Database_Operation_IDatabaseOperation
+{
+    /**
+     *
+     * @param PHPUnit_Extensions_Database_DB_IDatabaseConnection $connection
+     * @param PHPUnit_Extensions_Database_DataSet_IDataSet $dataSet
+     * @return void
+     */
+    public function execute(PHPUnit_Extensions_Database_DB_IDatabaseConnection $connection, PHPUnit_Extensions_Database_DataSet_IDataSet $dataSet)
+    {
+        if(!($connection instanceof Zend_Test_PHPUnit_Db_Connection)) {
+            require_once "Zend/Test/PHPUnit/Db/Exception.php";
+            throw new Zend_Test_PHPUnit_Db_Exception("Not a valid Zend_Test_PHPUnit_Db_Connection instance, ".get_class($connection)." given!");
+        }
+
+        foreach ($dataSet->getReverseIterator() AS $table) {
+            try {
+                $tableName = $table->getTableMetaData()->getTableName();
+                $this->_truncate($connection->getConnection(), $tableName);
+            } catch (Exception $e) {
+                throw new PHPUnit_Extensions_Database_Operation_Exception('TRUNCATE', 'TRUNCATE '.$tableName.'', array(), $table, $e->getMessage());
+            }
+        }
+    }
+
+    /**
+     * Truncate a given table.
+     *
+     * @param Zend_Db_Adapter_Abstract $db
+     * @param string $tableName
+     * @return void
+     */
+    protected function _truncate(Zend_Db_Adapter_Abstract $db, $tableName)
+    {
+        $tableName = $db->quoteIdentifier($tableName);
+        if($db instanceof Zend_Db_Adapter_Pdo_Sqlite) {
+            $db->query('DELETE FROM '.$tableName);
+        } else if($db instanceof Zend_Db_Adapter_Db2) {
+            /*if(strstr(PHP_OS, "WIN")) {
+                $file = tempnam(sys_get_temp_dir(), "zendtestdbibm_");
+                file_put_contents($file, "");
+                $db->query('IMPORT FROM '.$file.' OF DEL REPLACE INTO '.$tableName);
+                unlink($file);
+            } else {
+                $db->query('IMPORT FROM /dev/null OF DEL REPLACE INTO '.$tableName);
+            }*/
+            require_once "Zend/Exception.php";
+            throw Zend_Exception("IBM Db2 TRUNCATE not supported.");
+        } else if($this->_isMssqlOrOracle($db)) {
+            $db->query('TRUNCATE TABLE '.$tableName);
+        } else if($db instanceof Zend_Db_Adapter_Pdo_Pgsql) {
+            $db->query('TRUNCATE '.$tableName.' CASCADE');
+        } else {
+            $db->query('TRUNCATE '.$tableName);
+        }
+    }
+
+    /**
+     * Detect if an adapter is for Mssql or Oracle Databases.
+     *
+     * @param  Zend_Db_Adapter_Abstract $db
+     * @return bool
+     */
+    private function _isMssqlOrOracle($db)
+    {
+        return (
+            $db instanceof Zend_Db_Adapter_Pdo_Mssql ||
+            $db instanceof Zend_Db_Adapter_Sqlsrv ||
+            $db instanceof Zend_Db_Adapter_Pdo_Oci ||
+            $db instanceof Zend_Db_Adapter_Oracle
+        );
+    }
+}
\ No newline at end of file