1
0
Fork 0
MySQLDumper/library/Zend/Test/PHPUnit/Db/Operation/Truncate.php
2011-06-10 21:55:32 +00:00

130 Zeilen
Kein EOL
4,6 KiB
PHP

<?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-2011 Zend Technologies USA Inc. (http://www.zend.com)
* @license http://framework.zend.com/license/new-bsd New BSD License
* @version $Id$
*/
/**
* @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-2011 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, true);
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
);
}
}