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

PR Changes

Browse source
This commit is contained in:
zandercymatics 2023-10-02 11:38:41 -06:00
parent a84bbb5d3a
commit 1990f91e6a
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
5 changed files with 185 additions and 131 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -683,9 +683,9 @@ class MockEppLib(TestCase):
else:
return MagicMock(res_data=[self.mockDataInfoDomain])
elif isinstance(_request, commands.InfoContact):
# Default contact return
mocked_result = self.mockDataInfoContact
# For testing contact types...
mocked_result: info.InfoContactResultData
# For testing contact types
match getattr(_request, "id", None):
case "securityContact":
mocked_result = self.mockSecurityContact
@ -695,7 +695,8 @@ class MockEppLib(TestCase):
mocked_result = self.mockAdministrativeContact
case "regContact":
mocked_result = self.mockRegistrantContact
case "123":
case _:
# Default contact return
mocked_result = self.mockDataInfoContact
return MagicMock(res_data=[mocked_result])