phpstan-doctrine/tests/DoctrineIntegration/ORM/data/queryBuilder.php at ce8ea0577a89ed9d55b5194ba8a505a2888a568f · phpstan/phpstan-doctrine · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
<?php declare(strict_types = 1);

namespace PHPStan\DoctrineIntegration\ORM\QueryBuilder;

use Doctrine\ORM\EntityManager;
use Doctrine\ORM\Query;
use PHPStan\DoctrineIntegration\ORM\CustomRepositoryUsage\MyEntity;
use function PHPStan\Testing\assertType;

class Foo
{

	/**
	 * @var EntityManager
	 */
	private $entityManager;

	public function __construct(EntityManager $entityManager)
	{
		$this->entityManager = $entityManager;
	}

	public function doFoo(): Query
	{
		$queryBuilder = $this->entityManager->createQueryBuilder()
			->select('e')
			->from(MyEntity::class, 'e');
		$query = $queryBuilder->getQuery();
		assertType('\'SELECT e FROM PHPStan\\\\DoctrineIntegration\\\\ORM\\\\CustomRepositoryUsage\\\\MyEntity e\'', $query->getDQL());
		assertType('\'SELECT e FROM PHPStan\\\\DoctrineIntegration\\\\ORM\\\\CustomRepositoryUsage\\\\MyEntity e\'', $queryBuilder->getDQL());

		return $query;
	}

	public function doBar(): Query
	{
		$entityRepository = $this->entityManager->getRepository(MyEntity::class);
		$queryBuilder = $entityRepository->createQueryBuilder('e');
		$query = $queryBuilder->getQuery();

		assertType('\'SELECT e FROM PHPStan\\\\DoctrineIntegration\\\\ORM\\\\CustomRepositoryUsage\\\\MyEntity e\'', $query->getDQL());
		assertType('\'SELECT e FROM PHPStan\\\\DoctrineIntegration\\\\ORM\\\\CustomRepositoryUsage\\\\MyEntity e\'', $queryBuilder->getDQL());

		return $query;
	}

	/**
	 * @phpstan-param class-string $entityClass
	 */
	public function dynamicQueryBuilder(string $entityClass): Query
	{
		$entityRepository = $this->entityManager->getRepository($entityClass);
		$queryBuilder = $entityRepository->createQueryBuilder('e');
		return $queryBuilder->getQuery();
	}

	public function usingMethodThatReturnStatic(): ?MyEntity
	{
		$queryBuilder = $this->entityManager->createQueryBuilder();

		$queryBuilder
			->select('e')
			->from(MyEntity::class, 'e')
			->where('e.id = :id')
			->setParameters([
				'id' => 123,
			]);

		$result = $queryBuilder->getQuery()->getOneOrNullResult();
		assertType('PHPStan\DoctrineIntegration\ORM\CustomRepositoryUsage\MyEntity|null', $result);

		return $result;
	}

	public function getCustomQueryBuilder(): CustomQueryBuilder
	{
		assertType(CustomQueryBuilder::class, $this->entityManager->createQueryBuilder());
	    return $this->entityManager->createQueryBuilder();
	}
}

class CustomQueryBuilder extends \Doctrine\ORM\QueryBuilder
{
}