Return an intersection type when nameMock is used (#19)

Author: martinssipenko

extension.neon

@@ -52,6 +52,11 @@ services:
 		tags:
 			- phpstan.broker.dynamicStaticMethodReturnTypeExtension
 
+	-
+		class: PHPStan\Mockery\Type\MockDynamicNamedMockReturnTypeExtension
+		tags:
+			- phpstan.broker.dynamicStaticMethodReturnTypeExtension
+
 	-
 		class: PHPStan\Mockery\PhpDoc\TypeNodeResolverExtension
 		tags:

src/Mockery/Type/MockDynamicNamedMockReturnTypeExtension.php

@@ -0,0 +1,72 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Mockery\Type;
+
+use PhpParser\Node\Expr\StaticCall;
+use PHPStan\Analyser\Scope;
+use PHPStan\Reflection\MethodReflection;
+use PHPStan\Type\Constant\ConstantStringType;
+use PHPStan\Type\DynamicStaticMethodReturnTypeExtension;
+use PHPStan\Type\ObjectType;
+use PHPStan\Type\Type;
+use PHPStan\Type\TypeCombinator;
+
+class MockDynamicNamedMockReturnTypeExtension implements DynamicStaticMethodReturnTypeExtension
+{
+
+	public function getClass(): string
+	{
+		return 'Mockery';
+	}
+
+	public function isStaticMethodSupported(MethodReflection $methodReflection): bool
+	{
+		return $methodReflection->getName() === 'namedMock';
+	}
+
+	public function getTypeFromStaticMethodCall(
+		MethodReflection $methodReflection,
+		StaticCall $methodCall,
+		Scope $scope
+	): Type
+	{
+		$defaultReturnType = new ObjectType('Mockery\\MockInterface');
+
+		$args = $methodCall->args;
+		if (count($args) > 1) {
+			array_shift($args);
+		}
+
+		$types = [$defaultReturnType];
+		foreach ($args as $arg) {
+			$classType = $scope->getType($arg->value);
+			if (!$classType instanceof ConstantStringType) {
+				continue;
+			}
+
+			$value = $classType->getValue();
+			if (substr($value, 0, 6) === 'alias:') {
+				$value = substr($value, 6);
+			}
+			if (substr($value, 0, 9) === 'overload:') {
+				$value = substr($value, 9);
+			}
+			if (substr($value, -1) === ']' && strpos($value, '[') !== false) {
+				$value = substr($value, 0, strpos($value, '['));
+			}
+
+			if (strpos($value, ',') !== false) {
+				$interfaceNames = explode(',', str_replace(' ', '', $value));
+			} else {
+				$interfaceNames = [$value];
+			}
+
+			foreach ($interfaceNames as $name) {
+				$types[] = new ObjectType($name);
+			}
+		}
+
+		return TypeCombinator::intersect(...$types);
+	}
+
+}

tests/Mockery/MockeryTest.php

@@ -113,6 +113,15 @@ public function testMakePartial(): void
 		self::assertSame('foo', $fooMock->doFoo());
 	}
 
+	public function testNamedMock(): void
+	{
+		$fooMock = \Mockery::namedMock('FooBar', Foo::class);
+		$this->requireFoo($fooMock);
+
+		$fooMock->allows()->doFoo()->andReturns('foo');
+		self::assertSame('foo', $fooMock->doFoo());
+	}
+
 	private function requireFoo(Foo $foo): void
 	{
 	}