Allow enums in array codec

edgedb · May 26, 2023 · 2de7e3f · 2de7e3f
1 parent 03e4012
commit 2de7e3f
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/edgedb/protocol/codecs/array.pyx b/edgedb/protocol/codecs/array.pyx
@@ -39,7 +39,7 @@ cdef class BaseArrayCodec(BaseCodec):
 
         if not isinstance(
             self.sub_codec,
-            (ScalarCodec, TupleCodec, NamedTupleCodec, RangeCodec)
+            (ScalarCodec, TupleCodec, NamedTupleCodec, RangeCodec, EnumCodec)
         ):
             raise TypeError(
                 'only arrays of scalars are supported (got type {!r})'.format(

diff --git a/tests/test_enum.py b/tests/test_enum.py
@@ -99,3 +99,12 @@ async def test_enum_03(self):
         c_red = await self.client.query_single('SELECT <Color>"red"')
         c_red2 = await self.client.query_single('SELECT <Color>$0', c_red)
         self.assertIs(c_red, c_red2)
+
+    async def test_enum_04(self):
+        enums = await self.client.query_single(
+            'SELECT <array<Color>>$0', ['red', 'white']
+        )
+        enums2 = await self.client.query_single(
+            'SELECT <array<Color>>$0', enums
+        )
+        self.assertEqual(enums, enums2)