[rector] Rector fixes

rectorphp · Jun 24, 2024 · ef1ab31 · ef1ab31
1 parent 0b21e2f
commit ef1ab31
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/...lity/Rector/Class_/AddReturnDocBlockToCollectionPropertyGetterByToManyAttributeRector.php b/...lity/Rector/Class_/AddReturnDocBlockToCollectionPropertyGetterByToManyAttributeRector.php
@@ -4,6 +4,7 @@
 
 namespace Rector\Doctrine\CodeQuality\Rector\Class_;
 
+use PhpParser\Node\Attribute;
 use PhpParser\Node;
 use PhpParser\Node\Expr\ClassConstFetch;
 use PhpParser\Node\Stmt\Class_;
@@ -102,13 +103,13 @@ public function refactorWithScope(Node $node, Scope $scope): ?Node
 
             $property = $this->methodUniqueReturnedPropertyResolver->resolve($node, $classMethod);
 
-            if ($property === null) {
+            if (!$property instanceof Property) {
                 continue;
             }
 
             $collectionObjectType = $this->getCollectionObjectTypeFromToManyAttribute($property);
 
-            if ($collectionObjectType === null) {
+            if (!$collectionObjectType instanceof FullyQualifiedObjectType) {
                 return null;
             }
 
@@ -153,6 +154,6 @@ private function isDoctrineEntityClass(Class_ $class): bool
             ['Doctrine\ORM\Mapping\Entity', 'Doctrine\ORM\Mapping\Embeddable'],
         );
 
-        return $entityAttribute !== null;
+        return $entityAttribute instanceof Attribute;
     }
 }