Treat enumerations as classes

Author: sebastianbergmann

src/StaticAnalysis/CodeUnitFindingVisitor.php

@@ -21,6 +21,7 @@
 use PhpParser\Node\NullableType;
 use PhpParser\Node\Stmt\Class_;
 use PhpParser\Node\Stmt\ClassMethod;
+use PhpParser\Node\Stmt\Enum_;
 use PhpParser\Node\Stmt\Function_;
 use PhpParser\Node\Stmt\Interface_;
 use PhpParser\Node\Stmt\Trait_;
@@ -81,6 +82,10 @@ public function enterNode(Node $node): null
             $this->processClass($node);
         }
 
+        if ($node instanceof Enum_) {
+            $this->processClass($node);
+        }
+
         if ($node instanceof Trait_) {
             $this->processTrait($node);
         }
@@ -255,19 +260,21 @@ private function processInterface(Interface_ $node): void
         );
     }
 
-    private function processClass(Class_ $node): void
+    private function processClass(Class_|Enum_ $node): void
     {
         $name           = $node->name->toString();
         $namespacedName = $node->namespacedName->toString();
         $parentClass    = null;
         $interfaces     = [];
 
-        if ($node->extends instanceof Name) {
-            $parentClass = $node->extends->toString();
-        }
+        if (!$node instanceof Enum_) {
+            if ($node->extends instanceof Name) {
+                $parentClass = $node->extends->toString();
+            }
 
-        foreach ($node->implements as $interface) {
-            $interfaces[] = $interface->toString();
+            foreach ($node->implements as $interface) {
+                $interfaces[] = $interface->toString();
+            }
         }
 
         $this->classes[$namespacedName] = new \SebastianBergmann\CodeCoverage\StaticAnalysis\Class_(

tests/_files/Target/TargetEnumeration.php

@@ -0,0 +1,8 @@
+<?php declare(strict_types=1);
+namespace SebastianBergmann\CodeCoverage\TestFixture\Target;
+
+enum TargetEnumeration
+{
+    case Foo;
+    case Bar;
+}

tests/tests/Target/MapBuilderTest.php

@@ -20,6 +20,7 @@
 use SebastianBergmann\CodeCoverage\StaticAnalysis\ParsingFileAnalyser;
 use SebastianBergmann\CodeCoverage\TestFixture\Target\T1;
 use SebastianBergmann\CodeCoverage\TestFixture\Target\T2;
+use SebastianBergmann\CodeCoverage\TestFixture\Target\TargetEnumeration;
 use SebastianBergmann\CodeCoverage\TestFixture\Target\TraitOne;
 use SebastianBergmann\CodeCoverage\TestFixture\Target\TraitTwo;
 
@@ -39,6 +40,7 @@ public static function provider(): array
         $traitOne  = realpath(__DIR__ . '/../../_files/Target/TraitOne.php');
         $traitTwo  = realpath(__DIR__ . '/../../_files/Target/TraitTwo.php');
         $twoTraits = realpath(__DIR__ . '/../../_files/Target/two_traits.php');
+        $enum      = realpath(__DIR__ . '/../../_files/Target/TargetEnumeration.php');
 
         return [
             'generic' => [
@@ -279,6 +281,44 @@ public static function provider(): array
                     $twoTraits,
                 ],
             ],
+            'enumeration' => [
+                [
+                    'namespaces' => [
+                        'SebastianBergmann' => [
+                            $enum => range(4, 8),
+                        ],
+                        'SebastianBergmann\\CodeCoverage' => [
+                            $enum => range(4, 8),
+                        ],
+                        'SebastianBergmann\\CodeCoverage\\TestFixture' => [
+                            $enum => range(4, 8),
+                        ],
+                        'SebastianBergmann\\CodeCoverage\\TestFixture\\Target' => [
+                            $enum => range(4, 8),
+                        ],
+                    ],
+                    'traits' => [
+                    ],
+                    'classes' => [
+                        TargetEnumeration::class => [
+                            $enum => range(4, 8),
+                        ],
+                    ],
+                    'classesThatExtendClass' => [
+                    ],
+                    'classesThatImplementInterface' => [
+                    ],
+                    'methods' => [
+                    ],
+                    'functions' => [
+                    ],
+                    'reverseLookup' => [
+                    ],
+                ],
+                [
+                    $enum,
+                ],
+            ],
         ];
     }
 

tests/tests/Target/MapperTest.php

@@ -20,6 +20,7 @@
 use PHPUnit\Framework\TestCase;
 use SebastianBergmann\CodeCoverage\Filter;
 use SebastianBergmann\CodeCoverage\StaticAnalysis\ParsingFileAnalyser;
+use SebastianBergmann\CodeCoverage\TestFixture\Target\TargetEnumeration;
 use SebastianBergmann\CodeCoverage\TestFixture\Target\TraitOne;
 
 /**
@@ -157,6 +158,17 @@ public static function provider(): array
                     ],
                 ),
             ],
+
+            'enumeration' => [
+                [
+                    realpath(__DIR__ . '/../../_files/Target/TargetEnumeration.php') => range(4, 8),
+                ],
+                TargetCollection::fromArray(
+                    [
+                        Target::forClass(TargetEnumeration::class),
+                    ],
+                ),
+            ],
         ];
     }