zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-05-15 17:17:02 +02:00
Code Issues Projects Releases Packages Wiki Activity

refactor epp mockInfoDomainCommands function

Browse source
This commit is contained in:
Rachid Mrad 2023-10-13 15:43:24 -04:00
parent c6abca99ca
commit f94363a836
No known key found for this signature in database
GPG key ID: EF38E4CEC4A8F3CF
1 changed files with 39 additions and 35 deletions
Download patch file Download diff file Expand all files Collapse all files

74
src/registrar/tests/common.py
View file

@ -838,41 +838,45 @@ class MockEppLib(TestCase):
) )
def mockInfoDomainCommands(self, _request, cleaned): def mockInfoDomainCommands(self, _request, cleaned):
if getattr(_request, "name", None) == "security.gov": request_name = getattr(_request, "name", None)
return MagicMock(res_data=[self.infoDomainNoContact])
elif getattr(_request, "name", None) == "dnssec-dsdata.gov": # Define a dictionary to map request names to data and extension values
return MagicMock( request_mappings = {
res_data=[self.mockDataInfoDomain], "security.gov": (self.infoDomainNoContact, None),
extensions=[self.dnssecExtensionWithDsData], "dnssec-dsdata.gov": (
) self.mockDataInfoDomain,
elif getattr(_request, "name", None) == "dnssec-multdsdata.gov": self.dnssecExtensionWithDsData,
return MagicMock( ),
res_data=[self.mockDataInfoDomain], "dnssec-multdsdata.gov": (
extensions=[self.dnssecExtensionWithMultDsData], self.mockDataInfoDomain,
) self.dnssecExtensionWithMultDsData,
elif getattr(_request, "name", None) == "dnssec-keydata.gov": ),
return MagicMock( "dnssec-keydata.gov": (
res_data=[self.mockDataInfoDomain], self.mockDataInfoDomain,
extensions=[self.dnssecExtensionWithKeyData], self.dnssecExtensionWithKeyData,
) ),
elif getattr(_request, "name", None) == "dnssec-none.gov": "dnssec-none.gov": (self.mockDataInfoDomain, None),
# this case is not necessary, but helps improve readability "my-nameserver.gov": (
return MagicMock(res_data=[self.mockDataInfoDomain]) self.infoDomainTwoHosts
elif getattr(_request, "name", None) == "my-nameserver.gov": if self.mockedSendFunction.call_count == 5
if self.mockedSendFunction.call_count == 5: else self.infoDomainNoHost,
return MagicMock(res_data=[self.infoDomainTwoHosts]) None,
else: ),
return MagicMock(res_data=[self.infoDomainNoHost]) "nameserverwithip.gov": (self.infoDomainHasIP, None),
elif getattr(_request, "name", None) == "nameserverwithip.gov": "namerserversubdomain.gov": (self.infoDomainCheckHostIPCombo, None),
return MagicMock(res_data=[self.infoDomainHasIP]) "freeman.gov": (self.InfoDomainWithContacts, None),
elif getattr(_request, "name", None) == "namerserversubdomain.gov": "threenameserversDomain.gov": (self.infoDomainThreeHosts, None),
return MagicMock(res_data=[self.infoDomainCheckHostIPCombo]) }
elif getattr(_request, "name", None) == "freeman.gov":
return MagicMock(res_data=[self.InfoDomainWithContacts]) # Retrieve the corresponding values from the dictionary
elif getattr(_request, "name", None) == "threenameserversDomain.gov": res_data, extensions = request_mappings.get(
return MagicMock(res_data=[self.infoDomainThreeHosts]) request_name, (self.mockDataInfoDomain, None)
else: )
return MagicMock(res_data=[self.mockDataInfoDomain])
return MagicMock(
res_data=[res_data],
extensions=[extensions] if extensions is not None else [],
)
def mockInfoContactCommands(self, _request, cleaned): def mockInfoContactCommands(self, _request, cleaned):
mocked_result: info.InfoContactResultData mocked_result: info.InfoContactResultData