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.
115 lines
3.2 KiB
115 lines
3.2 KiB
<?php |
|
namespace yiiunit\framework\rbac; |
|
|
|
use Yii; |
|
use yii\console\Application; |
|
use yii\console\Controller; |
|
use yii\db\Connection; |
|
use yii\rbac\DbManager; |
|
|
|
/** |
|
* DbManagerTestCase |
|
*/ |
|
abstract class DbManagerTestCase extends ManagerTestCase |
|
{ |
|
protected static $database; |
|
protected static $driverName = 'mysql'; |
|
|
|
/** |
|
* @var Connection |
|
*/ |
|
protected static $db; |
|
|
|
protected static function runConsoleAction($route, $params = []) |
|
{ |
|
if (Yii::$app === null) { |
|
new Application([ |
|
'id' => 'Migrator', |
|
'basePath' => '@yiiunit', |
|
'components' => [ |
|
'db' => static::getConnection(), |
|
'authManager' => '\yii\rbac\DbManager', |
|
], |
|
]); |
|
} |
|
|
|
ob_start(); |
|
$result = Yii::$app->runAction('migrate/up', ['migrationPath' => '@yii/rbac/migrations/', 'interactive' => false]); |
|
echo "Result is ".$result; |
|
if ($result !== Controller::EXIT_CODE_NORMAL) { |
|
ob_end_flush(); |
|
} else { |
|
ob_end_clean(); |
|
} |
|
} |
|
|
|
public static function setUpBeforeClass() |
|
{ |
|
parent::setUpBeforeClass(); |
|
$databases = static::getParam('databases'); |
|
static::$database = $databases[static::$driverName]; |
|
$pdo_database = 'pdo_' . static::$driverName; |
|
|
|
if (!extension_loaded('pdo') || !extension_loaded($pdo_database)) { |
|
static::markTestSkipped('pdo and ' . $pdo_database . ' extension are required.'); |
|
} |
|
|
|
static::runConsoleAction('migrate/up', ['migrationPath' => '@yii/rbac/migrations/', 'interactive' => false]); |
|
} |
|
|
|
public static function tearDownAfterClass() |
|
{ |
|
static::runConsoleAction('migrate/down', ['migrationPath' => '@yii/rbac/migrations/', 'interactive' => false]); |
|
if (static::$db) { |
|
static::$db->close(); |
|
} |
|
Yii::$app = null; |
|
parent::tearDownAfterClass(); |
|
} |
|
|
|
protected function setUp() |
|
{ |
|
parent::setUp(); |
|
$this->auth = $this->createManager(); |
|
} |
|
|
|
protected function tearDown() |
|
{ |
|
parent::tearDown(); |
|
$this->auth->removeAll(); |
|
} |
|
|
|
/** |
|
* @throws \yii\base\InvalidParamException |
|
* @throws \yii\db\Exception |
|
* @throws \yii\base\InvalidConfigException |
|
* @return \yii\db\Connection |
|
*/ |
|
public static function getConnection() |
|
{ |
|
if (static::$db == null) { |
|
$db = new Connection; |
|
$db->dsn = static::$database['dsn']; |
|
if (isset(static::$database['username'])) { |
|
$db->username = static::$database['username']; |
|
$db->password = static::$database['password']; |
|
} |
|
if (isset(static::$database['attributes'])) { |
|
$db->attributes = static::$database['attributes']; |
|
} |
|
if (!$db->isActive) { |
|
$db->open(); |
|
} |
|
static::$db = $db; |
|
} |
|
return static::$db; |
|
} |
|
|
|
/** |
|
* @return \yii\rbac\ManagerInterface |
|
*/ |
|
protected function createManager() |
|
{ |
|
return new DbManager(['db' => $this->getConnection()]); |
|
} |
|
}
|
|
|