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
|
<?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: Insert.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 Inserting 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_Insert implements PHPUnit_Extensions_Database_Operation_IDatabaseOperation
{
/**
* @param PHPUnit_Extensions_Database_DB_IDatabaseConnection $connection
* @param PHPUnit_Extensions_Database_DataSet_IDataSet $dataSet
*/
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!");
}
$databaseDataSet = $connection->createDataSet();
$dsIterator = $dataSet->getIterator();
foreach($dsIterator as $table) {
$tableName = $table->getTableMetaData()->getTableName();
$db = $connection->getConnection();
for($i = 0; $i < $table->getRowCount(); $i++) {
$values = $this->buildInsertValues($table, $i);
try {
$db->insert($tableName, $values);
} catch (Exception $e) {
throw new PHPUnit_Extensions_Database_Operation_Exception("INSERT", "INSERT INTO ".$tableName." [..]", $values, $table, $e->getMessage());
}
}
}
}
/**
*
* @param PHPUnit_Extensions_Database_DataSet_ITable $table
* @param int $rowNum
* @return array
*/
protected function buildInsertValues(PHPUnit_Extensions_Database_DataSet_ITable $table, $rowNum)
{
$values = array();
foreach($table->getTableMetaData()->getColumns() as $columnName) {
$values[$columnName] = $table->getValue($rowNum, $columnName);
}
return $values;
}
}
|