allows() support

Author: ondrejmirtes

extension.neon

@@ -1,4 +1,19 @@
 services:
+	-
+		class: PHPStan\Mockery\Reflection\AllowsMethodsClassReflectionExtension
+		tags:
+			- phpstan.broker.methodsClassReflectionExtension
+
+	-
+		class: PHPStan\Mockery\Type\AllowsDynamicReturnTypeExtension
+		tags:
+			- phpstan.broker.dynamicMethodReturnTypeExtension
+
+	-
+		class: PHPStan\Mockery\Type\ExpectationAfterAllowsDynamicReturnTypeExtension
+		tags:
+			- phpstan.broker.dynamicMethodReturnTypeExtension
+
 	-
 		class: PHPStan\Mockery\Type\MockDynamicReturnTypeExtension
 		tags:

src/Mockery/Reflection/AllowsMethodReflection.php

@@ -0,0 +1,65 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Mockery\Reflection;
+
+use PHPStan\Reflection\ClassMemberReflection;
+use PHPStan\Reflection\ClassReflection;
+use PHPStan\Reflection\MethodReflection;
+use PHPStan\Reflection\TrivialParametersAcceptor;
+
+class AllowsMethodReflection implements MethodReflection
+{
+
+	/** @var ClassReflection */
+	private $declaringClass;
+
+	/** @var string */
+	private $name;
+
+	public function __construct(ClassReflection $declaringClass, string $name)
+	{
+		$this->declaringClass = $declaringClass;
+		$this->name = $name;
+	}
+
+	public function getDeclaringClass(): ClassReflection
+	{
+		return $this->declaringClass;
+	}
+
+	public function isStatic(): bool
+	{
+		return false;
+	}
+
+	public function isPrivate(): bool
+	{
+		return false;
+	}
+
+	public function isPublic(): bool
+	{
+		return true;
+	}
+
+	public function getName(): string
+	{
+		return $this->name;
+	}
+
+	public function getPrototype(): ClassMemberReflection
+	{
+		return $this;
+	}
+
+	/**
+	 * @return \PHPStan\Reflection\ParametersAcceptor[]
+	 */
+	public function getVariants(): array
+	{
+		return [
+			new TrivialParametersAcceptor(),
+		];
+	}
+
+}

src/Mockery/Reflection/AllowsMethodsClassReflectionExtension.php

@@ -0,0 +1,23 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Mockery\Reflection;
+
+use PHPStan\Mockery\Type\Allows;
+use PHPStan\Reflection\ClassReflection;
+use PHPStan\Reflection\MethodReflection;
+use PHPStan\Reflection\MethodsClassReflectionExtension;
+
+class AllowsMethodsClassReflectionExtension implements MethodsClassReflectionExtension
+{
+
+	public function hasMethod(ClassReflection $classReflection, string $methodName): bool
+	{
+		return $classReflection->getName() === Allows::class;
+	}
+
+	public function getMethod(ClassReflection $classReflection, string $methodName): MethodReflection
+	{
+		return new AllowsMethodReflection($classReflection, $methodName);
+	}
+
+}

src/Mockery/Type/Allows.php

@@ -0,0 +1,8 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Mockery\Type;
+
+interface Allows
+{
+
+}

src/Mockery/Type/AllowsDynamicReturnTypeExtension.php

@@ -0,0 +1,47 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Mockery\Type;
+
+use PhpParser\Node\Expr\MethodCall;
+use PHPStan\Analyser\Scope;
+use PHPStan\Reflection\MethodReflection;
+use PHPStan\Reflection\ParametersAcceptorSelector;
+use PHPStan\Type\DynamicMethodReturnTypeExtension;
+use PHPStan\Type\IntersectionType;
+use PHPStan\Type\ObjectType;
+use PHPStan\Type\Type;
+use PHPStan\Type\TypeCombinator;
+use PHPStan\Type\TypeWithClassName;
+
+class AllowsDynamicReturnTypeExtension implements DynamicMethodReturnTypeExtension
+{
+
+	public function getClass(): string
+	{
+		return \Mockery\MockInterface::class;
+	}
+
+	public function isMethodSupported(MethodReflection $methodReflection): bool
+	{
+		return $methodReflection->getName() === 'allows';
+	}
+
+	public function getTypeFromMethodCall(MethodReflection $methodReflection, MethodCall $methodCall, Scope $scope): Type
+	{
+		$calledOnType = $scope->getType($methodCall->var);
+		$defaultType = ParametersAcceptorSelector::selectSingle($methodReflection->getVariants())->getReturnType();
+		if (!$calledOnType instanceof IntersectionType || count($calledOnType->getTypes()) !== 2) {
+			return $defaultType;
+		}
+		$mockedType = $calledOnType->getTypes()[1];
+		if (!$mockedType instanceof TypeWithClassName) {
+			return $defaultType;
+		}
+
+		return TypeCombinator::intersect(
+			new AllowsObjectType($mockedType->getClassName()),
+			new ObjectType(Allows::class)
+		);
+	}
+
+}

src/Mockery/Type/AllowsObjectType.php

@@ -0,0 +1,10 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Mockery\Type;
+
+use PHPStan\Type\ObjectType;
+
+class AllowsObjectType extends ObjectType
+{
+
+}

src/Mockery/Type/ExpectationAfterAllowsDynamicReturnTypeExtension.php

@@ -0,0 +1,30 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Mockery\Type;
+
+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;
+
+class ExpectationAfterAllowsDynamicReturnTypeExtension implements DynamicMethodReturnTypeExtension
+{
+
+	public function getClass(): string
+	{
+		return Allows::class;
+	}
+
+	public function isMethodSupported(MethodReflection $methodReflection): bool
+	{
+		return true;
+	}
+
+	public function getTypeFromMethodCall(MethodReflection $methodReflection, MethodCall $methodCall, Scope $scope): Type
+	{
+		return new ObjectType(\Mockery\CompositeExpectation::class);
+	}
+
+}