SonataAdminBundle/Tests/Form/DataTransformer/ModelsToArrayTransformerTest.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\Form\DataTransformer;

use Doctrine\Common\Collections\ArrayCollection;
use Sonata\AdminBundle\Form\ChoiceList\ModelChoiceList;
use Sonata\AdminBundle\Tests\Fixtures\Entity\Form\FooEntity;
use Symfony\Component\Form\ChoiceList\LegacyChoiceListAdapter;

/**
 * @author Andrej Hudec <pulzarraider@gmail.com>
 */
class ModelsToArrayTransformerTest extends \PHPUnit_Framework_TestCase
{
    private $choiceList;
    private $modelChoiceList;

    private $modelManager;

    public function setUp()
    {
        $this->modelChoiceList = $this->getMockBuilder('Sonata\AdminBundle\Form\ChoiceList\ModelChoiceList')
            ->disableOriginalConstructor()
            ->getMock();

        // Symfony < 2.7 BC
        if (class_exists('Symfony\Component\Form\ChoiceList\LegacyChoiceListAdapter')) {
            $this->choiceList = new LegacyChoiceListAdapter($this->modelChoiceList);
        } else {
            $this->choiceList = $this->modelChoiceList;
        }

        $this->modelManager = $this->getMock('Sonata\AdminBundle\Model\ModelManagerInterface');

        // php 5.3 BC
        $modelManager = $this->modelManager;

        $this->modelChoiceList->expects($this->any())
            ->method('getModelManager')
            ->will($this->returnCallback(function () use ($modelManager) {
                return $modelManager;
            }));
    }

    /**
     * @dataProvider getTransformTests
     */
    public function testTransform($expected, $collection, $identifiers)
    {
        $transformer = new ModelsToArrayTransformer($this->choiceList);

        $this->modelChoiceList->expects($this->any())
            ->method('getIdentifierValues')
            ->will($this->returnCallback(function ($entity) use ($identifiers) {
                if ($entity instanceof FooEntity) {
                    return $identifiers;
                }

                return array();
            }));

        $this->modelChoiceList->expects($this->any())
            ->method('getIdentifier')
            ->will($this->returnCallback(function () use ($identifiers) {
                return $identifiers;
            }));

        $this->modelChoiceList->expects($this->any())
            ->method('getEntities')
            ->will($this->returnCallback(function () {
                return array('bcd' => new FooEntity(array('bcd')), 'efg' => new FooEntity(array('efg')), 'abc' => new FooEntity(array('abc')));
            }));

        $this->assertEquals($expected, $transformer->transform($collection));
    }

    public function getTransformTests()
    {
        return array(
            array(array(), null, array()),
            array(array(), array(), array()),
            array(array('id'), array(new FooEntity()), array('id')),
            array(array('id', 'id'), array(new FooEntity(), new FooEntity()), array('id')),
            array(array('abc', 'bcd', 'efg'), array(new FooEntity(array('abc')), new FooEntity(array('bcd')), new FooEntity(array('efg'))), array('id1', 'id2')),
        );
    }

    public function testReverseTransformWithException1()
    {
        $this->setExpectedException('Symfony\Component\Form\Exception\UnexpectedTypeException', 'Expected argument of type "\ArrayAccess", "NULL" given');

        $transformer = new ModelsToArrayTransformer($this->choiceList);

        $this->modelManager->expects($this->any())
            ->method('getModelCollectionInstance')
            ->will($this->returnValue(null));

        $transformer->reverseTransform(array());
    }

    public function testReverseTransformWithException2()
    {
        $this->setExpectedException('Symfony\Component\Form\Exception\UnexpectedTypeException', 'Expected argument of type "array", "integer" given');

        $transformer = new ModelsToArrayTransformer($this->choiceList);

        $this->modelManager->expects($this->any())
            ->method('getModelCollectionInstance')
            ->will($this->returnValue(new ArrayCollection()));

        $transformer->reverseTransform(123);
    }

    /**
     * @dataProvider getReverseTransformEmptyTests
     */
    public function testReverseTransformEmpty($keys)
    {
        $transformer = new ModelsToArrayTransformer($this->choiceList);

        $this->modelManager->expects($this->any())
            ->method('getModelCollectionInstance')
            ->will($this->returnValue(new ArrayCollection()));

        $this->assertInstanceOf('Doctrine\Common\Collections\ArrayCollection', $transformer->reverseTransform($keys));
    }

    public function getReverseTransformEmptyTests()
    {
        return array(
            array(null),
            array(''),
        );
    }

    public function testReverseTransform()
    {
        $transformer = new ModelsToArrayTransformer($this->choiceList);

        $this->modelManager->expects($this->any())
            ->method('getModelCollectionInstance')
            ->will($this->returnValue(new ArrayCollection()));

        $entity1 =  new FooEntity(array('foo'));
        $entity2 =  new FooEntity(array('bar'));
        $entity3 =  new FooEntity(array('baz'));

        $this->modelChoiceList->expects($this->any())
            ->method('getEntity')
            ->will($this->returnCallback(function ($key) use ($entity1, $entity2, $entity3) {
                switch ($key) {
                    case 'foo':
                        return $entity1;

                        break;
                    case 'bar':
                        return $entity2;

                        break;
                    case 'baz':
                        return $entity3;
                        break;
                }

                return;
            }));

        $collection = $transformer->reverseTransform(array('foo', 'bar'));
        $this->assertInstanceOf('Doctrine\Common\Collections\ArrayCollection', $collection);
        $this->assertEquals(array($entity1, $entity2), $collection->getValues());
        $this->assertCount(2, $collection);
    }

    public function testReverseTransformWithNonexistentEntityKey()
    {
        $this->setExpectedException('Symfony\Component\Form\Exception\TransformationFailedException', 'The entities with keys "nonexistent" could not be found');

        $transformer = new ModelsToArrayTransformer($this->choiceList);

        $this->modelManager->expects($this->any())
            ->method('getModelCollectionInstance')
            ->will($this->returnValue(new ArrayCollection()));

        $this->modelChoiceList->expects($this->any())
            ->method('getEntity')
            ->will($this->returnValue(false));

        $transformer->reverseTransform(array('nonexistent'));
    }
}