ActiveFixture.php 5.5 KB
Newer Older
Qiang Xue committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16
<?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\InvalidConfigException;
use yii\db\TableSchema;

/**
 * ActiveFixture represents a fixture backed up by a [[modelClass|ActiveRecord class]] or a [[tableName|database table]].
 *
Qiang Xue committed
17 18
 * Either [[modelClass]] or [[tableName]] must be set. You should also provide fixture data in the file
 * specified by [[dataFile]] or overriding [[getData()]] if you want to use code to generate the fixture data.
Qiang Xue committed
19
 *
Qiang Xue committed
20 21
 * When the fixture is being loaded, it will first call [[resetTable()]] to remove any existing data in the table.
 * It will then populate the table with the data returned by [[getData()]].
Qiang Xue committed
22 23 24
 *
 * After the fixture is loaded, you can access the loaded data via the [[data]] property. If you set [[modelClass]],
 * you will also be able to retrieve an instance of [[modelClass]] with the populated data via [[getModel()]].
Qiang Xue committed
25
 *
26 27 28
 * @property TableSchema $tableSchema The schema information of the database table associated with this
 * fixture. This property is read-only.
 *
Qiang Xue committed
29 30 31
 * @author Qiang Xue <qiang.xue@gmail.com>
 * @since 2.0
 */
Qiang Xue committed
32
class ActiveFixture extends BaseActiveFixture
Qiang Xue committed
33
{
34 35 36 37 38 39 40 41 42 43 44 45 46
    /**
     * @var string the name of the database table that this fixture is about. If this property is not set,
     * the table name will be determined via [[modelClass]].
     * @see modelClass
     */
    public $tableName;
    /**
     * @var string|boolean the file path or path alias of the data file that contains the fixture data
     * to be returned by [[getData()]]. If this is not set, it will default to `FixturePath/data/TableName.php`,
     * where `FixturePath` stands for the directory containing this fixture class, and `TableName` stands for the
     * name of the table associated with this fixture. You can set this property to be false to prevent loading any data.
     */
    public $dataFile;
47

48 49 50 51
    /**
     * @var TableSchema the table schema for the table associated with this fixture
     */
    private $_table;
Qiang Xue committed
52

53

54 55 56 57 58 59 60 61 62 63
    /**
     * @inheritdoc
     */
    public function init()
    {
        parent::init();
        if (!isset($this->modelClass) && !isset($this->tableName)) {
            throw new InvalidConfigException('Either "modelClass" or "tableName" must be set.');
        }
    }
Carsten Brandt committed
64

65 66 67
    /**
     * Loads the fixture.
     *
68
     * The default implementation will first clean up the table by calling [[resetTable()]].
69 70 71 72 73 74 75
     * It will then populate the table with the data returned by [[getData()]].
     *
     * If you override this method, you should consider calling the parent implementation
     * so that the data returned by [[getData()]] can be populated into the table.
     */
    public function load()
    {
76 77
        $this->resetTable();
        $this->data = [];
78 79 80 81 82 83 84 85 86 87 88 89 90 91
        $table = $this->getTableSchema();
        foreach ($this->getData() as $alias => $row) {
            $this->db->createCommand()->insert($table->fullName, $row)->execute();
            if ($table->sequenceName !== null) {
                foreach ($table->primaryKey as $pk) {
                    if (!isset($row[$pk])) {
                        $row[$pk] = $this->db->getLastInsertID($table->sequenceName);
                        break;
                    }
                }
            }
            $this->data[$alias] = $row;
        }
    }
Qiang Xue committed
92

93 94 95 96 97 98 99 100 101 102 103 104 105 106 107
    /**
     * Returns the fixture data.
     *
     * The default implementation will try to return the fixture data by including the external file specified by [[dataFile]].
     * The file should return an array of data rows (column name => column value), each corresponding to a row in the table.
     *
     * If the data file does not exist, an empty array will be returned.
     *
     * @return array the data rows to be inserted into the database table.
     */
    protected function getData()
    {
        if ($this->dataFile === null) {
            $class = new \ReflectionClass($this);
            $dataFile = dirname($class->getFileName()) . '/data/' . $this->getTableSchema()->fullName . '.php';
Mark committed
108

109 110 111 112 113
            return is_file($dataFile) ? require($dataFile) : [];
        } else {
            return parent::getData();
        }
    }
Qiang Xue committed
114

115 116 117 118 119 120 121 122 123 124 125 126
    /**
     * Removes all existing data from the specified table and resets sequence number to 1 (if any).
     * This method is called before populating fixture data into the table associated with this fixture.
     */
    protected function resetTable()
    {
        $table = $this->getTableSchema();
        $this->db->createCommand()->delete($table->fullName)->execute();
        if ($table->sequenceName !== null) {
            $this->db->createCommand()->resetSequence($table->fullName, 1)->execute();
        }
    }
Qiang Xue committed
127

128
    /**
129
     * @return TableSchema the schema information of the database table associated with this fixture.
130 131 132 133 134 135 136
     * @throws \yii\base\InvalidConfigException if the table does not exist
     */
    public function getTableSchema()
    {
        if ($this->_table !== null) {
            return $this->_table;
        }
Qiang Xue committed
137

138 139 140
        $db = $this->db;
        $tableName = $this->tableName;
        if ($tableName === null) {
141
            /* @var $modelClass \yii\db\ActiveRecord */
142 143 144
            $modelClass = $this->modelClass;
            $tableName = $modelClass::tableName();
        }
Qiang Xue committed
145

146 147 148 149
        $this->_table = $db->getSchema()->getTableSchema($tableName);
        if ($this->_table === null) {
            throw new InvalidConfigException("Table does not exist: {$tableName}");
        }
Qiang Xue committed
150

151 152
        return $this->_table;
    }
Qiang Xue committed
153
}