QueryRunTest.php 3.23 KB
Newer Older
1 2
<?php

3
namespace yiiunit\extensions\mongodb;
4

5
use yii\mongodb\Query;
6 7

/**
8
 * @group mongodb
9
 */
10
class QueryRunTest extends MongoDbTestCase
11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34
{
	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,
35 36 37 38 39
				'avatar' => [
					'width' => 50 + $i,
					'height' => 100 + $i,
					'url' => 'http://some.url/' . $i,
				],
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 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82
			];
		}
		$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([
83
				'name' => ['name1', 'name5']
84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106
			])
			->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 testOrder()
	{
		$connection = $this->getConnection();
107

108 109 110 111 112
		$query = new Query;
		$rows = $query->from('customer')
			->orderBy(['name' => SORT_DESC])
			->all($connection);
		$this->assertEquals('name9', $rows[0]['name']);
113 114 115 116 117 118

		$query = new Query;
		$rows = $query->from('customer')
			->orderBy(['avatar.height' => SORT_DESC])
			->all($connection);
		$this->assertEquals('name10', $rows[0]['name']);
119
	}
120 121 122 123 124 125 126 127 128 129 130 131

	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));
	}
132 133 134 135 136 137 138 139 140 141 142 143

	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']);
	}
144
}