Viewing file:  Truncate.php (4.35 KB) -rw-rw-rw-
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
<?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-2009 Zend Technologies USA Inc. (http://www.zend.com)
* @license http://framework.zend.com/license/new-bsd New BSD License
* @version $Id: Truncate.php 19106 2009-11-20 17:15:30Z beberlei $
*/
require_once "PHPUnit/Extensions/Database/Operation/IDatabaseOperation.php";
require_once "PHPUnit/Extensions/Database/DB/IDatabaseConnection.php";
require_once "PHPUnit/Extensions/Database/DataSet/IDataSet.php";
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-2009 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
);
}
}
|