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.
85 lines
2.9 KiB
85 lines
2.9 KiB
10 years ago
|
<?php
|
||
|
|
|
||
|
|
namespace yiiunit\extensions\mongodb\file;
|
||
|
|
|
||
|
|
use yiiunit\data\ar\mongodb\Customer;
|
||
|
|
use yiiunit\data\ar\mongodb\file\CustomerFile;
|
||
|
|
use yiiunit\extensions\mongodb\MongoDbTestCase;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @group mongodb
|
||
|
|
*/
|
||
|
|
class ActiveRelationTest extends MongoDbTestCase
|
||
|
|
{
|
||
|
|
protected function setUp()
|
||
|
|
{
|
||
|
|
parent::setUp();
|
||
|
|
\yiiunit\data\ar\mongodb\ActiveRecord::$db = $this->getConnection();
|
||
|
|
\yiiunit\data\ar\mongodb\file\ActiveRecord::$db = $this->getConnection();
|
||
|
|
$this->setUpTestRows();
|
||
|
|
}
|
||
|
|
|
||
|
|
protected function tearDown()
|
||
|
|
{
|
||
|
|
$this->dropCollection(Customer::collectionName());
|
||
|
|
$this->dropCollection(CustomerFile::collectionName());
|
||
|
|
parent::tearDown();
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Sets up test rows.
|
||
|
|
*/
|
||
|
|
protected function setUpTestRows()
|
||
|
|
{
|
||
|
|
$fileCollection = $this->getConnection()->getFileCollection(CustomerFile::collectionName());
|
||
|
|
$customers = [];
|
||
|
|
$files = [];
|
||
|
|
for ($i = 1; $i <= 5; $i++) {
|
||
|
|
$file = [
|
||
|
|
'tag' => 'tag' . $i,
|
||
|
|
'status' => $i,
|
||
|
|
];
|
||
|
|
$content = 'content' . $i;
|
||
|
|
$file['_id'] = $fileCollection->insertFileContent($content, $file);
|
||
|
|
$file['content'] = $content;
|
||
|
|
$files[] = $file;
|
||
|
|
|
||
|
|
$customers[] = [
|
||
|
|
'name' => 'name' . $i,
|
||
|
|
'email' => 'email' . $i,
|
||
|
|
'address' => 'address' . $i,
|
||
|
|
'status' => $i,
|
||
|
|
'file_id' => $file['_id'],
|
||
|
|
];
|
||
|
|
}
|
||
|
|
$customerCollection = $this->getConnection()->getCollection(Customer::collectionName());
|
||
|
|
$customers = $customerCollection->batchInsert($customers);
|
||
|
|
}
|
||
|
|
|
||
|
|
// Tests :
|
||
|
|
|
||
|
|
public function testFindLazy()
|
||
|
|
{
|
||
|
|
/* @var $customer Customer */
|
||
|
|
$customer = Customer::findOne(['status' => 2]);
|
||
|
|
$this->assertFalse($customer->isRelationPopulated('file'));
|
||
|
|
$file = $customer->file;
|
||
|
|
$this->assertTrue($customer->isRelationPopulated('file'));
|
||
|
|
$this->assertTrue($file instanceof CustomerFile);
|
||
|
|
$this->assertEquals((string) $file->_id, (string) $customer->file_id);
|
||
|
|
$this->assertEquals(1, count($customer->relatedRecords));
|
||
|
|
}
|
||
|
|
|
||
|
|
public function testFindEager()
|
||
|
|
{
|
||
|
|
/* @var $customers Customer[] */
|
||
|
|
$customers = Customer::find()->with('file')->all();
|
||
|
|
$this->assertEquals(5, count($customers));
|
||
|
|
$this->assertTrue($customers[0]->isRelationPopulated('file'));
|
||
|
|
$this->assertTrue($customers[1]->isRelationPopulated('file'));
|
||
|
|
$this->assertTrue($customers[0]->file instanceof CustomerFile);
|
||
|
|
$this->assertEquals((string) $customers[0]->file->_id, (string) $customers[0]->file_id);
|
||
|
|
$this->assertTrue($customers[1]->file instanceof CustomerFile);
|
||
|
|
$this->assertEquals((string) $customers[1]->file->_id, (string) $customers[1]->file_id);
|
||
|
|
}
|
||
|
|
}
|