Support for makePartial()

Author: ondrejmirtes

extension.neon

@@ -13,6 +13,11 @@ services:
 		arguments:
 			stubInterfaceName: PHPStan\Mockery\Type\Expects
 
+	-
+		class: PHPStan\Mockery\Type\MakePartialDynamicReturnTypeExtension
+		tags:
+			- phpstan.broker.dynamicMethodReturnTypeExtension
+
 	-
 		class: PHPStan\Mockery\Type\StubDynamicReturnTypeExtension
 		tags:

src/Mockery/Type/MakePartialDynamicReturnTypeExtension.php

@@ -0,0 +1,29 @@
+<?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\Type;
+
+class MakePartialDynamicReturnTypeExtension implements DynamicMethodReturnTypeExtension
+{
+
+	public function getClass(): string
+	{
+		return 'Mockery\\MockInterface';
+	}
+
+	public function isMethodSupported(MethodReflection $methodReflection): bool
+	{
+		return $methodReflection->getName() === 'makePartial';
+	}
+
+	public function getTypeFromMethodCall(MethodReflection $methodReflection, MethodCall $methodCall, Scope $scope): Type
+	{
+		return $scope->getType($methodCall->var);
+	}
+
+}

tests/Mockery/MockeryTest.php

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