You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
91 lines
2.0 KiB
91 lines
2.0 KiB
11 years ago
|
<?php
|
||
|
|
|
||
|
|
namespace yiiunit\extensions\mongo;
|
||
|
|
|
||
|
|
use yii\helpers\FileHelper;
|
||
|
|
use yii\mongo\Connection;
|
||
|
|
use Yii;
|
||
|
|
use yiiunit\TestCase;
|
||
|
|
|
||
|
|
class MongoTestCase extends TestCase
|
||
|
|
{
|
||
|
|
/**
|
||
|
|
* @var array Mongo connection configuration.
|
||
|
|
*/
|
||
|
|
protected $mongoConfig = [
|
||
|
|
'dsn' => 'mongodb://localhost:27017',
|
||
|
|
'dbName' => 'yii2test',
|
||
|
|
];
|
||
|
|
/**
|
||
|
|
* @var Connection Mongo connection instance.
|
||
|
|
*/
|
||
|
|
protected $mongo;
|
||
|
|
|
||
|
|
public static function setUpBeforeClass()
|
||
|
|
{
|
||
|
|
static::loadClassMap();
|
||
|
|
}
|
||
|
|
|
||
|
|
protected function setUp()
|
||
|
|
{
|
||
|
|
parent::setUp();
|
||
|
|
if (!extension_loaded('mongo')) {
|
||
|
|
$this->markTestSkipped('mongo extension required.');
|
||
|
|
}
|
||
|
|
$config = $this->getParam('mongo');
|
||
|
|
if (!empty($config)) {
|
||
|
|
$this->mongoConfig = $config;
|
||
|
|
}
|
||
|
|
$this->mockApplication();
|
||
|
|
static::loadClassMap();
|
||
|
|
}
|
||
|
|
|
||
|
|
protected function tearDown()
|
||
|
|
{
|
||
|
|
if ($this->mongo) {
|
||
|
|
$this->mongo->close();
|
||
|
|
}
|
||
|
|
$this->destroyApplication();
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Adds sphinx extension files to [[Yii::$classPath]],
|
||
|
|
* avoiding the necessity of usage Composer autoloader.
|
||
|
|
*/
|
||
|
|
protected static function loadClassMap()
|
||
|
|
{
|
||
|
|
$baseNameSpace = 'yii/mongo';
|
||
|
|
$basePath = realpath(__DIR__. '/../../../../extensions/mongo');
|
||
|
|
$files = FileHelper::findFiles($basePath);
|
||
|
|
foreach ($files as $file) {
|
||
|
|
$classRelativePath = str_replace($basePath, '', $file);
|
||
|
|
$classFullName = str_replace(['/', '.php'], ['\\', ''], $baseNameSpace . $classRelativePath);
|
||
|
|
Yii::$classMap[$classFullName] = $file;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @param boolean $reset whether to clean up the test database
|
||
|
|
* @param boolean $open whether to open test database
|
||
|
|
* @return \yii\mongo\Connection
|
||
|
|
*/
|
||
|
|
public function getConnection($reset = false, $open = true)
|
||
|
|
{
|
||
|
|
if (!$reset && $this->mongo) {
|
||
|
|
return $this->mongo;
|
||
|
|
}
|
||
|
|
$db = new Connection;
|
||
|
|
$db->dsn = $this->mongoConfig['dsn'];
|
||
|
|
if (isset($this->mongoConfig['dbName'])) {
|
||
|
|
$db->dbName = $this->mongoConfig['dbName'];
|
||
|
|
}
|
||
|
|
if (isset($this->mongoConfig['options'])) {
|
||
|
|
$db->options = $this->mongoConfig['options'];
|
||
|
|
}
|
||
|
|
if ($open) {
|
||
|
|
$db->open();
|
||
|
|
}
|
||
|
|
$this->mongo = $db;
|
||
|
|
return $db;
|
||
|
|
}
|
||
|
|
}
|