Correct parsing for union

antidot · Jul 19, 2024 · efa3811 · efa3811
1 parent 0ecb9ca
commit efa3811
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/src/pyckson/parsers/base.py b/src/pyckson/parsers/base.py
@@ -104,5 +104,8 @@ def __init__(self, value_parsers: list[Parser]):
     def parse(self, json_value):
         for parser in self.value_parsers:
             if hasattr(parser, 'cls') and isinstance(json_value, parser.cls):
-                return parser.parse(json_value)
+                try:
+                    return parser.parse(json_value)
+                except:
+                    pass
         raise TypeError(f'{json_value} is not compatible with Union type in Pyckson.')
diff --git a/tests/parsers/test_base.py b/tests/parsers/test_base.py
@@ -44,6 +44,14 @@ def test_should_not_raise_if_parser_does_not_have_cls(self):
 
         assert_that(result).is_equal_to(5)
 
+    def test_should_parse_list_of_list_in_union(self):
+        parser = UnionParser([ListParser(BasicParserWithCast(int)), ListParser(ListParser(BasicParserWithCast(int)))])
+
+        result = parser.parse([[5], [6]])
+
+        assert result == [[5], [6]]
+
+
 
 class TestListParser:
     def test_should_accept_list(self):