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

Update availability API to use EPP availability check

Browse source
This commit is contained in:
Erin 2023-10-17 08:37:03 -07:00
parent 1904c7e461
commit 3ed4c0e4fb
No known key found for this signature in database
GPG key ID: 1CAD275313C62460
5 changed files with 108 additions and 25 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -30,6 +30,7 @@ from epplibwrapper import (
info,
RegistryError,
ErrorCode,
responses,
)
logger = logging.getLogger(__name__)
@ -824,7 +825,41 @@ class MockEppLib(TestCase):
raise RegistryError(
code=ErrorCode.OBJECT_ASSOCIATION_PROHIBITS_OPERATION
)
elif isinstance(_request, commands.CheckDomain):
if "gsa.gov" in getattr(_request, "names", None):
return MagicMock(
res_data=[
responses.check.CheckDomainResultData(
name="gsa.gov", avail=True, reason=None
),
]
)
elif "GSA.gov" in getattr(_request, "names", None):
return MagicMock(
res_data=[
responses.check.CheckDomainResultData(
name="GSA.gov", avail=True, reason=None
),
]
)
elif "igorvilleremixed.gov" in getattr(_request, "names", None):
return MagicMock(
res_data=[
responses.check.CheckDomainResultData(
name="igorvilleremixed.gov", avail=False, reason=None
),
]
)
elif "errordomain.gov" in getattr(_request, "names", None):
raise RegistryError("Registry cannot find domain availability.")
else:
return MagicMock(
res_data=[
responses.check.CheckDomainResultData(
name="domainnotfound.gov", avail=False, reason="In Use"
)
],
)
return MagicMock(res_data=[self.mockDataInfoHosts])
def setUp(self):