<?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: DbTableDataSet.php 20096 2010-01-06 02:05:09Z bkarwin $

 */

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-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_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);

    }

}