SonataAdminBundle/Tests/Model/AuditManagerTest.php

<?php

/*
 * This file is part of the Sonata Project package.
 *
 * (c) Thomas Rabaix <thomas.rabaix@sonata-project.org>
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace Sonata\AdminBundle\Model;

/**
 * Test for AuditManager.
 *
 * @author Andrej Hudec <pulzarraider@gmail.com>
 */
class AuditManagerTest extends \PHPUnit_Framework_TestCase
{
    public function testGetReader()
    {
        $container = $this->getMock('Symfony\Component\DependencyInjection\ContainerInterface');

        $fooReader = $this->getMock('Sonata\AdminBundle\Model\AuditReaderInterface');
        $barReader = $this->getMock('Sonata\AdminBundle\Model\AuditReaderInterface');

        $container->expects($this->any())
            ->method('get')
            ->will($this->returnCallback(function ($id) use ($fooReader, $barReader) {
                switch ($id) {
                    case 'foo_reader':
                        return $fooReader;
                        break;

                    case 'bar_reader':
                        return $barReader;
                        break;
                }

                return;
            }));

        $auditManager = new AuditManager($container);

        $this->assertFalse($auditManager->hasReader('Foo\Foo1'));

        $auditManager->setReader('foo_reader', array('Foo\Foo1', 'Foo\Foo2'));

        $this->assertTrue($auditManager->hasReader('Foo\Foo1'));
        $this->assertSame($fooReader, $auditManager->getReader('Foo\Foo1'));
    }

    public function testGetReaderWithException()
    {
        $this->setExpectedException('\RuntimeException', 'The class "Foo\Foo" does not have any reader manager');

        $container = $this->getMock('Symfony\Component\DependencyInjection\ContainerInterface');
        $auditManager = new AuditManager($container);

        $auditManager->getReader('Foo\Foo');
    }
}