Sonata\DoctrineORMAdminBundle\Tests\Datagrid\PagerTest

Complexity

Total Complexity

3

Size/Duplication

Total Lines	70
Duplicated Lines	0 %

Coupling/Cohesion

Components	0
Dependencies	7

Importance

Changes

0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	testComputeNbResultFoCompositeId()	0	20	1
A	dataGetComputeNbResult()	0	7	1
A	testComputeNbResult()	0	33	1

<?php

declare(strict_types=1);

/*
 * 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\DoctrineORMAdminBundle\Tests\Datagrid;

use Doctrine\ORM\AbstractQuery;
use Doctrine\ORM\Tools\SchemaTool;
use PHPUnit\Framework\TestCase;
use Sonata\DoctrineORMAdminBundle\Datagrid\Pager;
use Sonata\DoctrineORMAdminBundle\Datagrid\ProxyQuery;
use Sonata\DoctrineORMAdminBundle\Tests\Filter\QueryBuilder;
use Sonata\DoctrineORMAdminBundle\Tests\Fixtures\Entity\ORM\Product;
use Sonata\DoctrineORMAdminBundle\Tests\Fixtures\Entity\ORM\Store;
use Sonata\DoctrineORMAdminBundle\Tests\Fixtures\Entity\ORM\StoreProduct;
use Symfony\Bridge\Doctrine\Test\DoctrineTestHelper;

class PagerTest extends TestCase
{
    public function testComputeNbResultFoCompositeId()
    {
        $em = DoctrineTestHelper::createTestEntityManager();
        $classes = [
            $em->getClassMetadata(Product::class),
            $em->getClassMetadata(StoreProduct::class),
            $em->getClassMetadata(Store::class),
        ];
        $schemaTool = new SchemaTool($em);
        $schemaTool->createSchema($classes);

        $qb = $em->createQueryBuilder()
            ->select('o')
            ->from(StoreProduct::class, 'o');
        $pq = new ProxyQuery($qb);
        $pager = new Pager();
        $pager->setCountColumn($em->getClassMetadata(StoreProduct::class)->getIdentifierFieldNames());
        $pager->setQuery($pq);
        $this->assertEquals(0, $pager->computeNbResult());
    }

    public function dataGetComputeNbResult()
    {
        return [
            [true],
            [false],
        ];
    }

    /**
     * @dataProvider dataGetComputeNbResult
     *
     * @param bool $distinct
     */
    public function testComputeNbResult($distinct)
    {
        $query = $this->getMockBuilder(AbstractQuery::class)
            ->disableOriginalConstructor()
            ->setMethods(['getSingleScalarResult'])
            ->getMockForAbstractClass();

        $query->expects($this->once())
            ->method('getSingleScalarResult');

        $queryBuilder = $this->getMockBuilder(QueryBuilder::class)
            ->disableOriginalConstructor()
            ->setMethods(['getQuery', 'select', 'resetDQLPart'])
            ->getMock();

        $queryBuilder->expects($this->once())
            ->method('getQuery')
            ->willReturn($query);

        $queryBuilder->expects($this->once())
            ->method('select');

        $proxyQuery = new ProxyQuery($queryBuilder);
        $proxyQuery->setDistinct($distinct);

        $queryBuilder->expects($this->once())
            ->method('resetDQLPart')
            ->willReturn($proxyQuery);

        $pager = new Pager();
        $pager->setQuery($proxyQuery);
        $pager->computeNbResult();
    }
}