103 lines
3.2 KiB
PHP
103 lines
3.2 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: DbTableDataSet.php 16607 2009-07-09 21:51:46Z beberlei $
|
|
*/
|
|
|
|
require_once "PHPUnit/Extensions/Database/DataSet/QueryDataSet.php";
|
|
|
|
require_once "PHPUnit/Extensions/Database/DB/IDatabaseConnection.php";
|
|
|
|
/**
|
|
* @see Zend_Test_PHPUnit_Db_DataSet_DbTable
|
|
*/
|
|
require_once "Zend/Test/PHPUnit/Db/DataSet/DbTable.php";
|
|
|
|
/**
|
|
* Aggregate several Zend_Db_Table instances into a dataset.
|
|
*
|
|
* @uses Zend_Db_Table
|
|
* @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_DataSet_DbTableDataSet extends PHPUnit_Extensions_Database_DataSet_AbstractDataSet
|
|
{
|
|
/**
|
|
* @var array
|
|
*/
|
|
protected $tables = array();
|
|
|
|
/**
|
|
* Add a Table dataset representation by specifiying an arbitrary select query.
|
|
*
|
|
* By default a select * will be done on the given tablename.
|
|
*
|
|
* @param Zend_Db_Table_Abstract $table
|
|
* @param string|Zend_Db_Select $query
|
|
* @param string $where
|
|
* @param string $order
|
|
* @param string $count
|
|
* @param string $offset
|
|
*/
|
|
public function addTable(Zend_Db_Table_Abstract $table, $where = null, $order = null, $count = null, $offset = null)
|
|
{
|
|
$tableName = $table->info('name');
|
|
$this->tables[$tableName] = new Zend_Test_PHPUnit_Db_DataSet_DbTable($table, $where, $order, $count, $offset);
|
|
}
|
|
|
|
/**
|
|
* Creates an iterator over the tables in the data set. If $reverse is
|
|
* true a reverse iterator will be returned.
|
|
*
|
|
* @param bool $reverse
|
|
* @return PHPUnit_Extensions_Database_DB_TableIterator
|
|
*/
|
|
protected function createIterator($reverse = FALSE)
|
|
{
|
|
return new PHPUnit_Extensions_Database_DataSet_DefaultTableIterator($this->tables, $reverse);
|
|
}
|
|
|
|
/**
|
|
* Returns a table object for the given table.
|
|
*
|
|
* @param string $tableName
|
|
* @return PHPUnit_Extensions_Database_DB_Table
|
|
*/
|
|
public function getTable($tableName)
|
|
{
|
|
if (!isset($this->tables[$tableName])) {
|
|
throw new InvalidArgumentException("$tableName is not a table in the current database.");
|
|
}
|
|
|
|
return $this->tables[$tableName];
|
|
}
|
|
|
|
/**
|
|
* Returns a list of table names for the database
|
|
*
|
|
* @return Array
|
|
*/
|
|
public function getTableNames()
|
|
{
|
|
return array_keys($this->tables);
|
|
}
|
|
} |