PrintABrick/tests/AppBundle/Repository/LDraw/ModelRepositoryTest.php

<?php

namespace Tests\AppBundle\Repository\LDraw;

use AppBundle\Entity\LDraw\Model;
use AppBundle\Repository\LDraw\ModelRepository;
use Tests\AppBundle\BaseTest;
use Tests\AppBundle\Fixtures\LoadBaseData;

class ModelRepositoryTest extends BaseTest
{
    /** @var ModelRepository */
    private $repository;

    public function setUp()
    {
        parent::setUp();
        $this->setUpDb([LoadBaseData::class]);

        $this->repository = $this->em->getRepository(Model::class);
    }

    public function testFindOneByNumber()
    {
        $model = $this->repository->findOneByNumber('25');
        $this->assertEquals(1, $model->getId());

        $model = $this->repository->findOneByNumber(1);
        $this->assertEquals(1, $model->getId());
    }

    public function testFindOneByName()
    {
        $model = $this->repository->findOneByName('Name');
        $this->assertEquals(1, $model->getId());

        $model = $this->repository->findOneByName('Not');
        $this->assertNull($model);
    }

    public function testFindAllSiblings()
    {
        $model = $this->repository->findOneByNumber(2);
        $siblings = $this->repository->findAllSiblings($model);

        $this->assertCount(1, $siblings);
    }

    public function testCount()
    {
        $this->assertEquals(3, $this->repository->count());
    }

    public function testGetOrCreate()
    {
        $this->assertCount(3, $this->repository->findAll());

        $model = $this->repository->getOrCreate('25');
        $this->repository->save($model);
        $this->assertCount(3, $this->repository->findAll());

        $model = $this->repository->getOrCreate(33);
        $this->repository->save($model);
        $this->assertCount(4, $this->repository->findAll());
    }
}