AssertRuleHelper.php

<?php declare(strict_types = 1);

namespace PHPStan\Rules\PHPUnit;

use PhpParser\Node;
use PHPStan\Analyser\Scope;
use PHPStan\Type\ObjectType;
use function in_array;
use function strtolower;

class AssertRuleHelper
{

	/**
	 * @phpstan-assert-if-true Node\Expr\MethodCall|Node\Expr\StaticCall $node
	 */
	public static function isMethodOrStaticCallOnAssert(Node $node, Scope $scope): bool
	{
		if ($node instanceof Node\Expr\MethodCall) {
			$calledOnType = $scope->getType($node->var);
		} elseif ($node instanceof Node\Expr\StaticCall) {
			if ($node->class instanceof Node\Name) {
				$class = (string) $node->class;
				if (
					$scope->isInClass()
					&& in_array(
						strtolower($class),
						[
							'self',
							'static',
							'parent',
						],
						true,
					)
				) {
					$calledOnType = new ObjectType($scope->getClassReflection()->getName());
				} else {
					$calledOnType = new ObjectType($class);
				}
			} else {
				$calledOnType = $scope->getType($node->class);
			}
		} else {
			return false;
		}

		$testCaseType = new ObjectType('PHPUnit\Framework\Assert');

		return $testCaseType->isSuperTypeOf($calledOnType)->yes();
	}

}