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

updated test to remove theeNS and stop settings if not ready or DNS needed

Browse source
This commit is contained in:
Alysia Broddrick 2023-10-06 11:48:45 -07:00
parent 8b17d184eb
commit a4057a669f
No known key found for this signature in database
GPG key ID: 03917052CD0F06B7
4 changed files with 44 additions and 38 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -727,17 +727,11 @@ class MockEppLib(TestCase):
],
)
# Bc we have multiple tests utilizing 3 nameservers,
# easier to set a flag for it
threeNS = False
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":
if self.threeNS:
return MagicMock(res_data=[self.infoDomainThreeHosts])
elif self.mockedSendFunction.call_count == 5:
if self.mockedSendFunction.call_count == 5:
return MagicMock(res_data=[self.infoDomainTwoHosts])
else:
return MagicMock(res_data=[self.infoDomainNoHost])
@ -747,8 +741,8 @@ class MockEppLib(TestCase):
return MagicMock(res_data=[self.infoDomainCheckHostIPCombo])
elif getattr(_request, "name", None) == "freeman.gov":
return MagicMock(res_data=[self.InfoDomainWithContacts])
# elif getattr(_request, "name", None) == "failednameserver.gov":
# return MagicMock(res_data=[self.infoDomainUpdateFail])
elif getattr(_request, "name", None) == "threenameserversDomain.gov":
return MagicMock(res_data=[self.infoDomainThreeHosts])
return MagicMock(res_data=[self.mockDataInfoDomain])
def mockSend(self, _request, cleaned):