Add missed tests related to quoted and lazy_import

Author: mtovt

tests/test_parser/test_properties/test_init.py

@@ -129,6 +129,26 @@ class TestListProperty:
     def test_is_base_type(self, list_property_factory):
         assert list_property_factory().is_base_type is False
 
+    @pytest.mark.parametrize("quoted", (True, False))
+    def test_get_base_json_type_string_base_inner(self, list_property_factory, quoted):
+        p = list_property_factory()
+        assert p.get_base_json_type_string(quoted=quoted) == "List[str]"
+
+    @pytest.mark.parametrize("quoted", (True, False))
+    def test_get_base_json_type_string_model_inner(self, list_property_factory, model_property_factory, quoted):
+        m = model_property_factory()
+        p = list_property_factory(inner_property=m)
+        assert p.get_base_json_type_string(quoted=quoted) == "List[Dict[str, Any]]"
+
+    def test_get_lazy_import_base_inner(self, list_property_factory):
+        p = list_property_factory()
+        assert p.get_lazy_imports(prefix="..") == set()
+
+    def test_get_lazy_import_model_inner(self, list_property_factory, model_property_factory):
+        m = model_property_factory()
+        p = list_property_factory(inner_property=m)
+        assert p.get_lazy_imports(prefix="..") == {'from ..models.my_module import MyClass'}
+
     @pytest.mark.parametrize(
         "required, nullable, expected",
         (
@@ -209,6 +229,15 @@ class TestUnionProperty:
     def test_is_base_type(self, union_property_factory):
         assert union_property_factory().is_base_type is False
 
+    def test_get_lazy_import_base_inner(self, union_property_factory):
+        p = union_property_factory()
+        assert p.get_lazy_imports(prefix="..") == set()
+
+    def test_get_lazy_import_model_inner(self, union_property_factory, model_property_factory):
+        m = model_property_factory()
+        p = union_property_factory(inner_properties=[m])
+        assert p.get_lazy_imports(prefix="..") == {'from ..models.my_module import MyClass'}
+
     @pytest.mark.parametrize(
         "nullable,required,no_optional,json,expected",
         [

tests/test_parser/test_properties/test_model_property.py

@@ -54,6 +54,18 @@ def test_get_imports(self, model_property_factory):
             "from typing import cast",
         }
 
+    @pytest.mark.parametrize(
+        "quoted,expected",
+        [
+            (False, "MyClass"),
+            (True, "'MyClass'"),
+        ]
+    )
+    def test_get_type_string_parent(self, model_property_factory, quoted, expected):
+        parent = model_property_factory()
+        m = model_property_factory()
+        assert m.get_type_string(model_parent=parent, quoted=quoted) == expected
+
     def test_get_lazy_imports(self, model_property_factory):
         prop = model_property_factory(required=False, nullable=True)