web/enmi/Zend/Test/PHPUnit/Db/DataSet/DbTableDataSet.php
changeset 19 1c2f13fd785c
parent 0 4eba9c11703f
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/web/enmi/Zend/Test/PHPUnit/Db/DataSet/DbTableDataSet.php	Thu Jan 20 19:30:54 2011 +0100
@@ -0,0 +1,103 @@
+<?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);
+    }
+}
\ No newline at end of file