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

Remove extraneous print statements and rename extendedValues to threeNS

Browse source
This commit is contained in:
Rebecca Hsieh 2023-10-04 09:14:09 -07:00
parent 5fcae0cb80
commit fbad2abd9d
No known key found for this signature in database
GPG key ID: 644527A2F375A379
3 changed files with 9 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -644,14 +644,15 @@ class MockEppLib(TestCase):
],
)
# TODO-848: Fix naming
extendedValues = False
# 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.extendedValues:
if self.threeNS:
return MagicMock(res_data=[self.infoDomainThreeHosts])
elif self.mockedSendFunction.call_count == 5:
return MagicMock(res_data=[self.infoDomainTwoHosts])