TestMethodsHelper - class methods cache

Author: ondrejmirtes

src/Rules/PHPUnit/TestMethodsHelper.php

@@ -9,6 +9,7 @@
 use PHPStan\Reflection\MethodReflection;
 use PHPStan\Type\FileTypeMapper;
 use PHPUnit\Framework\TestCase;
+use function array_key_exists;
 use function str_starts_with;
 use function strtolower;
 
@@ -19,6 +20,9 @@ final class TestMethodsHelper
 
 	private PHPUnitVersion $PHPUnitVersion;
 
+	/** @var array<string, array<ReflectionMethod>> */
+	private array $methodCache = [];
+
 	public function __construct(
 		FileTypeMapper $fileTypeMapper,
 		PHPUnitVersion $PHPUnitVersion
@@ -44,8 +48,11 @@ public function getTestMethodReflection(ClassReflection $classReflection, Method
 	 */
 	public function getTestMethods(ClassReflection $classReflection, Scope $scope): array
 	{
+		if (array_key_exists($classReflection->getName(), $this->methodCache)) {
+			return $this->methodCache[$classReflection->getName()];
+		}
 		if (!$classReflection->is(TestCase::class)) {
-			return [];
+			return $this->methodCache[$classReflection->getName()] = [];
 		}
 
 		$testMethods = [];
@@ -87,7 +94,7 @@ public function getTestMethods(ClassReflection $classReflection, Scope $scope):
 			$testMethods[] = $reflectionMethod;
 		}
 
-		return $testMethods;
+		return $this->methodCache[$classReflection->getName()] = $testMethods;
 	}
 
 	private function hasTestAnnotation(?ResolvedPhpDocBlock $phpDoc): bool