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.
83 lines
2.1 KiB
83 lines
2.1 KiB
11 years ago
|
<?php
|
||
|
|
|
||
|
|
namespace yiiunit\extensions\mongo;
|
||
|
|
|
||
|
|
use yiiunit\data\ar\mongo\ActiveRecord;
|
||
|
|
use yiiunit\data\ar\mongo\Customer;
|
||
|
|
use yiiunit\data\ar\mongo\CustomerOrder;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @group mongo
|
||
|
|
*/
|
||
|
|
class ActiveRelationTest extends MongoTestCase
|
||
|
|
{
|
||
|
|
protected function setUp()
|
||
|
|
{
|
||
|
|
parent::setUp();
|
||
|
|
ActiveRecord::$db = $this->getConnection();
|
||
|
|
$this->setUpTestRows();
|
||
|
|
}
|
||
|
|
|
||
|
|
protected function tearDown()
|
||
|
|
{
|
||
|
|
$this->dropCollection(Customer::collectionName());
|
||
|
|
$this->dropCollection(CustomerOrder::collectionName());
|
||
|
|
parent::tearDown();
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Sets up test rows.
|
||
|
|
*/
|
||
|
|
protected function setUpTestRows()
|
||
|
|
{
|
||
|
|
$customerCollection = $this->getConnection()->getCollection('customer');
|
||
|
|
|
||
|
|
$customers = [];
|
||
|
|
for ($i = 1; $i <= 5; $i++) {
|
||
|
|
$customers[] = [
|
||
|
|
'name' => 'name' . $i,
|
||
|
|
'email' => 'email' . $i,
|
||
|
|
'address' => 'address' . $i,
|
||
|
|
'status' => $i,
|
||
|
|
];
|
||
|
|
}
|
||
|
|
$customerCollection->batchInsert($customers);
|
||
|
|
|
||
|
|
$customerOrderCollection = $this->getConnection()->getCollection('customer_order');
|
||
|
|
$customerOrders = [];
|
||
|
|
foreach ($customers as $customer) {
|
||
|
|
$customerOrders[] = [
|
||
|
|
'customer_id' => $customer['_id'],
|
||
|
|
'number' => $customer['status'],
|
||
|
|
];
|
||
|
|
$customerOrders[] = [
|
||
|
|
'customer_id' => $customer['_id'],
|
||
|
|
'number' => $customer['status'] + 1,
|
||
|
|
];
|
||
|
|
}
|
||
|
|
$customerOrderCollection->batchInsert($customerOrders);
|
||
|
|
}
|
||
|
|
|
||
|
|
// Tests :
|
||
|
|
|
||
|
|
public function testFindLazy()
|
||
|
|
{
|
||
|
|
/** @var CustomerOrder $order */
|
||
|
|
$order = CustomerOrder::find(['number' => 2]);
|
||
|
|
$this->assertFalse($order->isRelationPopulated('customer'));
|
||
|
|
$index = $order->customer;
|
||
|
|
$this->assertTrue($order->isRelationPopulated('customer'));
|
||
|
|
$this->assertTrue($index instanceof Customer);
|
||
|
|
$this->assertEquals(1, count($order->populatedRelations));
|
||
|
|
}
|
||
|
|
|
||
|
|
public function testFindEager()
|
||
|
|
{
|
||
|
|
$orders = CustomerOrder::find()->with('customer')->all();
|
||
|
|
$this->assertEquals(10, count($orders));
|
||
|
|
$this->assertTrue($orders[0]->isRelationPopulated('customer'));
|
||
|
|
$this->assertTrue($orders[1]->isRelationPopulated('customer'));
|
||
|
|
$this->assertTrue($orders[0]->index instanceof ArticleIndex);
|
||
|
|
$this->assertTrue($orders[1]->index instanceof ArticleIndex);
|
||
|
|
}
|
||
|
|
}
|