Refactoring MyCLabsEnum type

Author: moufmouf

src/Mappers/Root/MyCLabsEnumTypeMapper.php

@@ -14,7 +14,7 @@
 use phpDocumentor\Reflection\Type;
 use phpDocumentor\Reflection\Types\Object_;
 use ReflectionMethod;
-use TheCodingMachine\GraphQLite\Types\MyclabsEnumType;
+use TheCodingMachine\GraphQLite\Types\MyCLabsEnumType;
 use function is_a;
 use function str_replace;
 use function strpos;
@@ -88,16 +88,7 @@ private function mapByClassName(string $enumClass): ?EnumType
             return $this->cache[$enumClass];
         }
 
-        $consts         = $enumClass::toArray();
-        $constInstances = [];
-        foreach ($consts as $key => $value) {
-            $constInstances[$key] = ['value' => $enumClass::$key()];
-        }
-
-        return $this->cache[$enumClass] = new MyclabsEnumType([
-            'name' => 'MyCLabsEnum_' . str_replace('\\', '__', $enumClass),
-            'values' => $constInstances,
-        ]);
+        return $this->cache[$enumClass] = new MyCLabsEnumType($enumClass);
     }
 
     /**

src/Types/MyCLabsEnumType.php

@@ -0,0 +1,46 @@
+<?php
+
+declare(strict_types=1);
+
+namespace TheCodingMachine\GraphQLite\Types;
+
+use GraphQL\Type\Definition\EnumType;
+use InvalidArgumentException;
+use MyCLabs\Enum\Enum;
+use function str_replace;
+
+/**
+ * An extension of the EnumType to support Myclabs enum.
+ *
+ * This implementation is needed to overwrite the default "serialize" method, that expects to see the exact same object
+ * (while there can be several instances of the same enum value with MyclabsEnum)
+ */
+class MyCLabsEnumType extends EnumType
+{
+    public function __construct(string $enumClassName)
+    {
+        $consts         = $enumClassName::toArray();
+        $constInstances = [];
+        foreach ($consts as $key => $value) {
+            $constInstances[$key] = ['value' => $enumClassName::$key()];
+        }
+
+        parent::__construct([
+            'name' => 'MyCLabsEnum_' . str_replace('\\', '__', $enumClassName),
+            'values' => $constInstances,
+        ]);
+    }
+
+    /**
+     * @param mixed $value
+     *
+     * @return mixed
+     */
+    public function serialize($value)
+    {
+        if (! $value instanceof Enum) {
+            throw new InvalidArgumentException('Expected a Myclabs Enum instance');
+        }
+        return $value->getKey();
+    }
+}

src/Types/MyclabsEnumType.php

@@ -0,0 +1,18 @@
+<?php
+
+namespace TheCodingMachine\GraphQLite\Types;
+
+use InvalidArgumentException;
+use PHPUnit\Framework\TestCase;
+use TheCodingMachine\GraphQLite\Fixtures\Integration\Models\ProductTypeEnum;
+
+class MyclabsEnumTypeTest extends TestCase
+{
+    public function testException()
+    {
+        $enumType = new MyCLabsEnumType(ProductTypeEnum::class);
+        $this->expectException(InvalidArgumentException::class);
+        $this->expectExceptionMessage('Expected a Myclabs Enum instance');
+        $enumType->serialize('foo');
+    }
+}