Merge pull request #33 from lovasoa/master

Fix invalid serialization for lists of unions
adamboche · May 29, 2020 · 4821724 · 4821724
2 parents 58bfc9f + 78c654c
commit 4821724
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/src/marshmallow_union/__init__.py b/src/marshmallow_union/__init__.py
@@ -13,6 +13,7 @@ class MarshmallowUnionException(Exception):
 
 class ExceptionGroup(MarshmallowUnionException):
     """Collection of possibly multiple exceptions."""
+
     def __init__(self, msg: str, errors):
         self.msg = msg
         self.errors = errors
@@ -63,8 +64,8 @@ def _serialize(self, value: t.Any, attr: str, obj: str, **kwargs):
         for candidate_field in fields:
 
             try:
-                return candidate_field.serialize(
-                    attr, obj, error_store=error_store, **kwargs
+                return candidate_field._serialize(
+                    value, attr, obj, error_store=error_store, **kwargs
                 )
             except ValueError as e:
                 error_store.store_error({attr: e})

diff --git a/tests/test_union.py b/tests/test_union.py
@@ -52,13 +52,22 @@ class IntStrSchema(marshmallow.Schema):
     x = marshmallow_union.Union([marshmallow.fields.Int(), marshmallow.fields.String()])
 
 
+class ListUnionSchema(marshmallow.Schema):
+    """Schema with a list of unions."""
+
+    l = marshmallow.fields.List(
+        marshmallow_union.Union([marshmallow.fields.Int(), marshmallow.fields.String()])
+    )
+
+
 @pytest.mark.parametrize(
     "data, schema",
     [
         ({"name": "Alice", "number_or_numbers": 25}, PersonSchema()),
         ({"name": "Alice", "number_or_numbers": [25, 50]}, PersonSchema()),
         ({"name": "Alice", "number_or_numbers": [25, 50]}, OtherSchema()),
         ({"x": 5}, IntStrSchema()),
+        ({"l": ["h", 5, "n", 1]}, ListUnionSchema()),
         ({"x": "hello"}, IntStrSchema()),
         ({"items": {"a": 42, "b": [17]}}, MappingSchema()),
     ],