Merge pull request #301 from phpDocumentor/feature/add-support-for-us…

…ing-traits-in-enums Add support for using traits in enums
phpDocumentor · Dec 4, 2022 · 8ccd850 · 8ccd850
2 parents 9fac0fb + 8021b36
commit 8ccd850
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/src/phpDocumentor/Reflection/Php/Factory/TraitUse.php b/src/phpDocumentor/Reflection/Php/Factory/TraitUse.php
@@ -7,6 +7,7 @@
 use InvalidArgumentException;
 use phpDocumentor\Reflection\Fqsen;
 use phpDocumentor\Reflection\Php\Class_;
+use phpDocumentor\Reflection\Php\Enum_;
 use phpDocumentor\Reflection\Php\ProjectFactoryStrategy;
 use phpDocumentor\Reflection\Php\StrategyContainer;
 use phpDocumentor\Reflection\Php\Trait_;
@@ -31,8 +32,12 @@ public function create(ContextStack $context, object $object, StrategyContainer
 
         $class = $context->peek();
 
-        if ($class instanceof Class_ === false && $class instanceof Trait_ === false) {
-            throw new InvalidArgumentException('Traits can only be used in class or trait');
+        if (
+            $class instanceof Class_ === false
+            && $class instanceof Trait_ === false
+            && $class instanceof Enum_ === false
+        ) {
+            throw new InvalidArgumentException('Traits can only be used in classes, enums or other traits');
         }
 
         foreach ($object->traits as $trait) {

diff --git a/tests/unit/phpDocumentor/Reflection/Php/Factory/TraitUseTest.php b/tests/unit/phpDocumentor/Reflection/Php/Factory/TraitUseTest.php
@@ -8,6 +8,7 @@
 use phpDocumentor\Reflection\Element;
 use phpDocumentor\Reflection\Fqsen;
 use phpDocumentor\Reflection\Php\Class_ as Class_Element;
+use phpDocumentor\Reflection\Php\Enum_ as Enum_Element;
 use phpDocumentor\Reflection\Php\Interface_;
 use phpDocumentor\Reflection\Php\ProjectFactoryStrategies;
 use phpDocumentor\Reflection\Php\Trait_ as Trait_Element;
@@ -26,6 +27,7 @@ public function consumerProvider(): array
         return [
             [new Class_Element(new Fqsen('\MyClass'))],
             [new Trait_Element(new Fqsen('\MyTrait'))],
+            [new Enum_Element(new Fqsen('\MyEnum'), null)],
         ];
     }