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.
282 lines
8.2 KiB
282 lines
8.2 KiB
<?php |
|
|
|
namespace yiiunit\extensions\mongodb; |
|
|
|
use yii\mongodb\Query; |
|
|
|
/** |
|
* @group mongodb |
|
*/ |
|
class QueryRunTest extends MongoDbTestCase |
|
{ |
|
protected function setUp() |
|
{ |
|
parent::setUp(); |
|
$this->setUpTestRows(); |
|
} |
|
|
|
protected function tearDown() |
|
{ |
|
$this->dropCollection('customer'); |
|
parent::tearDown(); |
|
} |
|
|
|
/** |
|
* Sets up test rows. |
|
*/ |
|
protected function setUpTestRows() |
|
{ |
|
$collection = $this->getConnection()->getCollection('customer'); |
|
$rows = []; |
|
for ($i = 1; $i <= 10; $i++) { |
|
$rows[] = [ |
|
'name' => 'name' . $i, |
|
'address' => 'address' . $i, |
|
'avatar' => [ |
|
'width' => 50 + $i, |
|
'height' => 100 + $i, |
|
'url' => 'http://some.url/' . $i, |
|
], |
|
]; |
|
} |
|
$collection->batchInsert($rows); |
|
} |
|
|
|
// Tests : |
|
|
|
public function testAll() |
|
{ |
|
$connection = $this->getConnection(); |
|
$query = new Query; |
|
$rows = $query->from('customer')->all($connection); |
|
$this->assertEquals(10, count($rows)); |
|
} |
|
|
|
public function testDirectMatch() |
|
{ |
|
$connection = $this->getConnection(); |
|
$query = new Query; |
|
$rows = $query->from('customer') |
|
->where(['name' => 'name1']) |
|
->all($connection); |
|
$this->assertEquals(1, count($rows)); |
|
$this->assertEquals('name1', $rows[0]['name']); |
|
} |
|
|
|
public function testIndexBy() |
|
{ |
|
$connection = $this->getConnection(); |
|
$query = new Query; |
|
$rows = $query->from('customer') |
|
->indexBy('name') |
|
->all($connection); |
|
$this->assertEquals(10, count($rows)); |
|
$this->assertNotEmpty($rows['name1']); |
|
} |
|
|
|
public function testInCondition() |
|
{ |
|
$connection = $this->getConnection(); |
|
$query = new Query; |
|
$rows = $query->from('customer') |
|
->where([ |
|
'name' => ['name1', 'name5'] |
|
]) |
|
->all($connection); |
|
$this->assertEquals(2, count($rows)); |
|
$this->assertEquals('name1', $rows[0]['name']); |
|
$this->assertEquals('name5', $rows[1]['name']); |
|
} |
|
|
|
public function testOrCondition() |
|
{ |
|
$connection = $this->getConnection(); |
|
$query = new Query; |
|
$rows = $query->from('customer') |
|
->where(['name' => 'name1']) |
|
->orWhere(['address' => 'address5']) |
|
->all($connection); |
|
$this->assertEquals(2, count($rows)); |
|
$this->assertEquals('name1', $rows[0]['name']); |
|
$this->assertEquals('address5', $rows[1]['address']); |
|
} |
|
|
|
public function testCombinedInAndCondition() |
|
{ |
|
$connection = $this->getConnection(); |
|
$query = new Query; |
|
$rows = $query->from('customer') |
|
->where([ |
|
'name' => ['name1', 'name5'] |
|
]) |
|
->andWhere(['name' => 'name1']) |
|
->all($connection); |
|
$this->assertEquals(1, count($rows)); |
|
$this->assertEquals('name1', $rows[0]['name']); |
|
} |
|
|
|
public function testCombinedInLikeAndCondition() |
|
{ |
|
$connection = $this->getConnection(); |
|
$query = new Query; |
|
$rows = $query->from('customer') |
|
->where([ |
|
'name' => ['name1', 'name5', 'name10'] |
|
]) |
|
->andWhere(['LIKE', 'name', 'me1']) |
|
->andWhere(['name' => 'name10']) |
|
->all($connection); |
|
$this->assertEquals(1, count($rows)); |
|
$this->assertEquals('name10', $rows[0]['name']); |
|
} |
|
|
|
public function testNestedCombinedInAndCondition() |
|
{ |
|
$connection = $this->getConnection(); |
|
$query = new Query; |
|
$rows = $query->from('customer') |
|
->where([ |
|
'and', |
|
['name' => ['name1', 'name2', 'name3']], |
|
['name' => 'name1'] |
|
]) |
|
->orWhere([ |
|
'and', |
|
['name' => ['name4', 'name5', 'name6']], |
|
['name' => 'name6'] |
|
]) |
|
->all($connection); |
|
$this->assertEquals(2, count($rows)); |
|
$this->assertEquals('name1', $rows[0]['name']); |
|
$this->assertEquals('name6', $rows[1]['name']); |
|
} |
|
|
|
public function testOrder() |
|
{ |
|
$connection = $this->getConnection(); |
|
|
|
$query = new Query; |
|
$rows = $query->from('customer') |
|
->orderBy(['name' => SORT_DESC]) |
|
->all($connection); |
|
$this->assertEquals('name9', $rows[0]['name']); |
|
|
|
$query = new Query; |
|
$rows = $query->from('customer') |
|
->orderBy(['avatar.height' => SORT_DESC]) |
|
->all($connection); |
|
$this->assertEquals('name10', $rows[0]['name']); |
|
} |
|
|
|
public function testMatchPlainId() |
|
{ |
|
$connection = $this->getConnection(); |
|
$query = new Query; |
|
$row = $query->from('customer')->one($connection); |
|
$query = new Query; |
|
$rows = $query->from('customer') |
|
->where(['_id' => $row['_id']->__toString()]) |
|
->all($connection); |
|
$this->assertEquals(1, count($rows)); |
|
} |
|
|
|
public function testRegex() |
|
{ |
|
$connection = $this->getConnection(); |
|
$query = new Query; |
|
$rows = $query->from('customer') |
|
->where(['REGEX', 'name', '/me1/']) |
|
->all($connection); |
|
$this->assertEquals(2, count($rows)); |
|
$this->assertEquals('name1', $rows[0]['name']); |
|
$this->assertEquals('name10', $rows[1]['name']); |
|
} |
|
|
|
public function testLike() |
|
{ |
|
$connection = $this->getConnection(); |
|
|
|
$query = new Query; |
|
$rows = $query->from('customer') |
|
->where(['LIKE', 'name', 'me1']) |
|
->all($connection); |
|
$this->assertEquals(2, count($rows)); |
|
$this->assertEquals('name1', $rows[0]['name']); |
|
$this->assertEquals('name10', $rows[1]['name']); |
|
|
|
$query = new Query; |
|
$rowsUppercase = $query->from('customer') |
|
->where(['LIKE', 'name', 'ME1']) |
|
->all($connection); |
|
$this->assertEquals($rows, $rowsUppercase); |
|
} |
|
|
|
public function testModify() |
|
{ |
|
$connection = $this->getConnection(); |
|
|
|
$query = new Query(); |
|
|
|
$searchName = 'name5'; |
|
$newName = 'new name'; |
|
$row = $query->from('customer') |
|
->where(['name' => $searchName]) |
|
->modify(['$set' => ['name' => $newName]], ['new' => false], $connection); |
|
$this->assertEquals($searchName, $row['name']); |
|
|
|
$searchName = 'name7'; |
|
$newName = 'new name'; |
|
$row = $query->from('customer') |
|
->where(['name' => $searchName]) |
|
->modify(['$set' => ['name' => $newName]], ['new' => true], $connection); |
|
$this->assertEquals($newName, $row['name']); |
|
|
|
$row = $query->from('customer') |
|
->where(['name' => 'not existing name']) |
|
->modify(['$set' => ['name' => 'new name']], ['new' => false], $connection); |
|
$this->assertNull($row); |
|
} |
|
|
|
/** |
|
* @see https://github.com/yiisoft/yii2/issues/4879 |
|
* |
|
* @depends testInCondition |
|
*/ |
|
public function testInConditionIgnoreKeys() |
|
{ |
|
$connection = $this->getConnection(); |
|
$query = new Query; |
|
$rows = $query->from('customer') |
|
/*->where([ |
|
'name' => [ |
|
0 => 'name1', |
|
15 => 'name5' |
|
] |
|
])*/ |
|
->where(['in', 'name', [ |
|
10 => 'name1', |
|
15 => 'name5' |
|
]]) |
|
->all($connection); |
|
$this->assertEquals(2, count($rows)); |
|
$this->assertEquals('name1', $rows[0]['name']); |
|
$this->assertEquals('name5', $rows[1]['name']); |
|
} |
|
|
|
/** |
|
* @see https://github.com/yiisoft/yii2/issues/7010 |
|
*/ |
|
public function testSelect() |
|
{ |
|
$connection = $this->getConnection(); |
|
$query = new Query; |
|
$rows = $query->from('customer') |
|
->select(['name' => true, '_id' => false]) |
|
->limit(1) |
|
->all($connection); |
|
$row = array_pop($rows); |
|
$this->assertArrayHasKey('name', $row); |
|
$this->assertArrayNotHasKey('address', $row); |
|
$this->assertArrayNotHasKey('_id', $row); |
|
} |
|
}
|
|
|