zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-07-25 20:18:38 +02:00
Code Issues Projects Releases Packages Wiki Activity

fixed failing tests

Browse source
This commit is contained in:
Alysia Broddrick 2023-10-02 09:21:03 -07:00
parent 18c3e0e96f
commit ed3d8ccf72
No known key found for this signature in database
GPG key ID: 03917052CD0F06B7
2 changed files with 4 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -638,8 +638,7 @@ class MockEppLib(TestCase):
# TODO-848: Fix naming
extendedValues = False
# TODO-848: Rename later - was getting complex err
def _getattrshelper(self, _request):
def _getattrInfoDomain(self, _request):
if getattr(_request, "name", None) == "security.gov":
return MagicMock(res_data=[self.infoDomainNoContact])
elif getattr(_request, "name", None) == "my-nameserver.gov":
@ -653,7 +652,7 @@ class MockEppLib(TestCase):
return MagicMock(res_data=[self.infoDomainHasIP])
# elif getattr(_request, "name", None) == "failednameserver.gov":
# return MagicMock(res_data=[self.infoDomainUpdateFail])
# return MagicMock(res_data=[self.mockDataInfoDomain])
return MagicMock(res_data=[self.mockDataInfoDomain])
def mockSend(self, _request, cleaned):
"""Mocks the registry.send function used inside of domain.py
@ -661,8 +660,7 @@ class MockEppLib(TestCase):
returns objects that simulate what would be in a epp response
but only relevant pieces for tests"""
if isinstance(_request, commands.InfoDomain):
# TODO-848: Fix name here
return self._getattrshelper(_request)
return self._getattrInfoDomain(_request)
elif isinstance(_request, commands.InfoContact):
return MagicMock(res_data=[self.mockDataInfoContact])
elif (