1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
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
);
}
}
|
