Mirrors/yii2
2
0
Fork 0
mirror of https://github.com/yiisoft/yii2.git synced 2026-03-04 22:44:53 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add tests to ensure ActiveQuery is using DB connection set by model (#18286)

Browse Source
This commit is contained in:
Bizley
2020-09-17 19:07:32 +02:00
committed by GitHub
parent c48254cafa
commit 9790ea3a35
1 changed files with 78 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

78
tests/framework/db/ActiveQueryModelConnectionTest.php Normal file
View File

@@ -0,0 +1,78 @@
<?php
/**
* @link http://www.yiiframework.com/
* @copyright Copyright (c) 2008 Yii Software LLC
* @license http://www.yiiframework.com/license/
*/
namespace yiiunit\framework\db;
use yii\db\ActiveQuery;
use yii\db\ActiveRecord as DefaultActiveRecord;
use yiiunit\data\ar\ActiveRecord;
use yiiunit\TestCase;
class ActiveQueryModelConnectionTest extends TestCase
{
private $globalConnection;
private $modelConnection;
protected function setUp()
{
$this->globalConnection = $this->getMockBuilder('yii\db\Connection')->getMock();
$this->modelConnection = $this->getMockBuilder('yii\db\Connection')->getMock();
$this->mockApplication([
'components' => [
'db' => $this->globalConnection
]
]);
ActiveRecord::$db = $this->modelConnection;
}
private function prepareConnectionMock($connection)
{
$command = $this->getMockBuilder('yii\db\Command')->getMock();
$command->method('queryOne')->willReturn(false);
$connection->method('createCommand')->willReturn($command);
$builder = $this->getMockBuilder('yii\db\QueryBuilder')->disableOriginalConstructor()->getMock();
$connection->expects($this->once())->method('getQueryBuilder')->willReturn($builder);
}
public function testEnsureModelConnectionForOne()
{
$this->globalConnection->expects($this->never())->method('getQueryBuilder');
$this->prepareConnectionMock($this->modelConnection);
$query = new ActiveQuery(ActiveRecord::className());
$query->one();
}
public function testEnsureGlobalConnectionForOne()
{
$this->modelConnection->expects($this->never())->method('getQueryBuilder');
$this->prepareConnectionMock($this->globalConnection);
$query = new ActiveQuery(DefaultActiveRecord::className());
$query->one();
}
public function testEnsureModelConnectionForAll()
{
$this->globalConnection->expects($this->never())->method('getQueryBuilder');
$this->prepareConnectionMock($this->modelConnection);
$query = new ActiveQuery(ActiveRecord::className());
$query->all();
}
public function testEnsureGlobalConnectionForAll()
{
$this->modelConnection->expects($this->never())->method('getQueryBuilder');
$this->prepareConnectionMock($this->globalConnection);
$query = new ActiveQuery(DefaultActiveRecord::className());
$query->all();
}
}