Fix return type of entity manager dynamic return

The methods `EntityManagerInterface#getReference()` and
`EntityManagerInterface#getPartialReference()` have nullable return.

That's properly documented for `getReference()` but had to be fixed for
`getPartialReference()`.

Author: lcobucci

src/Type/Doctrine/EntityManagerFindDynamicReturnTypeExtension.php

@@ -43,12 +43,7 @@ public function getTypeFromMethodCall(
 			return $mixedType;
 		}
 
-		$type = new ObjectType($argType->getValue());
-		if ($methodReflection->getName() === 'find') {
-			$type = TypeCombinator::addNull($type);
-		}
-
-		return $type;
+		return TypeCombinator::addNull(new ObjectType($argType->getValue()));
 	}
 
 }

tests/DoctrineIntegration/ORM/data/entityManagerDynamicReturn.php

@@ -32,12 +32,22 @@ public function findDynamicType(): void
 	public function getReferenceDynamicType(): void
 	{
 		$test = $this->entityManager->getReference(MyEntity::class, 1);
+
+		if ($test === null) {
+			throw new RuntimeException('Sorry, but no...');
+		}
+
 		$test->doSomething();
 	}
 
 	public function getPartialReferenceDynamicType(): void
 	{
 		$test = $this->entityManager->getPartialReference(MyEntity::class, 1);
+
+		if ($test === null) {
+			throw new RuntimeException('Sorry, but no...');
+		}
+
 		$test->doSomething();
 	}
 }