From 39270316e0160e26fa3150d224e8f4d14367d6e9 Mon Sep 17 00:00:00 2001
From: Abdul Malik Ikhsan <samsonasik@gmail.com>
Date: Sat, 22 Feb 2025 00:07:55 +0700
Subject: [PATCH] [Attribute] Add CoversTrait support on
 CoversAnnotationWithValueToAttributeRector

---
 .../Fixture/covers_trait.php.inc              | 33 +++++++++++++++++++
 .../Source/ExistingTrait.php                  | 19 +++++++++++
 ...rsAnnotationWithValueToAttributeRector.php | 14 ++++++++
 3 files changed, 66 insertions(+)
 create mode 100644 rules-tests/AnnotationsToAttributes/Rector/Class_/CoversAnnotationWithValueToAttributeRector/Fixture/covers_trait.php.inc
 create mode 100644 rules-tests/AnnotationsToAttributes/Rector/Class_/CoversAnnotationWithValueToAttributeRector/Source/ExistingTrait.php

diff --git a/rules-tests/AnnotationsToAttributes/Rector/Class_/CoversAnnotationWithValueToAttributeRector/Fixture/covers_trait.php.inc b/rules-tests/AnnotationsToAttributes/Rector/Class_/CoversAnnotationWithValueToAttributeRector/Fixture/covers_trait.php.inc
new file mode 100644
index 00000000..b73915a5
--- /dev/null
+++ b/rules-tests/AnnotationsToAttributes/Rector/Class_/CoversAnnotationWithValueToAttributeRector/Fixture/covers_trait.php.inc
@@ -0,0 +1,33 @@
+<?php
+
+namespace Rector\PHPUnit\Tests\AnnotationsToAttributes\Rector\Class_\CoversAnnotationWithValueToAttributeRector\Fixture;
+
+use PHPUnit\Framework\TestCase;
+
+/**
+ * @covers \Rector\PHPUnit\Tests\AnnotationsToAttributes\Rector\Class_\CoversAnnotationWithValueToAttributeRector\Source\ExistingTrait
+ */
+final class CoversTrait extends TestCase
+{
+    public function test()
+    {
+    }
+}
+
+?>
+-----
+<?php
+
+namespace Rector\PHPUnit\Tests\AnnotationsToAttributes\Rector\Class_\CoversAnnotationWithValueToAttributeRector\Fixture;
+
+use PHPUnit\Framework\TestCase;
+
+#[\PHPUnit\Framework\Attributes\CoversTrait(\Rector\PHPUnit\Tests\AnnotationsToAttributes\Rector\Class_\CoversAnnotationWithValueToAttributeRector\Source\ExistingTrait::class)]
+final class CoversTrait extends TestCase
+{
+    public function test()
+    {
+    }
+}
+
+?>
diff --git a/rules-tests/AnnotationsToAttributes/Rector/Class_/CoversAnnotationWithValueToAttributeRector/Source/ExistingTrait.php b/rules-tests/AnnotationsToAttributes/Rector/Class_/CoversAnnotationWithValueToAttributeRector/Source/ExistingTrait.php
new file mode 100644
index 00000000..c9bf4548
--- /dev/null
+++ b/rules-tests/AnnotationsToAttributes/Rector/Class_/CoversAnnotationWithValueToAttributeRector/Source/ExistingTrait.php
@@ -0,0 +1,19 @@
+<?php
+declare(strict_types=1);
+
+namespace Rector\PHPUnit\Tests\AnnotationsToAttributes\Rector\Class_\CoversAnnotationWithValueToAttributeRector\Source;
+
+trait ExistingTrait
+{
+    public function __construct()
+    {
+    }
+
+    public function foo(): void
+    {
+    }
+
+    public function bar(): void
+    {
+    }
+}
diff --git a/rules/AnnotationsToAttributes/Rector/Class_/CoversAnnotationWithValueToAttributeRector.php b/rules/AnnotationsToAttributes/Rector/Class_/CoversAnnotationWithValueToAttributeRector.php
index 559e90d1..f80fb912 100644
--- a/rules/AnnotationsToAttributes/Rector/Class_/CoversAnnotationWithValueToAttributeRector.php
+++ b/rules/AnnotationsToAttributes/Rector/Class_/CoversAnnotationWithValueToAttributeRector.php
@@ -38,6 +38,11 @@ final class CoversAnnotationWithValueToAttributeRector extends AbstractRector im
      */
     private const COVERTS_CLASS_ATTRIBUTE = 'PHPUnit\Framework\Attributes\CoversClass';
 
+    /**
+     * @var string
+     */
+    private const COVERTS_TRAIT_ATTRIBUTE = 'PHPUnit\Framework\Attributes\CoversTrait';
+
     /**
      * @var string
      */
@@ -150,6 +155,15 @@ private function createAttributeGroup(string $annotationValue): AttributeGroup
             $attributeValue = [$this->getClass($annotationValue) . '::class', $this->getMethod($annotationValue)];
         } else {
             $attributeClass = self::COVERTS_CLASS_ATTRIBUTE;
+
+            if ($this->reflectionProvider->hasClass($annotationValue)) {
+                $classReflection = $this->reflectionProvider->getClass($annotationValue);
+
+                if ($classReflection->isTrait()) {
+                    $attributeClass = self::COVERTS_TRAIT_ATTRIBUTE;
+                }
+            }
+
             $attributeValue = [trim($annotationValue) . '::class'];
         }