assignToPropertyTypeInferer = $assignToPropertyTypeInferer; $this->staticTypeMapper = $staticTypeMapper; $this->constructorAssignDetector = $constructorAssignDetector; } public function getRuleDefinition() : RuleDefinition { return new RuleDefinition('Add typed property from assigned mock', [new CodeSample(<<<'CODE_SAMPLE' use PHPUnit\Framework\TestCase; final class SomeTest extends TestCase { private $someProperty; protected function setUp(): void { $this->someProperty = $this->createMock(SomeMockedClass::class); } } CODE_SAMPLE , <<<'CODE_SAMPLE' use PHPUnit\Framework\TestCase; final class SomeTest extends TestCase { private \PHPUnit\Framework\MockObject\MockObject $someProperty; protected function setUp(): void { $this->someProperty = $this->createMock(SomeMockedClass::class); } } CODE_SAMPLE )]); } public function getNodeTypes() : array { return [Class_::class]; } /** * @param Class_ $node */ public function refactor(Node $node) : ?Node { if (!$this->isObjectType($node, new ObjectType(ClassName::TEST_CASE_CLASS))) { return null; } $hasChanged = \false; foreach ($node->getProperties() as $property) { // already typed if ($property->type instanceof Node) { continue; } if (\count($property->props) !== 1) { continue; } $propertyName = (string) $this->getName($property); $type = $this->assignToPropertyTypeInferer->inferPropertyInClassLike($property, $propertyName, $node); if (!$type instanceof Type) { continue; } $propertyType = $this->staticTypeMapper->mapPHPStanTypeToPhpParserNode($type, TypeKind::PROPERTY); if (!$propertyType instanceof Node) { continue; } if (!$this->isObjectType($propertyType, new ObjectType(self::MOCK_OBJECT_CLASS))) { continue; } if (!$this->constructorAssignDetector->isPropertyAssigned($node, $propertyName)) { if (!$propertyType instanceof NullableType) { continue; } $property->props[0]->default = $this->nodeFactory->createNull(); } $property->type = $propertyType; $hasChanged = \true; } if (!$hasChanged) { return null; } return $node; } public function provideMinPhpVersion() : int { return PhpVersionFeature::TYPED_PROPERTIES; } }