mirror of
https://github.com/cisagov/manage.get.gov.git
synced 2025-07-26 04:28:39 +02:00
added unit test
This commit is contained in:
parent
d31af94a15
commit
d498ba3818
2 changed files with 52 additions and 1 deletions
|
@ -184,7 +184,7 @@ class EPPLibWrapper:
|
|||
message = f"{cmd_type} failed and will be retried"
|
||||
info_message = f"{message} Error: {err}"
|
||||
if err.response:
|
||||
info_message = f"{info_message}. cltrid is {err.response.cl_tr_id} svtrid is {err.response.sv_tr_id}"
|
||||
info_message = f"{info_message}| cltrid is {err.response.cl_tr_id} svtrid is {err.response.sv_tr_id}"
|
||||
logger.info(f"{info_message}")
|
||||
return self._retry(command)
|
||||
else:
|
||||
|
|
|
@ -264,6 +264,57 @@ class TestClient(TestCase):
|
|||
# send() is called 5 times: send(login), send(command) fail, send(logout), send(login), send(command)
|
||||
self.assertEquals(mock_send.call_count, 5)
|
||||
|
||||
@less_console_noise_decorator
|
||||
@patch("epplibwrapper.client.Client")
|
||||
@patch("epplibwrapper.client.logger")
|
||||
def test_send_command_2002_failure_prompts_successful_retry(self, mock_logger, mock_client):
|
||||
"""Test when the send("InfoDomainCommand) call fails with a 2002, prompting a retry
|
||||
and the subsequent send("InfoDomainCommand) call succeeds
|
||||
Flow:
|
||||
Initialization succeeds
|
||||
Send command fails (with 2002 code) prompting retry
|
||||
Client closes and re-initializes, and command succeeds"""
|
||||
# Mock the Client instance and its methods
|
||||
# connect() and close() should succeed throughout
|
||||
mock_connect = MagicMock()
|
||||
mock_close = MagicMock()
|
||||
# create success and failure result messages
|
||||
send_command_success_result = self.fake_result(1000, "Command completed successfully")
|
||||
send_command_failure_result = self.fake_result(2002, "Registrar is not logging in.")
|
||||
# side_effect for send call, initial send(login) succeeds during initialization, next send(command)
|
||||
# fails, subsequent sends (logout, login, command) all succeed
|
||||
send_call_count = 0
|
||||
|
||||
# Create a mock command
|
||||
mock_command = MagicMock()
|
||||
mock_command.__class__.__name__ = "InfoDomainCommand"
|
||||
|
||||
def side_effect(*args, **kwargs):
|
||||
nonlocal send_call_count
|
||||
send_call_count += 1
|
||||
if send_call_count == 2:
|
||||
return send_command_failure_result
|
||||
else:
|
||||
return send_command_success_result
|
||||
|
||||
mock_send = MagicMock(side_effect=side_effect)
|
||||
mock_client.return_value.connect = mock_connect
|
||||
mock_client.return_value.close = mock_close
|
||||
mock_client.return_value.send = mock_send
|
||||
# Create EPPLibWrapper instance and initialize client
|
||||
wrapper = EPPLibWrapper()
|
||||
wrapper.send(mock_command, cleaned=True)
|
||||
# connect() is called twice, once during initialization of app, once during retry
|
||||
self.assertEquals(mock_connect.call_count, 2)
|
||||
# close() is called once, during retry
|
||||
mock_close.assert_called_once()
|
||||
# send() is called 5 times: send(login), send(command) fail, send(logout), send(login), send(command)
|
||||
self.assertEquals(mock_send.call_count, 5)
|
||||
# Assertion proper logging; note that the
|
||||
mock_logger.info.assert_called_once_with(
|
||||
"InfoDomainCommand failed and will be retried Error: Registrar is not logging in.| cltrid is cl_tr_id svtrid is sv_tr_id"
|
||||
)
|
||||
|
||||
@less_console_noise_decorator
|
||||
def fake_failure_send_concurrent_threads(self, command=None, cleaned=None):
|
||||
"""
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue