TestCaseCreateMockDynamicReturnTypeExtension.php

<?php

declare(strict_types=1);

namespace PHPStan\Type\PHPUnit;

use Nette\PhpGenerator\ClassType;
use PhpParser\Node\Expr\MethodCall;
use PHPStan\Analyser\Scope;
use PHPStan\Reflection\MethodReflection;
use PHPStan\Type\DynamicMethodReturnTypeExtension;
use PHPStan\Type\ObjectType;
use PHPStan\Type\Type;
use PHPUnit\Framework\TestCase;

class TestCaseCreateMockDynamicReturnTypeExtension implements DynamicMethodReturnTypeExtension
{
	public static function getClass(): string
	{
		return TestCase::class;
	}

	public function isMethodSupported(MethodReflection $methodReflection): bool
	{
		return $methodReflection->getName() === 'createMock';
	}

	public function getTypeFromMethodCall(
		MethodReflection $methodReflection,
		MethodCall $methodCall,
		Scope $scope
	): Type {
		$class = (string) $methodCall->args[0]->value->class;
		if ($class === 'static') {
			return $methodReflection->getReturnType();
		} elseif ($class === 'self') {
			$class = $scope->getClass();
		}
		$rc = new \ReflectionClass($class);
		if ($rc->isFinal() || $rc->isTrait()) {
			return $methodReflection->getReturnType();
		}
		$proxy = 'proxyClass___' . md5($class);
		$this->generateProxy($rc, $proxy);
		return new ObjectType($proxy, false);
	}

	private function generateProxy(\ReflectionClass $rc, string $name)
	{
		$class = new ClassType($name);
		$class->setAbstract();
		$class->addImplement(\PHPUnit_Framework_MockObject_MockObject::class);
		$rc->isInterface() ? $class->addImplement($rc->getName()) : $class->setExtends($rc->getName());
		if (!class_exists($name)) {
			eval((string) $class);
		}
	}
}