Add tests

Author: forest-benchling

tests/test_parser/test_properties/test_init.py

@@ -293,6 +293,46 @@ def test_get_type_string(self, mocker, nullable, required, no_optional, json, ex
         )
         assert p.get_type_string(no_optional=no_optional, json=json) == expected
 
+    def test_get_base_type_string(self, mocker):
+        from openapi_python_client.parser.properties import UnionProperty
+
+        inner_property_1 = mocker.MagicMock()
+        inner_property_1.get_type_string.side_effect = (
+            lambda no_optional=False, json=False: "inner_json_type_string_1" if json else "inner_type_string_1"
+        )
+        inner_property_2 = mocker.MagicMock()
+        inner_property_2.get_type_string.side_effect = (
+            lambda no_optional=False, json=False: "inner_json_type_string_2" if json else "inner_type_string_2"
+        )
+        p = UnionProperty(
+            name="test",
+            required=False,
+            default=None,
+            inner_properties=[inner_property_1, inner_property_2],
+            nullable=True,
+        )
+        assert p.get_base_type_string() == "Union[inner_type_string_1, inner_type_string_2]"
+
+    def test_get_base_json_type_string(self, mocker):
+        from openapi_python_client.parser.properties import UnionProperty
+
+        inner_property_1 = mocker.MagicMock()
+        inner_property_1.get_type_string.side_effect = (
+            lambda no_optional=False, json=False: "inner_json_type_string_1" if json else "inner_type_string_1"
+        )
+        inner_property_2 = mocker.MagicMock()
+        inner_property_2.get_type_string.side_effect = (
+            lambda no_optional=False, json=False: "inner_json_type_string_2" if json else "inner_type_string_2"
+        )
+        p = UnionProperty(
+            name="test",
+            required=False,
+            default=None,
+            inner_properties=[inner_property_1, inner_property_2],
+            nullable=True,
+        )
+        assert p.get_base_json_type_string() == "Union[inner_json_type_string_1, inner_json_type_string_2]"
+
     def test_get_imports(self, mocker):
         from openapi_python_client.parser.properties import UnionProperty