-
-
Notifications
You must be signed in to change notification settings - Fork 6.9k
/
FileFixtureTrait.php
64 lines (55 loc) · 1.95 KB
/
FileFixtureTrait.php
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
<?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;
/**
* FileFixtureTrait provides functionalities for loading data fixture from file.
*
* @author Leandro Guindani Gehlen <leandrogehlen@gmail.com>
* @since 2.0.14
*/
trait FileFixtureTrait
{
/**
* @var string the directory path or [path alias](guide:concept-aliases) that contains the fixture data
*/
public $dataDirectory;
/**
* @var string|bool the file path or [path alias](guide:concept-aliases) of the data file that contains the fixture data
* to be returned by [[getData()]]. You can set this property to be false to prevent loading any data.
*/
public $dataFile;
/**
* 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 the data array that will be stored in [[data]] after inserting into the database.
*
* @param string $file the data file path
* @param bool $throwException whether to throw exception if fixture data file does not exist.
* @return array the data to be put into the database
* @throws InvalidConfigException if the specified data file does not exist.
*/
protected function loadData($file, $throwException = true)
{
if ($file === null || $file === false) {
return [];
}
if (basename($file) === $file && $this->dataDirectory !== null) {
$file = $this->dataDirectory . '/' . $file;
}
$file = Yii::getAlias($file);
if (is_file($file)) {
return require $file;
}
if ($throwException) {
throw new InvalidConfigException("Fixture data file does not exist: {$file}");
}
return [];
}
}