1 <?php

     2 /**

     3  * Zend Framework

     4  *

     5  * LICENSE

     6  *

     7  * This source file is subject to the new BSD license that is bundled

     8  * with this package in the file LICENSE.txt.

     9  * It is also available through the world-wide-web at this URL:

    10  * http://framework.zend.com/license/new-bsd

    11  * If you did not receive a copy of the license and are unable to

    12  * obtain it through the world-wide-web, please send an email

    13  * to license@zend.com so we can send you a copy immediately.

    14  *

    15  * @category   Zend

    16  * @package    Zend_Test

    17  * @subpackage PHPUnit

    18  * @copyright  Copyright (c) 2005-2010 Zend Technologies USA Inc. (http://www.zend.com)

    19  * @license    http://framework.zend.com/license/new-bsd     New BSD License

    20  * @version    $Id: Truncate.php 20096 2010-01-06 02:05:09Z bkarwin $

    21  */

    22 

    23 /**

    24  * @see PHPUnit_Extensions_Database_Operation_IDatabaseOperation

    25  */

    26 require_once "PHPUnit/Extensions/Database/Operation/IDatabaseOperation.php";

    27 

    28 /**

    29  * @see PHPUnit_Extensions_Database_DB_IDatabaseConnection

    30  */

    31 require_once "PHPUnit/Extensions/Database/DB/IDatabaseConnection.php";

    32 

    33 /**

    34  * @see PHPUnit_Extensions_Database_DataSet_IDataSet

    35  */

    36 require_once "PHPUnit/Extensions/Database/DataSet/IDataSet.php";

    37 

    38 /**

    39  * @see PHPUnit_Extensions_Database_Operation_Exception

    40  */

    41 require_once "PHPUnit/Extensions/Database/Operation/Exception.php";

    42 

    43 /**

    44  * @see Zend_Test_PHPUnit_Db_Connection

    45  */

    46 require_once "Zend/Test/PHPUnit/Db/Connection.php";

    47 

    48 /**

    49  * Operation for Truncating on setup or teardown of a database tester.

    50  *

    51  * @uses       PHPUnit_Extensions_Database_Operation_IDatabaseOperation

    52  * @category   Zend

    53  * @package    Zend_Test

    54  * @subpackage PHPUnit

    55  * @copyright  Copyright (c) 2005-2010 Zend Technologies USA Inc. (http://www.zend.com)

    56  * @license    http://framework.zend.com/license/new-bsd     New BSD License

    57  */

    58 class Zend_Test_PHPUnit_Db_Operation_Truncate implements PHPUnit_Extensions_Database_Operation_IDatabaseOperation

    59 {

    60     /**

    61      *

    62      * @param PHPUnit_Extensions_Database_DB_IDatabaseConnection $connection

    63      * @param PHPUnit_Extensions_Database_DataSet_IDataSet $dataSet

    64      * @return void

    65      */

    66     public function execute(PHPUnit_Extensions_Database_DB_IDatabaseConnection $connection, PHPUnit_Extensions_Database_DataSet_IDataSet $dataSet)

    67     {

    68         if(!($connection instanceof Zend_Test_PHPUnit_Db_Connection)) {

    69             require_once "Zend/Test/PHPUnit/Db/Exception.php";

    70             throw new Zend_Test_PHPUnit_Db_Exception("Not a valid Zend_Test_PHPUnit_Db_Connection instance, ".get_class($connection)." given!");

    71         }

    72 

    73         foreach ($dataSet->getReverseIterator() AS $table) {

    74             try {

    75                 $tableName = $table->getTableMetaData()->getTableName();

    76                 $this->_truncate($connection->getConnection(), $tableName);

    77             } catch (Exception $e) {

    78                 throw new PHPUnit_Extensions_Database_Operation_Exception('TRUNCATE', 'TRUNCATE '.$tableName.'', array(), $table, $e->getMessage());

    79             }

    80         }

    81     }

    82 

    83     /**

    84      * Truncate a given table.

    85      *

    86      * @param Zend_Db_Adapter_Abstract $db

    87      * @param string $tableName

    88      * @return void

    89      */

    90     protected function _truncate(Zend_Db_Adapter_Abstract $db, $tableName)

    91     {

    92         $tableName = $db->quoteIdentifier($tableName);

    93         if($db instanceof Zend_Db_Adapter_Pdo_Sqlite) {

    94             $db->query('DELETE FROM '.$tableName);

    95         } else if($db instanceof Zend_Db_Adapter_Db2) {

    96             /*if(strstr(PHP_OS, "WIN")) {

    97                 $file = tempnam(sys_get_temp_dir(), "zendtestdbibm_");

    98                 file_put_contents($file, "");

    99                 $db->query('IMPORT FROM '.$file.' OF DEL REPLACE INTO '.$tableName);

   100                 unlink($file);

   101             } else {

   102                 $db->query('IMPORT FROM /dev/null OF DEL REPLACE INTO '.$tableName);

   103             }*/

   104             require_once "Zend/Exception.php";

   105             throw Zend_Exception("IBM Db2 TRUNCATE not supported.");

   106         } else if($this->_isMssqlOrOracle($db)) {

   107             $db->query('TRUNCATE TABLE '.$tableName);

   108         } else if($db instanceof Zend_Db_Adapter_Pdo_Pgsql) {

   109             $db->query('TRUNCATE '.$tableName.' CASCADE');

   110         } else {

   111             $db->query('TRUNCATE '.$tableName);

   112         }

   113     }

   114 

   115     /**

   116      * Detect if an adapter is for Mssql or Oracle Databases.

   117      *

   118      * @param  Zend_Db_Adapter_Abstract $db

   119      * @return bool

   120      */

   121     private function _isMssqlOrOracle($db)

   122     {

   123         return (

   124             $db instanceof Zend_Db_Adapter_Pdo_Mssql ||

   125             $db instanceof Zend_Db_Adapter_Sqlsrv ||

   126             $db instanceof Zend_Db_Adapter_Pdo_Oci ||

   127             $db instanceof Zend_Db_Adapter_Oracle

   128         );

   129     }

   130 }