doctrine / DoctrineBundle

class ContainerRepositoryFactoryTest extends TestCase

{

    public function testGetRepositoryReturnsService()

    {

        if (!interface_exists(ContainerInterface::class)) {

            $this->markTestSkipped('Symfony 3.3 is needed for this feature.');

        }

        $em = $this->createEntityManager([

            'Foo\CoolEntity' => 'my_repo',

        ]);

        $repo = new StubRepository($em, new ClassMetadata(''));

        $container = $this->createContainer([

            'my_repo' => $repo,

        ]);

        $factory = new ContainerRepositoryFactory($container);

        $this->assertSame($repo, $factory->getRepository($em, 'Foo\CoolEntity'));

    }

    public function testGetRepositoryReturnsEntityRepository()

    {

     * @expectedException \RuntimeException

     * @expectedExceptionMessage The service "my_repo" must extend EntityRepository (or a base class, like ServiceEntityRepository).

     */

    public function testServiceRepositoriesMustExtendEntityRepository()

    {

        if (!interface_exists(ContainerInterface::class)) {

            $this->markTestSkipped('Symfony 3.3 is needed for this feature.');

        }

        $repo = new \stdClass();

        $container = $this->createContainer([

            'my_repo' => $repo,

        ]);

        $em = $this->createEntityManager([

            'Foo\CoolEntity' => 'my_repo',

        ]);

        $factory = new ContainerRepositoryFactory($container);

        $factory->getRepository($em, 'Foo\CoolEntity');

    }

    /**

     * @expectedException \RuntimeException