98 lines
3.7 KiB
PHP
98 lines
3.7 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-2009 Zend Technologies USA Inc. (http://www.zend.com)
|
|
* @license http://framework.zend.com/license/new-bsd New BSD License
|
|
* @version $Id: Truncate.php 16607 2009-07-09 21:51:46Z 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 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
|
|
*/
|
|
private 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);
|
|
}
|
|
} else if($db instanceof Zend_Db_Adapter_Pdo_Mssql) {
|
|
$db->query('TRUNCATE TABLE '.$tableName);
|
|
} else {
|
|
$db->query('TRUNCATE '.$tableName);
|
|
}
|
|
}
|
|
} |