Add tests for DoctrineSelectableDynamicReturnTypeExtension

Author: vhenzl

phpstan.neon

@@ -1,4 +1,6 @@
 parameters:
 	ignoreErrors:
 		- '#but returns PHPUnit_Framework_MockObject_MockObject#'
+		- '#PHPUnit_Framework_MockObject_MockObject given#'
+		- '#Access to an undefined property PHPUnit_Framework_MockObject_MockObject::\$\w+#'
 

tests/PHPStan/Type/Doctrine/DoctrineSelectableDynamicReturnTypeExtensionTest.php

@@ -0,0 +1,73 @@
+<?php declare(strict_types = 1);
+
+namespace Tests\PHPStan\Type\Doctrine;
+
+use PhpParser\Node\Expr;
+use PhpParser\Node\Expr\MethodCall;
+use PHPStan\Analyser\Scope;
+use PHPStan\Reflection\MethodReflection;
+use PHPStan\Type\Doctrine\DoctrineSelectableDynamicReturnTypeExtension;
+use PHPStan\Type\ObjectType;
+use PHPStan\Type\Type;
+use PHPUnit\Framework\TestCase;
+
+final class DoctrineSelectableDynamicReturnTypeExtensionTest extends TestCase
+{
+
+	/** @var \PHPStan\Type\Doctrine\DoctrineSelectableDynamicReturnTypeExtension */
+	private $extension;
+
+	protected function setUp()
+	{
+		$this->extension = new DoctrineSelectableDynamicReturnTypeExtension();
+	}
+
+	/**
+	 * @return mixed[]
+	 */
+	public function dataIsMethodSupported(): array
+	{
+		return [
+			['matching', true],
+			['filter', false],
+			['foo', false],
+		];
+	}
+
+	/**
+	 * @dataProvider dataIsMethodSupported
+	 * @param string $method
+	 * @param bool $expectedResult
+	 */
+	public function testIsMethodSupported(string $method, bool $expectedResult)
+	{
+		$methodReflection = $this->createMock(MethodReflection::class);
+		$methodReflection->method('getName')->willReturn($method);
+		$this->assertSame($expectedResult, $this->extension->isMethodSupported($methodReflection));
+	}
+
+	public function testGetTypeFromMethodCall()
+	{
+		$methodReflection = $this->createMock(MethodReflection::class);
+
+		$scope = $this->createMock(Scope::class);
+		$scope->method('getType')->will(
+			self::returnCallback(
+				function (\PhpParser\Node\Expr $node): Type {
+					return new ObjectType($node->getType());
+				}
+			)
+		);
+
+		$var = $this->createMock(Expr::class);
+		$var->method('getType')->willReturn(\Doctrine\Common\Collections\Collection::class);
+		$methodCall = $this->createMock(MethodCall::class);
+		$methodCall->var = $var;
+
+		$resultType = $this->extension->getTypeFromMethodCall($methodReflection, $methodCall, $scope);
+
+		$this->assertInstanceOf(ObjectType::class, $resultType);
+		$this->assertSame(\Doctrine\Common\Collections\Collection::class, $resultType->describe());
+	}
+
+}