CommunityID/libs/Zend/Test/PHPUnit/Db/DataSet/QueryTable.php

86 lines
2.9 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: QueryTable.php 17117 2009-07-26 09:41:26Z beberlei $
*/
require_once "PHPUnit/Extensions/Database/DataSet/QueryTable.php";
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-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_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
);
}
}
}