<?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;
use yiiunit\data\ar\ActiveRecord;
use yii\db\Query;
use yii\db\BatchQueryResult;
use yiiunit\data\ar\Customer;

/**
 * @author Qiang Xue <qiang.xue@gmail.com>
 * @since 2.0
 */
class BatchQueryResultTest extends DatabaseTestCase
{
	public function setUp()
	{
		parent::setUp();
		ActiveRecord::$db = $this->getConnection();
	}

	public function testQuery()
	{
		$db = $this->getConnection();

		// initialize property test
		$query = new Query();
		$query->from('tbl_customer')->orderBy('id');
		$result = $query->batch(2, $db);
		$this->assertTrue($result instanceof BatchQueryResult);
		$this->assertEquals(2, $result->batchSize);
		$this->assertNull($result->dataReader);
		$this->assertTrue($result->query === $query);

		// normal query
		$query = new Query();
		$query->from('tbl_customer')->orderBy('id');
		$allRows = [];
		$batch = $query->batch(2, $db);
		foreach ($batch as $rows) {
			$allRows = array_merge($allRows, $rows);
		}
		$this->assertEquals(3, count($allRows));
		$this->assertEquals('user1', $allRows[0]['name']);
		$this->assertEquals('user2', $allRows[1]['name']);
		$this->assertEquals('user3', $allRows[2]['name']);
		// rewind
		$allRows = [];
		foreach ($batch as $rows) {
			$allRows = array_merge($allRows, $rows);
		}
		$this->assertEquals(3, count($allRows));
		// reset
		$batch->reset();
		$this->assertNull($batch->dataReader);

		// query with index
		$query = new Query();
		$query->from('tbl_customer')->indexBy('name');
		$allRows = [];
		foreach ($query->batch(2, $db) as $rows) {
			$allRows = array_merge($allRows, $rows);
		}
		$this->assertEquals(3, count($allRows));
		$this->assertEquals('address1', $allRows['user1']['address']);
		$this->assertEquals('address2', $allRows['user2']['address']);
		$this->assertEquals('address3', $allRows['user3']['address']);

		// query in batch 1
		$query = new Query();
		$query->from('tbl_customer')->orderBy('id');
		$allRows = [];
		foreach ($query->batch(1, $db) as $rows) {
			$allRows[] = $rows;
		}
		$this->assertEquals(3, count($allRows));
		$this->assertEquals('user1', $allRows[0]['name']);
		$this->assertEquals('user2', $allRows[1]['name']);
		$this->assertEquals('user3', $allRows[2]['name']);

		// each
		$query = new Query();
		$query->from('tbl_customer')->orderBy('id');
		$allRows = [];
		foreach ($query->each($db) as $rows) {
			$allRows[] = $rows;
		}
		$this->assertEquals(3, count($allRows));
		$this->assertEquals('user1', $allRows[0]['name']);
		$this->assertEquals('user2', $allRows[1]['name']);
		$this->assertEquals('user3', $allRows[2]['name']);

		// each with key
		$query = new Query();
		$query->from('tbl_customer')->orderBy('id')->indexBy('name');
		$allRows = [];
		foreach ($query->each($db) as $key => $row) {
			$allRows[$key] = $row;
		}
		$this->assertEquals(3, count($allRows));
		$this->assertEquals('address1', $allRows['user1']['address']);
		$this->assertEquals('address2', $allRows['user2']['address']);
		$this->assertEquals('address3', $allRows['user3']['address']);
	}

	public function testActiveQuery()
	{
		$db = $this->getConnection();

		$query = Customer::find()->orderBy('id');
		$customers = [];
		foreach ($query->batch(2, $db) as $models) {
			$customers = array_merge($customers, $models);
		}
		$this->assertEquals(3, count($customers));
		$this->assertEquals('user1', $customers[0]->name);
		$this->assertEquals('user2', $customers[1]->name);
		$this->assertEquals('user3', $customers[2]->name);

		// query in batch 1
		$query = Customer::find()->orderBy('id');
		$customers = [];
		foreach ($query->batch(1, $db) as $model) {
			$customers[] = $model;
		}
		$this->assertEquals(3, count($customers));
		$this->assertEquals('user1', $customers[0]->name);
		$this->assertEquals('user2', $customers[1]->name);
		$this->assertEquals('user3', $customers[2]->name);

		// batch with eager loading
		$query = Customer::find()->with('orders')->orderBy('id');
		$customers = [];
		foreach ($query->batch(2, $db) as $models) {
			$customers = array_merge($customers, $models);
			foreach ($models as $model) {
				$this->assertTrue($model->isRelationPopulated('orders'));
			}
		}
		$this->assertEquals(3, count($customers));
		$this->assertEquals(1, count($customers[0]->orders));
		$this->assertEquals(2, count($customers[1]->orders));
		$this->assertEquals(0, count($customers[2]->orders));
	}
}
