ActiveRelationTest.php 2.85 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 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 83 84 85
<?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);
    }
}