Doctrine\Tests\ORM\Query\SelectSqlMappingTest

Complexity

Total Complexity

4

Size/Duplication

Total Lines	58
Duplicated Lines	0 %

Importance

Changes

0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	setUp()	0	3	1
A	testTypeFromMathematicNodeFunction()	0	16	1
A	assertScalarTypes()	0	22	2

<?php

declare(strict_types=1);

namespace Doctrine\Tests\ORM\Query;

use Doctrine\DBAL\Types\FloatType;
use Doctrine\DBAL\Types\IntegerType;
use Doctrine\ORM\Query as ORMQuery;
use Doctrine\Tests\OrmTestCase;
use Exception;

Type Exception is not used in this file.

use ReflectionObject;
use function array_search;
use function sprintf;

class SelectSqlMappingTest extends OrmTestCase
{
    private $em;

    protected function setUp() : void
    {
        $this->em = $this->getTestEntityManager();
    }

    /**
     * Assert a valid SQL generation.
     *
     * @param string $dqlToBeTested
     * @param array  $types
     */
    public function assertScalarTypes($dqlToBeTested, array $types)
    {
        $query = $this->em->createQuery($dqlToBeTested);

        $query
            ->setHint(ORMQuery::HINT_FORCE_PARTIAL_LOAD, true)
            ->useQueryCache(false);

        $r      = new ReflectionObject($query);
        $method = $r->getMethod('getResultSetMapping');
        $method->setAccessible(true);
        /** @var ORMQuery\ResultSetMapping $mapping */
        $mapping = $method->invoke($query);
        foreach ($types as $key => $expectedType) {
            $alias = array_search($key, $mapping->scalarMappings);
            self::assertInstanceOf(
                $expectedType,
                $mapping->typeMappings[$alias],
                sprintf('The field "%s" was expected as a %s', $key, $expectedType)
            );
        }
        $query->free();
    }

    /**
     * @group DDC-2235
     */
    public function testTypeFromMathematicNodeFunction() : void
    {
        $this->assertScalarTypes(
            'SELECT p, 
          count(p.id) as count, 
          SUM(p.price) as sales, 
          AVG(p.price) as average, 
          ABS(p.price) as absolute,
          LENGTH(p.name) as length
          FROM Doctrine\Tests\Models\CMS\CmsProduct p',
            [
                'count' => IntegerType::class,
                'sales' => FloatType::class,
                'average' => FloatType::class,
                'absolute' => FloatType::class,
                'length' => IntegerType::class,
            ]
        );
    }
}