Included vendor/ to the project
This commit is contained in:
142
vendor/phpunit/dbunit/src/Database/DataSet.php
vendored
Normal file
142
vendor/phpunit/dbunit/src/Database/DataSet.php
vendored
Normal file
@@ -0,0 +1,142 @@
|
||||
<?php
|
||||
/*
|
||||
* This file is part of DbUnit.
|
||||
*
|
||||
* (c) Sebastian Bergmann <sebastian@phpunit.de>
|
||||
*
|
||||
* For the full copyright and license information, please view the LICENSE
|
||||
* file that was distributed with this source code.
|
||||
*/
|
||||
|
||||
namespace PHPUnit\DbUnit\Database;
|
||||
|
||||
use PHPUnit\DbUnit\DataSet\AbstractDataSet;
|
||||
use PHPUnit\DbUnit\DataSet\DefaultTableMetadata;
|
||||
use PHPUnit\DbUnit\DataSet\ITableMetadata;
|
||||
use PHPUnit\DbUnit\InvalidArgumentException;
|
||||
use PHPUnit\DbUnit\RuntimeException;
|
||||
|
||||
/**
|
||||
* Provides access to a database instance as a data set.
|
||||
*/
|
||||
class DataSet extends AbstractDataSet
|
||||
{
|
||||
/**
|
||||
* An array of ITable objects.
|
||||
*
|
||||
* @var array
|
||||
*/
|
||||
protected $tables = [];
|
||||
|
||||
/**
|
||||
* The database connection this dataset is using.
|
||||
*
|
||||
* @var Connection
|
||||
*/
|
||||
protected $databaseConnection;
|
||||
|
||||
/**
|
||||
* Creates a new dataset using the given database connection.
|
||||
*
|
||||
* @param Connection $databaseConnection
|
||||
*/
|
||||
public function __construct(Connection $databaseConnection)
|
||||
{
|
||||
$this->databaseConnection = $databaseConnection;
|
||||
}
|
||||
|
||||
/**
|
||||
* Creates the query necessary to pull all of the data from a table.
|
||||
*
|
||||
* @param ITableMetadata $tableMetaData
|
||||
*
|
||||
* @return string
|
||||
*/
|
||||
public static function buildTableSelect(ITableMetadata $tableMetaData, Connection $databaseConnection = null)
|
||||
{
|
||||
if ($tableMetaData->getTableName() == '') {
|
||||
$e = new RuntimeException('Empty Table Name');
|
||||
echo $e->getTraceAsString();
|
||||
throw $e;
|
||||
}
|
||||
|
||||
$columns = $tableMetaData->getColumns();
|
||||
if ($databaseConnection) {
|
||||
$columns = array_map([$databaseConnection, 'quoteSchemaObject'], $columns);
|
||||
}
|
||||
$columnList = implode(', ', $columns);
|
||||
|
||||
if ($databaseConnection) {
|
||||
$tableName = $databaseConnection->quoteSchemaObject($tableMetaData->getTableName());
|
||||
} else {
|
||||
$tableName = $tableMetaData->getTableName();
|
||||
}
|
||||
|
||||
$primaryKeys = $tableMetaData->getPrimaryKeys();
|
||||
if ($databaseConnection) {
|
||||
$primaryKeys = array_map([$databaseConnection, 'quoteSchemaObject'], $primaryKeys);
|
||||
}
|
||||
if (count($primaryKeys)) {
|
||||
$orderBy = 'ORDER BY ' . implode(' ASC, ', $primaryKeys) . ' ASC';
|
||||
} else {
|
||||
$orderBy = '';
|
||||
}
|
||||
|
||||
return "SELECT {$columnList} FROM {$tableName} {$orderBy}";
|
||||
}
|
||||
|
||||
/**
|
||||
* Creates an iterator over the tables in the data set. If $reverse is
|
||||
* true a reverse iterator will be returned.
|
||||
*
|
||||
* @param bool $reverse
|
||||
*
|
||||
* @return TableIterator
|
||||
*/
|
||||
protected function createIterator($reverse = false)
|
||||
{
|
||||
return new TableIterator($this->getTableNames(), $this, $reverse);
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns a table object for the given table.
|
||||
*
|
||||
* @param string $tableName
|
||||
*
|
||||
* @return Table
|
||||
*/
|
||||
public function getTable($tableName)
|
||||
{
|
||||
if (!in_array($tableName, $this->getTableNames())) {
|
||||
throw new InvalidArgumentException("$tableName is not a table in the current database.");
|
||||
}
|
||||
|
||||
if (empty($this->tables[$tableName])) {
|
||||
$this->tables[$tableName] = new Table($this->getTableMetaData($tableName), $this->databaseConnection);
|
||||
}
|
||||
|
||||
return $this->tables[$tableName];
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns a table meta data object for the given table.
|
||||
*
|
||||
* @param string $tableName
|
||||
*
|
||||
* @return DefaultTableMetadata
|
||||
*/
|
||||
public function getTableMetaData($tableName)
|
||||
{
|
||||
return new DefaultTableMetadata($tableName, $this->databaseConnection->getMetaData()->getTableColumns($tableName), $this->databaseConnection->getMetaData()->getTablePrimaryKeys($tableName));
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns a list of table names for the database
|
||||
*
|
||||
* @return array
|
||||
*/
|
||||
public function getTableNames()
|
||||
{
|
||||
return $this->databaseConnection->getMetaData()->getTableNames();
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user