Fix Mongo stubs

Author: ondrejmirtes

src/Type/Doctrine/ObjectMetadataResolver.php

@@ -79,7 +79,7 @@ private function getResolvedRepositoryClass(): string
 		if ($this->repositoryClass !== null) {
 			return $this->resolvedRepositoryClass = $this->repositoryClass;
 		} elseif ($objectManager !== null && get_class($objectManager) === 'Doctrine\ODM\MongoDB\DocumentManager') {
-			return $this->resolvedRepositoryClass = 'Doctrine\ODM\MongoDB\DocumentRepository';
+			return $this->resolvedRepositoryClass = 'Doctrine\ODM\MongoDB\Repository\DocumentRepository';
 		}
 
 		return $this->resolvedRepositoryClass = 'Doctrine\ORM\EntityRepository';

stubs/DocumentManager.stub

@@ -9,7 +9,7 @@ class DocumentManager implements ObjectManager
 {
 
 	/**
-	 * @template T
+	 * @template T of object
 	 * @phpstan-param class-string<T> $documentName
 	 * @phpstan-param mixed  $identifier
 	 * @phpstan-param integer|null $lockMode
@@ -19,7 +19,7 @@ class DocumentManager implements ObjectManager
 	public function find($documentName, $identifier, $lockMode = null, $lockVersion = null);
 
 	/**
-	 * @template T
+	 * @template T of object
 	 * @phpstan-param T $document
 	 * @phpstan-return T
 	 */
@@ -33,15 +33,15 @@ class DocumentManager implements ObjectManager
 	public function getRepository($documentName);
 
 	/**
-	 * @template T
+	 * @template T of object
 	 * @phpstan-param class-string<T> $documentName
 	 * @phpstan-param mixed $identifier
-	 * @phpstan-return T|null
+	 * @phpstan-return T
 	 */
 	public function getReference($documentName, $identifier);
 
 	/**
-	 * @template T
+	 * @template T of object
 	 * @phpstan-param class-string<T> $documentName
 	 * @phpstan-param mixed $identifier
 	 *

tests/DoctrineIntegration/ODM/data/documentManagerDynamicReturn-2.json

@@ -6,12 +6,12 @@
     },
     {
         "message": "Call to an undefined method PHPStan\\DoctrineIntegration\\ODM\\DocumentManagerDynamicReturn\\MyDocument::doSomethingElse().",
-        "line": 43,
+        "line": 38,
         "ignorable": true
     },
     {
         "message": "Call to an undefined method PHPStan\\DoctrineIntegration\\ODM\\DocumentManagerDynamicReturn\\MyDocument::doSomethingElse().",
-        "line": 55,
+        "line": 45,
         "ignorable": true
     }
 ]

tests/DoctrineIntegration/ODM/data/documentManagerDynamicReturn.php

@@ -34,23 +34,13 @@ public function findDynamicType(): void
 	public function getReferenceDynamicType(): void
 	{
 		$test = $this->documentManager->getReference(MyDocument::class, 'blah-123');
-
-		if ($test === null) {
-			throw new RuntimeException('Sorry, but no...');
-		}
-
 		$test->doSomething();
 		$test->doSomethingElse();
 	}
 
 	public function getPartialReferenceDynamicType(): void
 	{
 		$test = $this->documentManager->getPartialReference(MyDocument::class, 'blah-123');
-
-		if ($test === null) {
-			throw new RuntimeException('Sorry, but no...');
-		}
-
 		$test->doSomething();
 		$test->doSomethingElse();
 	}

tests/DoctrineIntegration/ODM/data/documentRepositoryDynamicReturn-7.json

@@ -0,0 +1,22 @@
+[
+    {
+        "message": "Call to an undefined method object::doSomething().",
+        "line": 31,
+        "ignorable": true
+    },
+    {
+        "message": "Call to an undefined method object::doSomethingElse().",
+        "line": 32,
+        "ignorable": true
+    },
+    {
+        "message": "Call to an undefined method object::doSomething().",
+        "line": 43,
+        "ignorable": true
+    },
+    {
+        "message": "Call to an undefined method object::doSomethingElse().",
+        "line": 44,
+        "ignorable": true
+    }
+]