Rewrite more tests to use TypeInferenceTestCase instead of LevelsTest…

…Case

tests/DoctrineIntegration/ODM/DocumentManagerTypeInferenceTest.php

@@ -0,0 +1,44 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\DoctrineIntegration\ODM;
+
+use PHPStan\Testing\TypeInferenceTestCase;
+use const PHP_VERSION_ID;
+
+class DocumentManagerTypeInferenceTest extends TypeInferenceTestCase
+{
+
+	/**
+	 * @return iterable<mixed>
+	 */
+	public function dataFileAsserts(): iterable
+	{
+		if (PHP_VERSION_ID >= 80000) {
+			return [];
+		}
+
+		yield from $this->gatherAssertTypes(__DIR__ . '/data/documentManagerDynamicReturn.php');
+		yield from $this->gatherAssertTypes(__DIR__ . '/data/documentRepositoryDynamicReturn.php');
+		yield from $this->gatherAssertTypes(__DIR__ . '/data/documentManagerMergeReturn.php');
+		yield from $this->gatherAssertTypes(__DIR__ . '/data/customRepositoryUsage.php');
+	}
+
+	/**
+	 * @dataProvider dataFileAsserts
+	 * @param mixed ...$args
+	 */
+	public function testFileAsserts(
+		string $assertType,
+		string $file,
+		...$args
+	): void
+	{
+		$this->assertFileAsserts($assertType, $file, ...$args);
+	}
+
+	public static function getAdditionalConfigFiles(): array
+	{
+		return [__DIR__ . '/phpstan.neon'];
+	}
+
+}

tests/DoctrineIntegration/ODM/data/customRepositoryUsage.php

@@ -7,6 +7,7 @@
 use Doctrine\ODM\MongoDB\Mapping\Annotations\Document;
 use Doctrine\ODM\MongoDB\Mapping\Annotations\Id;
 use RuntimeException;
+use function PHPStan\Testing\assertType;
 
 class Example
 {
@@ -23,13 +24,9 @@ public function __construct(DocumentManager $documentManager)
 	public function get(): void
 	{
 		$test = $this->repository->get('testing');
+		assertType(MyDocument::class, $test);
 		$test->doSomethingElse();
 	}
-
-	public function nonexistant(): void
-	{
-		$this->repository->nonexistant();
-	}
 }
 
 /**
@@ -62,6 +59,8 @@ public function get(string $id): MyDocument
 			throw new RuntimeException('Not found...');
 		}
 
+		assertType(MyDocument::class, $document);
+
 		return $document;
 	}
 }

tests/DoctrineIntegration/ODM/data/documentManagerDynamicReturn.php

@@ -6,6 +6,7 @@
 use Doctrine\ODM\MongoDB\Mapping\Annotations\Document;
 use Doctrine\ODM\MongoDB\Mapping\Annotations\Id;
 use RuntimeException;
+use function PHPStan\Testing\assertType;
 
 class Example
 {
@@ -27,20 +28,24 @@ public function findDynamicType(): void
 			throw new RuntimeException('Sorry, but no...');
 		}
 
+		assertType(MyDocument::class, $test);
+
 		$test->doSomething();
 		$test->doSomethingElse();
 	}
 
 	public function getReferenceDynamicType(): void
 	{
 		$test = $this->documentManager->getReference(MyDocument::class, 'blah-123');
+		assertType(MyDocument::class, $test);
 		$test->doSomething();
 		$test->doSomethingElse();
 	}
 
 	public function getPartialReferenceDynamicType(): void
 	{
 		$test = $this->documentManager->getPartialReference(MyDocument::class, 'blah-123');
+		assertType(MyDocument::class, $test);
 		$test->doSomething();
 		$test->doSomethingElse();
 	}

tests/DoctrineIntegration/ODM/data/documentManagerMergeReturn.php

@@ -5,6 +5,7 @@
 use Doctrine\ODM\MongoDB\DocumentManager;
 use Doctrine\ODM\MongoDB\Mapping\Annotations\Document;
 use Doctrine\ODM\MongoDB\Mapping\Annotations\Id;
+use function PHPStan\Testing\assertType;
 
 class Example
 {
@@ -21,6 +22,7 @@ public function __construct(DocumentManager $documentManager)
 	public function merge(): void
 	{
 		$test = $this->documentManager->merge(new MyDocument());
+		assertType(MyDocument::class, $test);
 		$test->doSomething();
 	}
 }

tests/DoctrineIntegration/ODM/data/documentRepositoryDynamicReturn.php

@@ -7,6 +7,7 @@
 use Doctrine\ODM\MongoDB\Mapping\Annotations\Document;
 use Doctrine\ODM\MongoDB\Mapping\Annotations\Id;
 use RuntimeException;
+use function PHPStan\Testing\assertType;
 
 class Example
 {
@@ -28,6 +29,8 @@ public function findDynamicType(): void
 			throw new RuntimeException('Sorry, but no...');
 		}
 
+		assertType('object', $test);
+
 		$test->doSomething();
 		$test->doSomethingElse();
 	}
@@ -40,13 +43,16 @@ public function findOneByDynamicType(): void
 			throw new RuntimeException('Sorry, but no...');
 		}
 
+		assertType('object', $test);
+
 		$test->doSomething();
 		$test->doSomethingElse();
 	}
 
 	public function findAllDynamicType(): void
 	{
 		$items = $this->repository->findAll();
+		assertType('array<int, object>', $items);
 
 		foreach ($items as $test) {
 			$test->doSomething();
@@ -57,6 +63,7 @@ public function findAllDynamicType(): void
 	public function findByDynamicType(): void
 	{
 		$items = $this->repository->findBy(['blah' => 'testing']);
+		assertType('array<int, object>', $items);
 
 		foreach ($items as $test) {
 			$test->doSomething();
@@ -85,6 +92,8 @@ public function findDynamicType(): void
 			throw new RuntimeException('Sorry, but no...');
 		}
 
+		assertType(MyDocument::class, $test);
+
 		$test->doSomething();
 		$test->doSomethingElse();
 	}
@@ -97,13 +106,16 @@ public function findOneByDynamicType(): void
 			throw new RuntimeException('Sorry, but no...');
 		}
 
+		assertType(MyDocument::class, $test);
+
 		$test->doSomething();
 		$test->doSomethingElse();
 	}
 
 	public function findAllDynamicType(): void
 	{
 		$items = $this->repository->findAll();
+		assertType('array<int, ' . MyDocument::class . '>', $items);
 
 		foreach ($items as $test) {
 			$test->doSomething();
@@ -114,6 +126,7 @@ public function findAllDynamicType(): void
 	public function findByDynamicType(): void
 	{
 		$items = $this->repository->findBy(['blah' => 'testing']);
+		assertType('array<int, ' . MyDocument::class . '>', $items);
 
 		foreach ($items as $test) {
 			$test->doSomething();

tests/DoctrineIntegration/ODM/phpstan.neon

@@ -1,6 +1,5 @@
 includes:
 	- ../../../extension.neon
-	- phar://phpstan.phar/conf/bleedingEdge.neon
 
 parameters:
 	doctrine: