SonataAdminBundle/Tests/Filter/ORM/CallbackFilterTest.php

<?php

/*
 * This file is part of the Sonata 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\Tests\Filter\ORM;

use Sonata\AdminBundle\Filter\ORM\CallbackFilter;

class CallbackFilterTest extends \PHPUnit_Framework_TestCase
{
    public function getFieldDescription(array $options)
    {
        $fieldDescription = $this->getMock('Sonata\AdminBundle\Admin\FieldDescriptionInterface');
        $fieldDescription->expects($this->once())
            ->method('getOptions')
            ->will($this->returnValue($options));

        $fieldDescription->expects($this->once())
            ->method('getName')
            ->will($this->returnValue('field_name'));

        return $fieldDescription;
    }

    public function testFilterClosure()
    {
        $builder = new QueryBuilder;

        $filter = new CallbackFilter;
        $filter->setFieldDescription($this->getFieldDescription(array(
            'callback' => function($builder, $alias, $field, $value) {
                $builder->andWhere(sprintf('CUSTOM QUERY %s.%s', $alias, $field));
                $builder->setParameter('value', $value);
            }
        )));
        ;

        $filter->filter($builder, 'alias', 'field', 'myValue');

        $this->assertEquals(array('CUSTOM QUERY alias.field'), $builder->query);
        $this->assertEquals(array('value' => 'myValue'), $builder->parameters);
    }

    public function testFilterMethod()
    {
        $builder = new QueryBuilder;

        $filter = new CallbackFilter;
        $filter->setFieldDescription($this->getFieldDescription(array(
            'callback' => array($this, 'customCallback')
        )));

        $filter->filter($builder, 'alias', 'field', 'myValue');

        $this->assertEquals(array('CUSTOM QUERY alias.field'), $builder->query);
        $this->assertEquals(array('value' => 'myValue'), $builder->parameters);
    }

    public function customCallback($builder, $alias, $field, $value) {
        $builder->andWhere(sprintf('CUSTOM QUERY %s.%s', $alias, $field));
        $builder->setParameter('value', $value);
    }

    /**
     * @expectedException RuntimeException
     */
    public function testFilterException()
    {
        $builder = new QueryBuilder;

        $filter = new CallbackFilter;
        $filter->setFieldDescription($this->getFieldDescription(array()));

        $filter->filter($builder, 'alias', 'field', 'myValue');
    }
}