Make assertion types class a parameter

Author: Slamdunk

extension.neon

@@ -20,10 +20,14 @@ services:
 			- phpstan.typeSpecifier.functionTypeSpecifyingExtension
 	-
 		class: PHPStan\Type\PHPUnit\Assert\AssertMethodTypeSpecifyingExtension
+		arguments:
+			classWithAssertionMethods: PHPUnit\Framework\TestCase
 		tags:
 			- phpstan.typeSpecifier.methodTypeSpecifyingExtension
 	-
 		class: PHPStan\Type\PHPUnit\Assert\AssertStaticMethodTypeSpecifyingExtension
+		arguments:
+			classWithAssertionMethods: PHPUnit\Framework\Assert
 		tags:
 			- phpstan.typeSpecifier.staticMethodTypeSpecifyingExtension
 	-

src/Type/PHPUnit/Assert/AssertMethodTypeSpecifyingExtension.php

@@ -14,17 +14,28 @@
 class AssertMethodTypeSpecifyingExtension implements MethodTypeSpecifyingExtension, TypeSpecifierAwareExtension
 {
 
+	/** @var class-string */
+	private $classWithAssertionMethods;
+
 	/** @var TypeSpecifier */
 	private $typeSpecifier;
 
+	/**
+	 * @param class-string $classWithAssertionMethods
+	 */
+	public function __construct(string $classWithAssertionMethods)
+	{
+		$this->classWithAssertionMethods = $classWithAssertionMethods;
+	}
+
 	public function setTypeSpecifier(TypeSpecifier $typeSpecifier): void
 	{
 		$this->typeSpecifier = $typeSpecifier;
 	}
 
 	public function getClass(): string
 	{
-		return 'PHPUnit\Framework\TestCase';
+		return $this->classWithAssertionMethods;
 	}
 
 	public function isMethodSupported(

src/Type/PHPUnit/Assert/AssertStaticMethodTypeSpecifyingExtension.php

@@ -14,17 +14,28 @@
 class AssertStaticMethodTypeSpecifyingExtension implements StaticMethodTypeSpecifyingExtension, TypeSpecifierAwareExtension
 {
 
+	/** @var class-string */
+	private $classWithAssertionMethods;
+
 	/** @var TypeSpecifier */
 	private $typeSpecifier;
 
+	/**
+	 * @param class-string $classWithAssertionMethods
+	 */
+	public function __construct(string $classWithAssertionMethods)
+	{
+		$this->classWithAssertionMethods = $classWithAssertionMethods;
+	}
+
 	public function setTypeSpecifier(TypeSpecifier $typeSpecifier): void
 	{
 		$this->typeSpecifier = $typeSpecifier;
 	}
 
 	public function getClass(): string
 	{
-		return 'PHPUnit\Framework\Assert';
+		return $this->classWithAssertionMethods;
 	}
 
 	public function isStaticMethodSupported(

tests/Rules/PHPUnit/AssertSameMethodDifferentTypesRuleTest.php

@@ -24,7 +24,7 @@ protected function getRule(): Rule
 	protected function getMethodTypeSpecifyingExtensions(): array
 	{
 		return [
-			new AssertMethodTypeSpecifyingExtension(),
+			new AssertMethodTypeSpecifyingExtension('PHPUnit\Framework\TestCase'),
 		];
 	}
 

tests/Rules/PHPUnit/AssertSameStaticMethodDifferentTypesRuleTest.php

@@ -24,7 +24,7 @@ protected function getRule(): Rule
 	protected function getStaticMethodTypeSpecifyingExtensions(): array
 	{
 		return [
-			new AssertStaticMethodTypeSpecifyingExtension(),
+			new AssertStaticMethodTypeSpecifyingExtension('PHPUnit\Framework\Assert'),
 		];
 	}