yii2-bootstrap/tests/unit/extensions/mongo/ActiveRecordTest.php

<?php

namespace yiiunit\extensions\mongo;

use yii\mongo\ActiveQuery;
use yiiunit\data\ar\mongo\ActiveRecord;
use yiiunit\data\ar\mongo\Customer;

/**
 * @group mongo
 */
class ActiveRecordTest extends MongoTestCase
{
	/**
	 * @var array[] list of test rows.
	 */
	protected $testRows = [];

	protected function setUp()
	{
		parent::setUp();
		ActiveRecord::$db = $this->getConnection();
		$this->setUpTestRows();
	}

	protected function tearDown()
	{
		$this->dropCollection(Customer::collectionName());
		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,
				'email' => 'email' . $i,
				'address' => 'address' . $i,
				'status' => $i,
			];
		}
		$collection->batchInsert($rows);
		$this->testRows = $rows;
	}

	// Tests :

	public function testFind()
	{
		// find one
		$result = Customer::find();
		$this->assertTrue($result instanceof ActiveQuery);
		$customer = $result->one();
		$this->assertTrue($customer instanceof Customer);

		// find all
		$customers = Customer::find()->all();
		$this->assertEquals(10, count($customers));
		$this->assertTrue($customers[0] instanceof Customer);
		$this->assertTrue($customers[1] instanceof Customer);

		// find by _id
		$testId = $this->testRows[0]['_id'];
		$customer = Customer::find($testId);
		$this->assertTrue($customer instanceof Customer);
		$this->assertEquals($testId, $customer->_id);

		// find by column values
		$customer = Customer::find(['name' => 'name5']);
		$this->assertTrue($customer instanceof Customer);
		$this->assertEquals($this->testRows[4]['_id'], $customer->_id);
		$this->assertEquals('name5', $customer->name);
		$customer = Customer::find(['name' => 'unexisting name']);
		$this->assertNull($customer);

		// find by attributes
		$customer = Customer::find()->where(['status' => 4])->one();
		$this->assertTrue($customer instanceof Customer);
		$this->assertEquals(4, $customer->status);

		// find count, sum, average, min, max, distinct
		$this->assertEquals(10, Customer::find()->count());
		$this->assertEquals(1, Customer::find()->where(['status' => 2])->count());
		$this->assertEquals((1+10)/2*10, Customer::find()->sum('status'));
		$this->assertEquals((1+10)/2, Customer::find()->average('status'));
		$this->assertEquals(1, Customer::find()->min('status'));
		$this->assertEquals(10, Customer::find()->max('status'));
		$this->assertEquals(range(1, 10), Customer::find()->distinct('status'));

		// scope
		$this->assertEquals(1, Customer::find()->activeOnly()->count());

		// asArray
		$testRow = $this->testRows[2];
		$customer = Customer::find()->where(['_id' => $testRow['_id']])->asArray()->one();
		$this->assertEquals($testRow, $customer);

		// indexBy
		$customers = Customer::find()->indexBy('name')->all();
		$this->assertTrue($customers['name1'] instanceof Customer);
		$this->assertTrue($customers['name2'] instanceof Customer);

		// indexBy callable
		$customers = Customer::find()->indexBy(function ($customer) {
			return $customer->status . '-' . $customer->status;
		})->all();
		$this->assertTrue($customers['1-1'] instanceof Customer);
		$this->assertTrue($customers['2-2'] instanceof Customer);
	}

	public function testInsert()
	{
		$record = new Customer;
		$record->name = 'new name';
		$record->email = 'new email';
		$record->address = 'new address';
		$record->status = 7;

		$this->assertTrue($record->isNewRecord);

		$record->save();

		$this->assertTrue($record->_id instanceof \MongoId);
		$this->assertFalse($record->isNewRecord);
	}

	/**
	 * @depends testInsert
	 */
	public function testUpdate()
	{
		$record = new Customer;
		$record->name = 'new name';
		$record->email = 'new email';
		$record->address = 'new address';
		$record->status = 7;
		$record->save();

		// save
		$record = Customer::find($record->_id);
		$this->assertTrue($record instanceof Customer);
		$this->assertEquals(7, $record->status);
		$this->assertFalse($record->isNewRecord);

		$record->status = 9;
		$record->save();
		$this->assertEquals(9, $record->status);
		$this->assertFalse($record->isNewRecord);
		$record2 = Customer::find($record->_id);
		$this->assertEquals(9, $record2->status);

		// updateAll
		$pk = ['_id' => $record->_id];
		$ret = Customer::updateAll(['status' => 55], $pk);
		$this->assertEquals(1, $ret);
		$record = Customer::find($pk);
		$this->assertEquals(55, $record->status);
	}

	/**
	 * @depends testInsert
	 */
	public function testDelete()
	{
		// delete
		$record = new Customer;
		$record->name = 'new name';
		$record->email = 'new email';
		$record->address = 'new address';
		$record->status = 7;
		$record->save();

		$record = Customer::find($record->_id);
		$record->delete();
		$record = Customer::find($record->_id);
		$this->assertNull($record);

		// deleteAll
		$record = new Customer;
		$record->name = 'new name';
		$record->email = 'new email';
		$record->address = 'new address';
		$record->status = 7;
		$record->save();

		$ret = Customer::deleteAll(['name' => 'new name']);
		$this->assertEquals(1, $ret);
		$records = Customer::find()->where(['name' => 'new name'])->all();
		$this->assertEquals(0, count($records));
	}

	public function testUpdateAllCounters()
	{
		$this->assertEquals(1, Customer::updateAllCounters(['status' => 10], ['status' => 10]));

		$record = Customer::find(['status' => 10]);
		$this->assertNull($record);
	}

	/**
	 * @depends testUpdateAllCounters
	 */
	public function testUpdateCounters()
	{
		$record = Customer::find($this->testRows[9]);

		$originalCounter = $record->status;
		$counterIncrement = 20;
		$record->updateCounters(['status' => $counterIncrement]);
		$this->assertEquals($originalCounter + $counterIncrement, $record->status);

		$refreshedRecord = Customer::find($record->_id);
		$this->assertEquals($originalCounter + $counterIncrement, $refreshedRecord->status);
	}

	/**
	 * @depends testUpdate
	 */
	public function testUpdateNestedAttribute()
	{
		$record = new Customer;
		$record->name = 'new name';
		$record->email = 'new email';
		$record->address = [
			'city' => 'SomeCity',
			'street' => 'SomeStreet',
		];
		$record->status = 7;
		$record->save();

		// save
		$record = Customer::find($record->_id);
		$newAddress = [
			'city' => 'AnotherCity'
		];
		$record->address = $newAddress;
		$record->save();
		$record2 = Customer::find($record->_id);
		$this->assertEquals($newAddress, $record2->address);
	}
}