Type-specifying extension for asserts (function and instance/static method calls)

Author: ondrejmirtes

extension.neon

@@ -6,6 +6,18 @@ parameters:
 			- markTestSkipped
 
 services:
+	-
+		class: PHPStan\Type\PHPUnit\Assert\AssertFunctionTypeSpecifyingExtension
+		tags:
+			- phpstan.typeSpecifier.functionTypeSpecifyingExtension
+	-
+		class: PHPStan\Type\PHPUnit\Assert\AssertMethodTypeSpecifyingExtension
+		tags:
+			- phpstan.typeSpecifier.methodTypeSpecifyingExtension
+	-
+		class: PHPStan\Type\PHPUnit\Assert\AssertStaticMethodTypeSpecifyingExtension
+		tags:
+			- phpstan.typeSpecifier.staticMethodTypeSpecifyingExtension
 	-
 		class: PHPStan\Type\PHPUnit\CreateMockDynamicReturnTypeExtension
 		tags:

src/Type/PHPUnit/Assert/AssertFunctionTypeSpecifyingExtension.php

@@ -0,0 +1,52 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Type\PHPUnit\Assert;
+
+use PhpParser\Node\Expr\FuncCall;
+use PHPStan\Analyser\Scope;
+use PHPStan\Analyser\SpecifiedTypes;
+use PHPStan\Analyser\TypeSpecifier;
+use PHPStan\Analyser\TypeSpecifierAwareExtension;
+use PHPStan\Analyser\TypeSpecifierContext;
+use PHPStan\Reflection\FunctionReflection;
+use PHPStan\Type\FunctionTypeSpecifyingExtension;
+
+class AssertFunctionTypeSpecifyingExtension implements FunctionTypeSpecifyingExtension, TypeSpecifierAwareExtension
+{
+
+	/** @var TypeSpecifier */
+	private $typeSpecifier;
+
+	public function setTypeSpecifier(TypeSpecifier $typeSpecifier): void
+	{
+		$this->typeSpecifier = $typeSpecifier;
+	}
+
+	public function isFunctionSupported(
+		FunctionReflection $functionReflection,
+		FuncCall $node,
+		TypeSpecifierContext $context
+	): bool
+	{
+		return AssertTypeSpecifyingExtensionHelper::isSupported(
+			$functionReflection->getName(),
+			$node->args
+		);
+	}
+
+	public function specifyTypes(
+		FunctionReflection $functionReflection,
+		FuncCall $node,
+		Scope $scope,
+		TypeSpecifierContext $context
+	): SpecifiedTypes
+	{
+		return AssertTypeSpecifyingExtensionHelper::specifyTypes(
+			$this->typeSpecifier,
+			$scope,
+			$functionReflection->getName(),
+			$node->args
+		);
+	}
+
+}

src/Type/PHPUnit/Assert/AssertMethodTypeSpecifyingExtension.php

@@ -0,0 +1,57 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Type\PHPUnit\Assert;
+
+use PhpParser\Node\Expr\MethodCall;
+use PHPStan\Analyser\Scope;
+use PHPStan\Analyser\SpecifiedTypes;
+use PHPStan\Analyser\TypeSpecifier;
+use PHPStan\Analyser\TypeSpecifierAwareExtension;
+use PHPStan\Analyser\TypeSpecifierContext;
+use PHPStan\Reflection\MethodReflection;
+use PHPStan\Type\MethodTypeSpecifyingExtension;
+
+class AssertMethodTypeSpecifyingExtension implements MethodTypeSpecifyingExtension, TypeSpecifierAwareExtension
+{
+
+	/** @var TypeSpecifier */
+	private $typeSpecifier;
+
+	public function setTypeSpecifier(TypeSpecifier $typeSpecifier): void
+	{
+		$this->typeSpecifier = $typeSpecifier;
+	}
+
+	public function getClass(): string
+	{
+		return \PHPUnit\Framework\TestCase::class;
+	}
+
+	public function isMethodSupported(
+		MethodReflection $methodReflection,
+		MethodCall $node,
+		TypeSpecifierContext $context
+	): bool
+	{
+		return AssertTypeSpecifyingExtensionHelper::isSupported(
+			$methodReflection->getName(),
+			$node->args
+		);
+	}
+
+	public function specifyTypes(
+		MethodReflection $functionReflection,
+		MethodCall $node,
+		Scope $scope,
+		TypeSpecifierContext $context
+	): SpecifiedTypes
+	{
+		return AssertTypeSpecifyingExtensionHelper::specifyTypes(
+			$this->typeSpecifier,
+			$scope,
+			$functionReflection->getName(),
+			$node->args
+		);
+	}
+
+}

src/Type/PHPUnit/Assert/AssertStaticMethodTypeSpecifyingExtension.php

@@ -0,0 +1,57 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Type\PHPUnit\Assert;
+
+use PhpParser\Node\Expr\StaticCall;
+use PHPStan\Analyser\Scope;
+use PHPStan\Analyser\SpecifiedTypes;
+use PHPStan\Analyser\TypeSpecifier;
+use PHPStan\Analyser\TypeSpecifierAwareExtension;
+use PHPStan\Analyser\TypeSpecifierContext;
+use PHPStan\Reflection\MethodReflection;
+use PHPStan\Type\StaticMethodTypeSpecifyingExtension;
+
+class AssertStaticMethodTypeSpecifyingExtension implements StaticMethodTypeSpecifyingExtension, TypeSpecifierAwareExtension
+{
+
+	/** @var TypeSpecifier */
+	private $typeSpecifier;
+
+	public function setTypeSpecifier(TypeSpecifier $typeSpecifier): void
+	{
+		$this->typeSpecifier = $typeSpecifier;
+	}
+
+	public function getClass(): string
+	{
+		return \PHPUnit\Framework\TestCase::class;
+	}
+
+	public function isStaticMethodSupported(
+		MethodReflection $methodReflection,
+		StaticCall $node,
+		TypeSpecifierContext $context
+	): bool
+	{
+		return AssertTypeSpecifyingExtensionHelper::isSupported(
+			$methodReflection->getName(),
+			$node->args
+		);
+	}
+
+	public function specifyTypes(
+		MethodReflection $functionReflection,
+		StaticCall $node,
+		Scope $scope,
+		TypeSpecifierContext $context
+	): SpecifiedTypes
+	{
+		return AssertTypeSpecifyingExtensionHelper::specifyTypes(
+			$this->typeSpecifier,
+			$scope,
+			$functionReflection->getName(),
+			$node->args
+		);
+	}
+
+}