Add #[attribute] support to NoEntityMockingRule

composer.json

@@ -15,7 +15,7 @@
         "phpunit/phpunit": "^13.0",
         "symfony/framework-bundle": "^6.2",
         "illuminate/container": "^11.0",
-        "phpecs/phpecs": "^2.2",
+        "symplify/easy-coding-standard": "^13.1",
         "tomasvotruba/class-leak": "^2.1",
         "rector/rector": "^2.4",
         "phpstan/extension-installer": "^1.4",

rector.php

@@ -17,6 +17,8 @@
         StringClassNameToClassConstantRector::class => [
             __DIR__ . '/src/Enum',
             __DIR__ . '/src/Testing/PHPUnitTestAnalyser.php',
+            __DIR__ . '/src/Rules/NoEntityOutsideEntityNamespaceRule.php',
             __DIR__ . '/tests/Naming/ClassToSuffixResolverTest.php',
+            __DIR__ . '/src/Doctrine/DoctrineEntityDocumentAnalyser.php',
         ],
     ]);

src/Doctrine/DoctrineEntityDocumentAnalyser.php

@@ -14,13 +14,40 @@
      */
     private const array ENTITY_DOCBLOCK_MARKERS = ['@Document', '@ORM\\Document', '@Entity', '@ORM\\Entity'];
 
+    /**
+     * @var string[]
+     */
+    private const array ENTITY_ATTRIBUTES = [
+        'Doctrine\\ORM\\Mapping\\Entity',
+        'Doctrine\\ODM\\MongoDB\\Mapping\\Annotations\\Document',
+    ];
+
     public static function isEntityClass(ClassReflection $classReflection): bool
     {
+        if (self::hasEntityAttribute($classReflection)) {
+            return true;
+        }
+
         $resolvedPhpDocBlock = $classReflection->getResolvedPhpDoc();
         if (! $resolvedPhpDocBlock instanceof ResolvedPhpDocBlock) {
             return false;
         }
 
         return array_any(self::ENTITY_DOCBLOCK_MARKERS, fn (string $entityDocBlockMarker): bool => str_contains($resolvedPhpDocBlock->getPhpDocString(), $entityDocBlockMarker));
     }
+
+    private static function hasEntityAttribute(ClassReflection $classReflection): bool
+    {
+        $attributeReflections = $classReflection->getNativeReflection()
+            ->getAttributes();
+
+        return array_any(
+            $attributeReflections,
+            static fn ($reflectionAttribute): bool => in_array(
+                $reflectionAttribute->getName(),
+                self::ENTITY_ATTRIBUTES,
+                true
+            )
+        );
+    }
 }

stubs/Doctrine/ORM/Mapping/Entity.php

@@ -0,0 +1,12 @@
+<?php
+
+namespace Doctrine\ORM\Mapping;
+
+if (class_exists('Doctrine\ORM\Mapping\Entity')) {
+    return;
+}
+
+#[\Attribute(\Attribute::TARGET_CLASS)]
+final class Entity
+{
+}

tests/Rules/PHPUnit/NoEntityMockingRule/Fixture/MockingAttributeEntity.php

@@ -0,0 +1,14 @@
+<?php
+
+namespace Symplify\PHPStanRules\Tests\Rules\PHPUnit\NoEntityMockingRule\Fixture;
+
+use PHPUnit\Framework\TestCase;
+use Symplify\PHPStanRules\Tests\Rules\PHPUnit\NoEntityMockingRule\Source\SomeAttributeEntity;
+
+final class MockingAttributeEntity extends TestCase
+{
+    public function test(): void
+    {
+        $someEntityMock = $this->createMock(SomeAttributeEntity::class);
+    }
+}

tests/Rules/PHPUnit/NoEntityMockingRule/NoEntityMockingRuleTest.php

@@ -29,6 +29,8 @@ public static function provideData(): Iterator
     {
         yield [__DIR__ . '/Fixture/MockingEntity.php', [[NoEntityMockingRule::ERROR_MESSAGE, 12]]];
 
+        yield [__DIR__ . '/Fixture/MockingAttributeEntity.php', [[NoEntityMockingRule::ERROR_MESSAGE, 12]]];
+
         yield [__DIR__ . '/Fixture/MockingDocument.php', [[NoEntityMockingRule::ERROR_MESSAGE, 12]]];
 
         yield [__DIR__ . '/Fixture/SkipMockingOtherObject.php', []];

tests/Rules/PHPUnit/NoEntityMockingRule/Source/SomeAttributeEntity.php

@@ -0,0 +1,10 @@
+<?php
+
+namespace Symplify\PHPStanRules\Tests\Rules\PHPUnit\NoEntityMockingRule\Source;
+
+use Doctrine\ORM\Mapping as ORM;
+
+#[ORM\Entity]
+class SomeAttributeEntity
+{
+}