Skip to content

Commit

Permalink
parser / properties / self LazyReferencePropertyProxy: quote resolved…
Browse files Browse the repository at this point in the history
… type string for typing
  • Loading branch information
Nementon committed Feb 21, 2021
1 parent d3d7bbe commit 905ffd4
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 4 deletions.
3 changes: 2 additions & 1 deletion openapi_python_client/parser/properties/__init__.py
Expand Up @@ -54,7 +54,7 @@ def get_instance_type_string(self) -> str:
def get_type_string(self, no_optional: bool = False) -> str:
resolved = self.resolve()
if resolved:
return resolved.get_type_string(no_optional)
return f"'{resolved.get_type_string(no_optional)}'"
return "LazyReferencePropertyProxy"

def get_imports(self, *, prefix: str) -> Set[str]:
Expand All @@ -72,6 +72,7 @@ def __deepcopy__(self, memo: Any) -> Property:
return copy.deepcopy(resolved, memo)

def __getattr__(self, name: str) -> Any:

if name == "nullable":
return not self._required
elif name == "required":
Expand Down
6 changes: 3 additions & 3 deletions tests/test_parser/test_properties/test_init.py
Expand Up @@ -1560,9 +1560,9 @@ def test_lazy_proxy_reference_resolved():
)

assert foobar
assert lazy_reference_proxy.get_instance_type_string() == foobar.get_instance_type_string()
assert lazy_reference_proxy.get_type_string(no_optional=False) == foobar.get_type_string(no_optional=False)
assert lazy_reference_proxy.get_type_string(no_optional=True) == foobar.get_type_string(no_optional=True)
assert lazy_reference_proxy.get_instance_type_string() == f"'{foobar.get_instance_type_string()}'"
assert lazy_reference_proxy.get_type_string(no_optional=False) == f"'{foobar.get_type_string(no_optional=False)}'"
assert lazy_reference_proxy.get_type_string(no_optional=True) == f"'{foobar.get_type_string(no_optional=True)}'"
assert lazy_reference_proxy.get_imports(prefix="..") == foobar.get_imports(prefix="..")
assert lazy_reference_proxy.name == "childProperty" and foobar.name == "fooBar"
assert lazy_reference_proxy.nullable == foobar.nullable
Expand Down

0 comments on commit 905ffd4

Please sign in to comment.