286 lines
7.6 KiB
PHP
286 lines
7.6 KiB
PHP
<?php
|
|
/**
|
|
* A factory class to manage the life cycle of test fixtures
|
|
*
|
|
* CakePHP(tm) : Rapid Development Framework (http://cakephp.org)
|
|
* Copyright (c) Cake Software Foundation, Inc. (http://cakefoundation.org)
|
|
*
|
|
* Licensed under The MIT License
|
|
* For full copyright and license information, please see the LICENSE.txt
|
|
* Redistributions of files must retain the above copyright notice.
|
|
*
|
|
* @copyright Copyright (c) Cake Software Foundation, Inc. (http://cakefoundation.org)
|
|
* @link http://cakephp.org CakePHP(tm) Project
|
|
* @package Cake.TestSuite.Fixture
|
|
* @since CakePHP(tm) v 2.0
|
|
* @license http://www.opensource.org/licenses/mit-license.php MIT License
|
|
*/
|
|
|
|
App::uses('ConnectionManager', 'Model');
|
|
App::uses('ClassRegistry', 'Utility');
|
|
|
|
/**
|
|
* A factory class to manage the life cycle of test fixtures
|
|
*
|
|
* @package Cake.TestSuite.Fixture
|
|
*/
|
|
class CakeFixtureManager {
|
|
|
|
/**
|
|
* Was this class already initialized?
|
|
*
|
|
* @var boolean
|
|
*/
|
|
protected $_initialized = false;
|
|
|
|
/**
|
|
* Default datasource to use
|
|
*
|
|
* @var DataSource
|
|
*/
|
|
protected $_db = null;
|
|
|
|
/**
|
|
* Holds the fixture classes that where instantiated
|
|
*
|
|
* @var array
|
|
*/
|
|
protected $_loaded = array();
|
|
|
|
/**
|
|
* Holds the fixture classes that where instantiated indexed by class name
|
|
*
|
|
* @var array
|
|
*/
|
|
protected $_fixtureMap = array();
|
|
|
|
/**
|
|
* Inspects the test to look for unloaded fixtures and loads them
|
|
*
|
|
* @param CakeTestCase $test the test case to inspect
|
|
* @return void
|
|
*/
|
|
public function fixturize($test) {
|
|
if (!$this->_initialized) {
|
|
ClassRegistry::config(array('ds' => 'test', 'testing' => true));
|
|
}
|
|
if (empty($test->fixtures) || !empty($this->_processed[get_class($test)])) {
|
|
$test->db = $this->_db;
|
|
return;
|
|
}
|
|
$this->_initDb();
|
|
$test->db = $this->_db;
|
|
if (!is_array($test->fixtures)) {
|
|
$test->fixtures = array_map('trim', explode(',', $test->fixtures));
|
|
}
|
|
if (isset($test->fixtures)) {
|
|
$this->_loadFixtures($test->fixtures);
|
|
}
|
|
|
|
$this->_processed[get_class($test)] = true;
|
|
}
|
|
|
|
/**
|
|
* Initializes this class with a DataSource object to use as default for all fixtures
|
|
*
|
|
* @return void
|
|
*/
|
|
protected function _initDb() {
|
|
if ($this->_initialized) {
|
|
return;
|
|
}
|
|
$db = ConnectionManager::getDataSource('test');
|
|
$db->cacheSources = false;
|
|
$this->_db = $db;
|
|
$this->_initialized = true;
|
|
}
|
|
|
|
/**
|
|
* Looks for fixture files and instantiates the classes accordingly
|
|
*
|
|
* @param array $fixtures the fixture names to load using the notation {type}.{name}
|
|
* @return void
|
|
* @throws UnexpectedValueException when a referenced fixture does not exist.
|
|
*/
|
|
protected function _loadFixtures($fixtures) {
|
|
foreach ($fixtures as $fixture) {
|
|
$fixtureFile = null;
|
|
$fixtureIndex = $fixture;
|
|
if (isset($this->_loaded[$fixture])) {
|
|
continue;
|
|
}
|
|
|
|
if (strpos($fixture, 'core.') === 0) {
|
|
$fixture = substr($fixture, strlen('core.'));
|
|
$fixturePaths[] = CAKE . 'Test' . DS . 'Fixture';
|
|
} elseif (strpos($fixture, 'app.') === 0) {
|
|
$fixture = substr($fixture, strlen('app.'));
|
|
$fixturePaths = array(
|
|
TESTS . 'Fixture'
|
|
);
|
|
} elseif (strpos($fixture, 'plugin.') === 0) {
|
|
$parts = explode('.', $fixture, 3);
|
|
$pluginName = $parts[1];
|
|
$fixture = $parts[2];
|
|
$fixturePaths = array(
|
|
CakePlugin::path(Inflector::camelize($pluginName)) . 'Test' . DS . 'Fixture',
|
|
TESTS . 'Fixture'
|
|
);
|
|
} else {
|
|
$fixturePaths = array(
|
|
TESTS . 'Fixture',
|
|
CAKE . 'Test' . DS . 'Fixture'
|
|
);
|
|
}
|
|
|
|
$loaded = false;
|
|
foreach ($fixturePaths as $path) {
|
|
$className = Inflector::camelize($fixture);
|
|
if (is_readable($path . DS . $className . 'Fixture.php')) {
|
|
$fixtureFile = $path . DS . $className . 'Fixture.php';
|
|
require_once $fixtureFile;
|
|
$fixtureClass = $className . 'Fixture';
|
|
$this->_loaded[$fixtureIndex] = new $fixtureClass();
|
|
$this->_fixtureMap[$fixtureClass] = $this->_loaded[$fixtureIndex];
|
|
$loaded = true;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (!$loaded) {
|
|
$firstPath = str_replace(array(APP, CAKE_CORE_INCLUDE_PATH, ROOT), '', $fixturePaths[0] . DS . $className . 'Fixture.php');
|
|
throw new UnexpectedValueException(__d('cake_dev', 'Referenced fixture class %s (%s) not found', $className, $firstPath));
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Runs the drop and create commands on the fixtures if necessary.
|
|
*
|
|
* @param CakeTestFixture $fixture the fixture object to create
|
|
* @param DataSource $db the datasource instance to use
|
|
* @param boolean $drop whether drop the fixture if it is already created or not
|
|
* @return void
|
|
*/
|
|
protected function _setupTable($fixture, $db = null, $drop = true) {
|
|
if (!$db) {
|
|
if (!empty($fixture->useDbConfig)) {
|
|
$db = ConnectionManager::getDataSource($fixture->useDbConfig);
|
|
} else {
|
|
$db = $this->_db;
|
|
}
|
|
}
|
|
if (!empty($fixture->created) && in_array($db->configKeyName, $fixture->created)) {
|
|
return;
|
|
}
|
|
|
|
$sources = (array)$db->listSources();
|
|
$table = $db->config['prefix'] . $fixture->table;
|
|
$exists = in_array($table, $sources);
|
|
|
|
if ($drop && $exists) {
|
|
$fixture->drop($db);
|
|
$fixture->create($db);
|
|
} elseif (!$exists) {
|
|
$fixture->create($db);
|
|
} else {
|
|
$fixture->created[] = $db->configKeyName;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Creates the fixtures tables and inserts data on them.
|
|
*
|
|
* @param CakeTestCase $test the test to inspect for fixture loading
|
|
* @return void
|
|
*/
|
|
public function load(CakeTestCase $test) {
|
|
if (empty($test->fixtures)) {
|
|
return;
|
|
}
|
|
$fixtures = $test->fixtures;
|
|
if (empty($fixtures) || !$test->autoFixtures) {
|
|
return;
|
|
}
|
|
|
|
foreach ($fixtures as $f) {
|
|
if (!empty($this->_loaded[$f])) {
|
|
$fixture = $this->_loaded[$f];
|
|
$db = ConnectionManager::getDataSource($fixture->useDbConfig);
|
|
$db->begin();
|
|
$this->_setupTable($fixture, $db, $test->dropTables);
|
|
$fixture->insert($db);
|
|
$db->commit();
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Truncates the fixtures tables
|
|
*
|
|
* @param CakeTestCase $test the test to inspect for fixture unloading
|
|
* @return void
|
|
*/
|
|
public function unload(CakeTestCase $test) {
|
|
$fixtures = !empty($test->fixtures) ? $test->fixtures : array();
|
|
foreach (array_reverse($fixtures) as $f) {
|
|
if (isset($this->_loaded[$f])) {
|
|
$fixture = $this->_loaded[$f];
|
|
if (!empty($fixture->created)) {
|
|
foreach ($fixture->created as $ds) {
|
|
$db = ConnectionManager::getDataSource($ds);
|
|
$fixture->truncate($db);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Creates a single fixture table and loads data into it.
|
|
*
|
|
* @param string $name of the fixture
|
|
* @param DataSource $db DataSource instance or leave null to get DataSource from the fixture
|
|
* @param boolean $dropTables Whether or not tables should be dropped and re-created.
|
|
* @return void
|
|
* @throws UnexpectedValueException if $name is not a previously loaded class
|
|
*/
|
|
public function loadSingle($name, $db = null, $dropTables = true) {
|
|
$name .= 'Fixture';
|
|
if (isset($this->_fixtureMap[$name])) {
|
|
$fixture = $this->_fixtureMap[$name];
|
|
if (!$db) {
|
|
$db = ConnectionManager::getDataSource($fixture->useDbConfig);
|
|
}
|
|
$this->_setupTable($fixture, $db, $dropTables);
|
|
$fixture->truncate($db);
|
|
$fixture->insert($db);
|
|
} else {
|
|
throw new UnexpectedValueException(__d('cake_dev', 'Referenced fixture class %s not found', $name));
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Drop all fixture tables loaded by this class
|
|
*
|
|
* This will also close the session, as failing to do so will cause
|
|
* fatal errors with database sessions.
|
|
*
|
|
* @return void
|
|
*/
|
|
public function shutDown() {
|
|
if (session_id()) {
|
|
session_write_close();
|
|
}
|
|
foreach ($this->_loaded as $fixture) {
|
|
if (!empty($fixture->created)) {
|
|
foreach ($fixture->created as $ds) {
|
|
$db = ConnectionManager::getDataSource($ds);
|
|
$fixture->drop($db);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|