mirror of
https://github.com/yiisoft/yii2.git
synced 2025-11-16 06:17:56 +08:00
112 lines
2.9 KiB
PHP
112 lines
2.9 KiB
PHP
<?php
|
|
/**
|
|
* @link http://www.yiiframework.com/
|
|
* @copyright Copyright (c) 2008 Yii Software LLC
|
|
* @license http://www.yiiframework.com/license/
|
|
*/
|
|
|
|
namespace yii\test;
|
|
|
|
use Yii;
|
|
use yii\base\ArrayAccessTrait;
|
|
use yii\base\InvalidConfigException;
|
|
|
|
/**
|
|
* BaseActiveFixture is the base class for fixture classes that support accessing fixture data as ActiveRecord objects.
|
|
*
|
|
* @author Qiang Xue <qiang.xue@gmail.com>
|
|
* @since 2.0
|
|
*/
|
|
abstract class BaseActiveFixture extends DbFixture implements \IteratorAggregate, \ArrayAccess, \Countable
|
|
{
|
|
use ArrayAccessTrait;
|
|
|
|
/**
|
|
* @var string the AR model class associated with this fixture.
|
|
* @see tableName
|
|
*/
|
|
public $modelClass;
|
|
/**
|
|
* @var boolean whether to create the corresponding DB schema for this fixture.
|
|
* By setting this property to be true, the [[loadSchema()]] method will be called when the fixture is loaded.
|
|
*/
|
|
public $loadSchema = true;
|
|
/**
|
|
* @var boolean whether to load fixture data.
|
|
* By setting this property to be true, the [[loadData()]] method will be called when the fixture is loaded.
|
|
*/
|
|
public $loadData = true;
|
|
/**
|
|
* @var array the data rows. Each array element represents one row of data (column name => column value).
|
|
*/
|
|
public $data = [];
|
|
/**
|
|
* @var \yii\db\ActiveRecord[] the loaded AR models
|
|
*/
|
|
private $_models = [];
|
|
|
|
|
|
/**
|
|
* @inheritdoc
|
|
*/
|
|
public function load()
|
|
{
|
|
if ($this->loadSchema) {
|
|
$this->loadSchema();
|
|
}
|
|
if ($this->loadData) {
|
|
$this->loadData();
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Returns the AR model by the specified model name.
|
|
* A model name is the key of the corresponding data row returned by [[loadData()]].
|
|
* @param string $name the model name.
|
|
* @return null|\yii\db\ActiveRecord the AR model, or null if the model cannot be found in the database
|
|
* @throws \yii\base\InvalidConfigException if [[modelClass]] is not set.
|
|
*/
|
|
public function getModel($name)
|
|
{
|
|
if (!isset($this->data[$name])) {
|
|
return null;
|
|
}
|
|
if (array_key_exists($name, $this->_models)) {
|
|
return $this->_models[$name];
|
|
}
|
|
|
|
if ($this->modelClass === null) {
|
|
throw new InvalidConfigException('The "modelClass" property must be set.');
|
|
}
|
|
$row = $this->data[$name];
|
|
/** @var \yii\db\ActiveRecord $modelClass */
|
|
$modelClass = $this->modelClass;
|
|
/** @var \yii\db\ActiveRecord $model */
|
|
$model = new $modelClass;
|
|
$keys = [];
|
|
foreach ($model->primaryKey() as $key) {
|
|
$keys[$key] = isset($row[$key]) ? $row[$key] : null;
|
|
}
|
|
return $this->_models[$name] = $modelClass::find($keys);
|
|
}
|
|
|
|
/**
|
|
* Creates the database schema needed by this fixture.
|
|
* You may override this method by creating the DB table associated with this fixture
|
|
* and other relevant DB elements, such as views, triggers.
|
|
*/
|
|
protected function loadSchema()
|
|
{
|
|
}
|
|
|
|
/**
|
|
* Loads the fixture data.
|
|
* The default implementation will first reset the DB table and then populate it with the data
|
|
* returned by [[getData()]].
|
|
*/
|
|
protected function loadData()
|
|
{
|
|
return [];
|
|
}
|
|
}
|