Add test

Author: VincentLanglet

tests/Type/Doctrine/ObjectMetadataResolverTest.php

@@ -0,0 +1,61 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Reflection\Doctrine;
+
+use Doctrine\ODM\MongoDB\Mapping\Annotations\Document;
+use Doctrine\ODM\MongoDB\Mapping\Annotations\Id;
+use Doctrine\ORM\Mapping as ORM;
+use PHPStan\Type\Doctrine\ObjectMetadataResolver;
+
+final class ObjectMetadataResolverTest extends \PHPStan\Testing\PHPStanTestCase
+{
+
+	public function testGetRepositoryClassWithCustomObjectManager(): void
+	{
+		$objectMetadataResolver = new ObjectMetadataResolver($this->createReflectionProvider(), __DIR__ . '/custom-object-manager.php', null);
+
+		self::assertSame('Doctrine\ODM\MongoDB\Repository\DocumentRepository', $objectMetadataResolver->getRepositoryClass(MyDocument::class));
+		self::assertSame('Doctrine\ORM\EntityRepository', $objectMetadataResolver->getRepositoryClass(MyEntity::class));
+	}
+
+}
+
+/**
+ * @ORM\Entity
+ */
+class MyEntity
+{
+
+	/**
+	 * @ORM\Id()
+	 * @ORM\GeneratedValue()
+	 * @ORM\Column(type="integer")
+	 *
+	 * @var int
+	 */
+	private $id;
+
+	public function doSomethingElse(): void
+	{
+	}
+
+}
+
+/**
+ * @Document
+ */
+class MyDocument
+{
+
+	/**
+	 * @Id(strategy="NONE", type="string")
+	 *
+	 * @var string
+	 */
+	private $id;
+
+	public function doSomethingElse(): void
+	{
+	}
+
+}

tests/Type/Doctrine/custom-object-manager.php

@@ -0,0 +1,185 @@
+<?php declare(strict_types = 1);
+
+use Doctrine\Common\Annotations\AnnotationReader;
+use Doctrine\Common\Cache\ArrayCache;
+use Doctrine\ODM\MongoDB;
+use Doctrine\ORM;
+use Doctrine\Persistence\Mapping\ClassMetadataFactory;
+
+$config = new MongoDB\Configuration();
+$config->setProxyDir(__DIR__);
+$config->setProxyNamespace('PHPstan\Doctrine\OdmProxies');
+$config->setMetadataCacheImpl(new ArrayCache());
+$config->setHydratorDir(__DIR__);
+$config->setHydratorNamespace('PHPstan\Doctrine\OdmHydrators');
+
+$config->setMetadataDriverImpl(
+	new MongoDB\Mapping\Driver\AnnotationDriver(
+		new AnnotationReader(),
+		[__DIR__ . '/ObjectMetadataResolverTest.php']
+	)
+);
+
+$documentManager = MongoDB\DocumentManager::create(
+	null,
+	$config
+);
+
+$config = new ORM\Configuration();
+$config->setProxyDir(__DIR__);
+$config->setProxyNamespace('PHPstan\Doctrine\OrmProxies');
+$config->setMetadataCacheImpl(new ArrayCache());
+
+$config->setMetadataDriverImpl(
+	new ORM\Mapping\Driver\AnnotationDriver(
+		new AnnotationReader(),
+		[__DIR__ . '/ObjectMetadataResolverTest.php']
+	)
+);
+
+$entityManager = ORM\EntityManager::create(
+	[
+		'driver' => 'pdo_sqlite',
+		'memory' => true,
+	],
+	$config
+);
+
+$metadataFactory = new class($documentManager, $entityManager) implements ClassMetadataFactory
+{
+
+	/** @var MongoDB\DocumentManager */
+	private $documentManager;
+
+	/** @var ORM\EntityManager */
+	private $entityManager;
+
+	public function __construct(MongoDB\DocumentManager $documentManager, ORM\EntityManager $entityManager)
+	{
+		$this->documentManager = $documentManager;
+		$this->entityManager = $entityManager;
+	}
+
+	public function getAllMetadata()
+	{
+		return array_merge(
+			$this->documentManager->getMetadataFactory()->getAllMetadata(),
+			$this->entityManager->getMetadataFactory()->getAllMetadata()
+		);
+	}
+
+	public function getMetadataFor($className): void
+	{
+		throw new \Exception(__FILE__);
+	}
+
+	public function isTransient($className): void
+	{
+		throw new \Exception(__FILE__);
+	}
+
+	public function hasMetadataFor($className): void
+	{
+		throw new \Exception(__FILE__);
+	}
+
+	public function setMetadataFor($className, $class): void
+	{
+		throw new \Exception(__FILE__);
+	}
+
+};
+
+return new class($documentManager, $entityManager, $metadataFactory) implements \Doctrine\Persistence\ObjectManager
+{
+
+	/** @var MongoDB\DocumentManager */
+	private $documentManager;
+
+	/** @var ORM\EntityManager */
+	private $entityManager;
+
+	/** @var ClassMetadataFactory */
+	private $classMetadataFactory;
+
+	public function __construct(MongoDB\DocumentManager $documentManager, ORM\EntityManager $entityManager, ClassMetadataFactory $classMetadataFactory)
+	{
+		$this->documentManager = $documentManager;
+		$this->entityManager = $entityManager;
+		$this->classMetadataFactory = $classMetadataFactory;
+	}
+
+	public function getRepository($className)
+	{
+		if (strpos($className, 'Entity') !== false) {
+			return $this->entityManager->getRepository($className);
+		}
+
+		return $this->documentManager->getRepository($className);
+	}
+
+	public function getClassMetadata($className)
+	{
+		if (strpos($className, 'Entity') !== false) {
+			return $this->entityManager->getClassMetadata($className);
+		}
+
+		return $this->documentManager->getClassMetadata($className);
+	}
+
+	public function getMetadataFactory()
+	{
+		return $this->classMetadataFactory;
+	}
+
+	public function find($className, $id): void
+	{
+		throw new \Exception(__FILE__);
+	}
+
+	public function persist($object): void
+	{
+		throw new \Exception(__FILE__);
+	}
+
+	public function remove($object): void
+	{
+		throw new \Exception(__FILE__);
+	}
+
+	public function merge($object): void
+	{
+		throw new \Exception(__FILE__);
+	}
+
+	public function clear($objectName = null): void
+	{
+		throw new \Exception(__FILE__);
+	}
+
+	public function detach($object): void
+	{
+		throw new \Exception(__FILE__);
+	}
+
+	public function refresh($object): void
+	{
+		throw new \Exception(__FILE__);
+	}
+
+	public function flush(): void
+	{
+		throw new \Exception(__FILE__);
+	}
+
+	public function initializeObject($obj): void
+	{
+		throw new \Exception(__FILE__);
+	}
+
+	public function contains($object): void
+	{
+		throw new \Exception(__FILE__);
+	}
+
+};