def test_init(self) -> None:
service_name_mock = MagicMock()
service_name_mock.name = "service_name"
result = InternalImportRecord(service_name_mock, "name", "alias")
assert result.source.render() == ".service_name"
assert result.name == "name"
assert result.alias == "alias"
def get_import_record(self) -> ImportRecord:
"""
Get import record required for using type annotation.
"""
if self.inline:
return self.get_typing_import_record()
return InternalImportRecord(ServiceModuleName.literals, name=self.name)
def _get_client_import(self, service_name: ServiceName) -> ExternalImport:
client_import = ExternalImport(
source=ImportString.parent() +
ImportString(ServiceModuleName.client.value),
name=Client.get_class_name(service_name),
)
client_import.import_record = InternalImportRecord(
ServiceModuleName.client, client_import.name)
return client_import
def get_import_record(self) -> ImportRecord:
"""
Get import record required for using type annotation.
"""
if self.service_name is not None:
return ImportRecord(
source=ImportString(self.service_name.module_name,
self.module_name.name),
alias=self.scope,
)
return InternalImportRecord(self.module_name, alias=self.scope)
def get_import_record(self) -> ImportRecord:
"""
Get import record required for using type annotation.
"""
if self.inline:
return ImportRecord(
ImportString("typing"),
"Literal",
min_version=(3, 8),
fallback=ImportRecord(ImportString("typing_extensions"), "Literal"),
)
return InternalImportRecord(ServiceModuleName.literals, name=self.name)
def test_get_external(self, ImportRecordMock: MagicMock,
ImportStringMock: MagicMock) -> None:
service_name_mock = MagicMock()
service_name_mock.name = "service_name"
ImportRecordMock.return_value = "ImportRecord"
ImportStringMock().__add__.return_value = "module_name.service_name"
assert (InternalImportRecord(
service_name_mock, "name",
"alias").get_external("module_name") == "ImportRecord")
ImportStringMock.assert_called_with("module_name")
ImportRecordMock.assert_called_with(source="module_name.service_name",
name="name",
alias="alias")
def get_import_record(self) -> ImportRecord:
"""
Get import record required for using type annotation.
"""
return InternalImportRecord(ServiceModuleName.type_defs, name=self.name)
def test_get_external(self, ImportRecordMock: MagicMock,
ImportStringMock: MagicMock) -> None:
service_name_mock = MagicMock()
service_name_mock.name = "service_name"
result = InternalImportRecord(service_name_mock, "name", "alias")
assert result.get_external("module_name") is result
