Fix DoctrineSelectableClassReflectionExtension

Author: ondrejmirtes

src/Reflection/Doctrine/DoctrineSelectableClassReflectionExtension.php

@@ -13,7 +13,7 @@ class DoctrineSelectableClassReflectionExtension implements MethodsClassReflecti
 	/** @var ReflectionProvider */
 	private $reflectionProvider;
 
-	public function setBroker(ReflectionProvider $reflectionProvider): void
+	public function __construct(ReflectionProvider $reflectionProvider)
 	{
 		$this->reflectionProvider = $reflectionProvider;
 	}

tests/Reflection/Doctrine/DoctrineSelectableClassReflectionExtensionTest.php

@@ -3,24 +3,22 @@
 namespace PHPStan\Reflection\Doctrine;
 
 use Doctrine\Common\Collections\Collection;
-use PHPStan\Broker\Broker;
+use PHPStan\Reflection\ReflectionProvider;
 use PHPStan\Testing\PHPStanTestCase;
 
 final class DoctrineSelectableClassReflectionExtensionTest extends PHPStanTestCase
 {
 
-	/** @var Broker */
-	private $broker;
+	/** @var ReflectionProvider */
+	private $reflectionProvider;
 
 	/** @var DoctrineSelectableClassReflectionExtension */
 	private $extension;
 
 	protected function setUp(): void
 	{
-		$this->broker = $this->createBroker();
-
-		$this->extension = new DoctrineSelectableClassReflectionExtension();
-		$this->extension->setBroker($this->broker);
+		$this->reflectionProvider = $this->createReflectionProvider();
+		$this->extension = new DoctrineSelectableClassReflectionExtension($this->reflectionProvider);
 	}
 
 	/**
@@ -39,13 +37,13 @@ public function dataHasMethod(): array
 	 */
 	public function testHasMethod(string $className, string $method, bool $expectedResult): void
 	{
-		$classReflection = $this->broker->getClass($className);
+		$classReflection = $this->reflectionProvider->getClass($className);
 		self::assertSame($expectedResult, $this->extension->hasMethod($classReflection, $method));
 	}
 
 	public function testGetMethod(): void
 	{
-		$classReflection = $this->broker->getClass(Collection::class);
+		$classReflection = $this->reflectionProvider->getClass(Collection::class);
 		$methodReflection = $this->extension->getMethod($classReflection, 'matching');
 		self::assertSame('matching', $methodReflection->getName());
 	}