mirror of
https://github.com/cisagov/manage.get.gov.git
synced 2025-08-05 09:21:54 +02:00
epplibwrapper test_client
This commit is contained in:
David Kennedy
parent
6c8ec6044b
commit
4e00124fdd
1 changed files with 80 additions and 75 deletions
|
|
@ -26,7 +26,8 @@ class TestClient(TestCase):
|
|||
|
||||
def fake_result(self, code, msg):
|
||||
"""Helper function to create a fake Result object"""
|
||||
return Result(code=code, msg=msg, res_data=[], cl_tr_id="cl_tr_id", sv_tr_id="sv_tr_id")
|
||||
with less_console_noise():
|
||||
return Result(code=code, msg=msg, res_data=[], cl_tr_id="cl_tr_id", sv_tr_id="sv_tr_id")
|
||||
|
||||
@patch("epplibwrapper.client.Client")
|
||||
def test_initialize_client_success(self, mock_client):
|
||||
|
|
@ -268,8 +269,9 @@ class TestClient(TestCase):
|
|||
Raises a ConcurrentObjectUseError, which gevent throws when accessing
|
||||
the same thread from two different locations.
|
||||
"""
|
||||
# This error is thrown when two threads are being used concurrently
|
||||
raise ConcurrentObjectUseError("This socket is already used by another greenlet")
|
||||
with less_console_noise():
|
||||
# This error is thrown when two threads are being used concurrently
|
||||
raise ConcurrentObjectUseError("This socket is already used by another greenlet")
|
||||
|
||||
def do_nothing(self, command=None):
|
||||
"""
|
||||
|
|
@ -281,62 +283,65 @@ class TestClient(TestCase):
|
|||
"""
|
||||
Simulates receiving a success response from EPP.
|
||||
"""
|
||||
mock = MagicMock(
|
||||
code=1000,
|
||||
msg="Command completed successfully",
|
||||
res_data=None,
|
||||
cl_tr_id="xkw1uo#2023-10-17T15:29:09.559376",
|
||||
sv_tr_id="5CcH4gxISuGkq8eqvr1UyQ==-35a",
|
||||
extensions=[],
|
||||
msg_q=None,
|
||||
)
|
||||
return mock
|
||||
with less_console_noise():
|
||||
mock = MagicMock(
|
||||
code=1000,
|
||||
msg="Command completed successfully",
|
||||
res_data=None,
|
||||
cl_tr_id="xkw1uo#2023-10-17T15:29:09.559376",
|
||||
sv_tr_id="5CcH4gxISuGkq8eqvr1UyQ==-35a",
|
||||
extensions=[],
|
||||
msg_q=None,
|
||||
)
|
||||
return mock
|
||||
|
||||
def fake_info_domain_received(self, command=None, cleaned=None):
|
||||
"""
|
||||
Simulates receiving a response by reading from a predefined XML file.
|
||||
"""
|
||||
location = Path(__file__).parent / "utility" / "infoDomain.xml"
|
||||
xml = (location).read_bytes()
|
||||
return xml
|
||||
with less_console_noise():
|
||||
location = Path(__file__).parent / "utility" / "infoDomain.xml"
|
||||
xml = (location).read_bytes()
|
||||
return xml
|
||||
|
||||
def get_fake_epp_result(self):
|
||||
"""Mimics a return from EPP by returning a dictionary in the same format"""
|
||||
result = {
|
||||
"cl_tr_id": None,
|
||||
"code": 1000,
|
||||
"extensions": [],
|
||||
"msg": "Command completed successfully",
|
||||
"msg_q": None,
|
||||
"res_data": [
|
||||
info.InfoDomainResultData(
|
||||
roid="DF1340360-GOV",
|
||||
statuses=[
|
||||
common.Status(
|
||||
state="serverTransferProhibited",
|
||||
description=None,
|
||||
lang="en",
|
||||
),
|
||||
common.Status(state="inactive", description=None, lang="en"),
|
||||
],
|
||||
cl_id="gov2023-ote",
|
||||
cr_id="gov2023-ote",
|
||||
cr_date=datetime.datetime(2023, 8, 15, 23, 56, 36, tzinfo=tzlocal()),
|
||||
up_id="gov2023-ote",
|
||||
up_date=datetime.datetime(2023, 8, 17, 2, 3, 19, tzinfo=tzlocal()),
|
||||
tr_date=None,
|
||||
name="test3.gov",
|
||||
registrant="TuaWnx9hnm84GCSU",
|
||||
admins=[],
|
||||
nsset=None,
|
||||
keyset=None,
|
||||
ex_date=datetime.date(2024, 8, 15),
|
||||
auth_info=info.DomainAuthInfo(pw="2fooBAR123fooBaz"),
|
||||
)
|
||||
],
|
||||
"sv_tr_id": "wRRNVhKhQW2m6wsUHbo/lA==-29a",
|
||||
}
|
||||
return result
|
||||
with less_console_noise():
|
||||
result = {
|
||||
"cl_tr_id": None,
|
||||
"code": 1000,
|
||||
"extensions": [],
|
||||
"msg": "Command completed successfully",
|
||||
"msg_q": None,
|
||||
"res_data": [
|
||||
info.InfoDomainResultData(
|
||||
roid="DF1340360-GOV",
|
||||
statuses=[
|
||||
common.Status(
|
||||
state="serverTransferProhibited",
|
||||
description=None,
|
||||
lang="en",
|
||||
),
|
||||
common.Status(state="inactive", description=None, lang="en"),
|
||||
],
|
||||
cl_id="gov2023-ote",
|
||||
cr_id="gov2023-ote",
|
||||
cr_date=datetime.datetime(2023, 8, 15, 23, 56, 36, tzinfo=tzlocal()),
|
||||
up_id="gov2023-ote",
|
||||
up_date=datetime.datetime(2023, 8, 17, 2, 3, 19, tzinfo=tzlocal()),
|
||||
tr_date=None,
|
||||
name="test3.gov",
|
||||
registrant="TuaWnx9hnm84GCSU",
|
||||
admins=[],
|
||||
nsset=None,
|
||||
keyset=None,
|
||||
ex_date=datetime.date(2024, 8, 15),
|
||||
auth_info=info.DomainAuthInfo(pw="2fooBAR123fooBaz"),
|
||||
)
|
||||
],
|
||||
"sv_tr_id": "wRRNVhKhQW2m6wsUHbo/lA==-29a",
|
||||
}
|
||||
return result
|
||||
|
||||
def test_send_command_close_failure_recovers(self):
|
||||
"""
|
||||
|
|
@ -350,28 +355,28 @@ class TestClient(TestCase):
|
|||
- Subsequently, the client re-initializes the connection.
|
||||
- A retry of the command execution post-reinitialization succeeds.
|
||||
"""
|
||||
with less_console_noise():
|
||||
expected_result = self.get_fake_epp_result()
|
||||
wrapper = None
|
||||
# Trigger a retry
|
||||
# Do nothing on connect, as we aren't testing it and want to connect while
|
||||
# mimicking the rest of the client as closely as possible (which is not entirely possible with MagicMock)
|
||||
with patch.object(EPPLibWrapper, "_connect", self.do_nothing):
|
||||
with patch.object(SocketTransport, "send", self.fake_failure_send_concurrent_threads):
|
||||
wrapper = EPPLibWrapper()
|
||||
tested_command = commands.InfoDomain(name="test.gov")
|
||||
try:
|
||||
wrapper.send(tested_command, cleaned=True)
|
||||
except RegistryError as err:
|
||||
expected_error = "InfoDomain failed to execute due to an unknown error."
|
||||
self.assertEqual(err.args[0], expected_error)
|
||||
else:
|
||||
self.fail("Registry error was not thrown")
|
||||
|
||||
expected_result = self.get_fake_epp_result()
|
||||
wrapper = None
|
||||
# Trigger a retry
|
||||
# Do nothing on connect, as we aren't testing it and want to connect while
|
||||
# mimicking the rest of the client as closely as possible (which is not entirely possible with MagicMock)
|
||||
with patch.object(EPPLibWrapper, "_connect", self.do_nothing):
|
||||
with patch.object(SocketTransport, "send", self.fake_failure_send_concurrent_threads):
|
||||
wrapper = EPPLibWrapper()
|
||||
tested_command = commands.InfoDomain(name="test.gov")
|
||||
try:
|
||||
wrapper.send(tested_command, cleaned=True)
|
||||
except RegistryError as err:
|
||||
expected_error = "InfoDomain failed to execute due to an unknown error."
|
||||
self.assertEqual(err.args[0], expected_error)
|
||||
else:
|
||||
self.fail("Registry error was not thrown")
|
||||
|
||||
# After a retry, try sending again to see if the connection recovers
|
||||
with patch.object(EPPLibWrapper, "_connect", self.do_nothing):
|
||||
with patch.object(SocketTransport, "send", self.fake_success_send), patch.object(
|
||||
SocketTransport, "receive", self.fake_info_domain_received
|
||||
):
|
||||
result = wrapper.send(tested_command, cleaned=True)
|
||||
self.assertEqual(expected_result, result.__dict__)
|
||||
# After a retry, try sending again to see if the connection recovers
|
||||
with patch.object(EPPLibWrapper, "_connect", self.do_nothing):
|
||||
with patch.object(SocketTransport, "send", self.fake_success_send), patch.object(
|
||||
SocketTransport, "receive", self.fake_info_domain_received
|
||||
):
|
||||
result = wrapper.send(tested_command, cleaned=True)
|
||||
self.assertEqual(expected_result, result.__dict__)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue