add converter unit test for postgresql enum

gabledata · Jan 18, 2024 · e806095 · e806095
1 parent 22fe4e5
commit e806095
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/tests/unit/converters/test_postgresql.py b/tests/unit/converters/test_postgresql.py
@@ -4,6 +4,7 @@
 from recap.types import (
     BoolType,
     BytesType,
+    EnumType,
     FloatType,
     IntType,
     ListType,
@@ -239,6 +240,20 @@
                 scale=0,
             ),
         ),
+        (
+            {
+                'COLUMN_NAME': 'test_enum',
+                'DATA_TYPE': 'USER-DEFINED',
+                'CHARACTER_MAXIMUM_LENGTH': None,
+                'CHARACTER_OCTET_LENGTH': None,
+                'ENUM_VALUES': ['sad', 'ok', 'happy'],
+                'UDT_NAME': None,
+                'ATTNDIMS': 0,
+            },
+            EnumType(
+                symbols=['sad', 'ok', 'happy'],
+            ),
+        )
     ],
 )
 def test_postgresql_converter(column_props, expected):