web/Zend/Test/PHPUnit/Db/DataSet/QueryTable.php
changeset 0 4eba9c11703f
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/web/Zend/Test/PHPUnit/Db/DataSet/QueryTable.php	Mon Dec 13 18:29:26 2010 +0100
@@ -0,0 +1,91 @@
+<?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: QueryTable.php 20096 2010-01-06 02:05:09Z bkarwin $
+ */
+
+/**
+ * @see PHPUnit_Extensions_Database_DataSet_QueryTable
+ */
+require_once "PHPUnit/Extensions/Database/DataSet/QueryTable.php";
+
+/**
+ * @see PHPUnit_Extensions_Database_DB_IDatabaseConnection
+ */
+require_once "PHPUnit/Extensions/Database/DB/IDatabaseConnection.php";
+
+/**
+ * Represent a PHPUnit Database Extension table with Queries using a Zend_Db adapter for assertion against other tables.
+ *
+ * @uses       PHPUnit_Extensions_Database_DataSet_QueryTable
+ * @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_QueryTable extends PHPUnit_Extensions_Database_DataSet_QueryTable
+{
+    /**
+     * Creates a new database query table object.
+     *
+     * @param string $table_name
+     * @param string $query
+     * @param PHPUnit_Extensions_Database_DB_IDatabaseConnection $databaseConnection
+     */
+    public function __construct($tableName, $query, PHPUnit_Extensions_Database_DB_IDatabaseConnection $databaseConnection)
+    {
+        if( !($databaseConnection instanceof Zend_Test_PHPUnit_Db_Connection) ) {
+            require_once "Zend/Test/PHPUnit/Db/Exception.php";
+            throw new Zend_Test_PHPUnit_Db_Exception("Zend_Test_PHPUnit_Db_DataSet_QueryTable only works with Zend_Test_PHPUnit_Db_Connection connections-");
+        }
+        parent::__construct($tableName, $query, $databaseConnection);
+    }
+
+    /**
+     * Load data from the database.
+     *
+     * @return void
+     */
+    protected function loadData()
+    {
+        if($this->data === null) {
+            $stmt = $this->databaseConnection->getConnection()->query($this->query);
+            $this->data = $stmt->fetchAll(Zend_Db::FETCH_ASSOC);
+        }
+    }
+
+    /**
+     * Create Table Metadata
+     */
+    protected function createTableMetaData()
+    {
+        if ($this->tableMetaData === NULL)
+        {
+            $this->loadData();
+            $keys = array();
+            if(count($this->data) > 0) {
+                $keys = array_keys($this->data[0]);
+            }
+            $this->tableMetaData = new PHPUnit_Extensions_Database_DataSet_DefaultTableMetaData(
+                $this->tableName, $keys
+            );
+        }
+    }
+}
\ No newline at end of file