[PhpUnitBridge] Add enum_exists mock

alexandre-daubois · alexandre-daubois · commit f53feaeae196 · 2022-12-08T13:25:53.000+01:00
diff --git a/src/Symfony/Bridge/PhpUnit/CHANGELOG.md b/src/Symfony/Bridge/PhpUnit/CHANGELOG.md
@@ -1,6 +1,11 @@
 CHANGELOG
 =========
 
+6.3
+---
+
+ * Add support for mocking the `enum_exists` function
+
 6.2
 ---
 
diff --git a/src/Symfony/Bridge/PhpUnit/ClassExistsMock.php b/src/Symfony/Bridge/PhpUnit/ClassExistsMock.php
@@ -49,6 +49,17 @@ public static function trait_exists($name, $autoload = true)
         return isset(self::$classes[$name]) ? (bool) self::$classes[$name] : \trait_exists($name, $autoload);
     }
 
+    public static function enum_exists($name, $autoload = true)
+    {
+        $name = ltrim($name, '\\');
+
+        if (\PHP_VERSION_ID < 80100) {
+            throw new \LogicException('"enum_exists" cannot be used with a version of PHP earlier than 8.1.');
+        }
+
+        return isset(self::$classes[$name]) ? (bool) self::$classes[$name] : \enum_exists($name, $autoload);
+    }
+
     public static function register($class)
     {
         $self = static::class;
@@ -61,7 +72,7 @@ public static function register($class)
             $mockedNs[] = substr($class, 6, strrpos($class, '\\') - 6);
         }
         foreach ($mockedNs as $ns) {
-            foreach (['class', 'interface', 'trait'] as $type) {
+            foreach (['class', 'interface', 'trait', 'enum'] as $type) {
                 if (\function_exists($ns.'\\'.$type.'_exists')) {
                     continue;
                 }
diff --git a/src/Symfony/Bridge/PhpUnit/Tests/EnumExistsMockTest.php b/src/Symfony/Bridge/PhpUnit/Tests/EnumExistsMockTest.php
@@ -0,0 +1,62 @@
+<?php
+
+/*
+ * This file is part of the Symfony package.
+ *
+ * (c) Fabien Potencier <fabien@symfony.com>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Symfony\Bridge\PhpUnit\Tests;
+
+use PHPUnit\Framework\TestCase;
+use Symfony\Bridge\PhpUnit\ClassExistsMock;
+
+class EnumExistsMockTest extends TestCase
+{
+    public static function setUpBeforeClass(): void
+    {
+        ClassExistsMock::register(__CLASS__);
+    }
+
+    protected function setUp(): void
+    {
+        ClassExistsMock::withMockedClasses([
+            ExistingEnum::class => false,
+            'NonExistingEnum' => true,
+        ]);
+    }
+
+    /**
+     * @requires PHP 8.1
+     */
+    public function testEnumExists()
+    {
+        $this->assertFalse(enum_exists(ExistingEnum::class));
+        $this->assertFalse(enum_exists(ExistingEnum::class, false));
+        $this->assertFalse(enum_exists('\\'.ExistingEnum::class));
+        $this->assertFalse(enum_exists('\\'.ExistingEnum::class, false));
+        $this->assertTrue(enum_exists('NonExistingEnum'));
+        $this->assertTrue(enum_exists('NonExistingEnum', false));
+        $this->assertTrue(enum_exists('\\NonExistingEnum'));
+        $this->assertTrue(enum_exists('\\NonExistingEnum', false));
+        $this->assertTrue(enum_exists(ExistingEnumReal::class));
+        $this->assertTrue(enum_exists(ExistingEnumReal::class, false));
+        $this->assertTrue(enum_exists('\\'.ExistingEnumReal::class));
+        $this->assertTrue(enum_exists('\\'.ExistingEnumReal::class, false));
+        $this->assertFalse(enum_exists('NonExistingClassReal'));
+        $this->assertFalse(enum_exists('NonExistingClassReal', false));
+        $this->assertFalse(enum_exists('\\NonExistingEnumReal'));
+        $this->assertFalse(enum_exists('\\NonExistingEnumReal', false));
+    }
+}
+
+enum ExistingEnum
+{
+}
+
+enum ExistingEnumReal
+{
+}
