Update test to expect a union(nullable, enum)

gabledata · Jan 18, 2024 · 22fe4e5 · 22fe4e5
1 parent fb2d487
commit 22fe4e5
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/tests/integration/clients/test_postgresql.py b/tests/integration/clients/test_postgresql.py
@@ -280,8 +280,13 @@ def test_struct_method_arrays_no_enforce_dimensions(self):
                     ),
                 ],
             ),
-            EnumType(
-                default=None, symbols=["sad", "ok", "happy"], name="test_enum_mood"
+            UnionType(
+                default=None,
+                name="test_enum_mood",
+                types=[
+                    NullType(),
+                    EnumType(default=None, symbols=["sad", "ok", "happy"]),
+                ],
             ),
         ]
         validate_results(test_types_struct, expected_fields)
@@ -465,6 +470,14 @@ def test_struct_method_arrays_enforce_dimensions(self):
                     ),
                 ],
             ),
+            UnionType(
+                default=None,
+                name="test_enum_mood",
+                types=[
+                    NullType(),
+                    EnumType(default=None, symbols=["sad", "ok", "happy"]),
+                ],
+            ),
         ]
         validate_results(test_types_struct, expected_fields)