--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/web/enmi/Zend/Test/PHPUnit/Db/Operation/Truncate.php Thu Jan 20 19:30:54 2011 +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