zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-07-25 12:08:40 +02:00
Code Issues Projects Releases Packages Wiki Activity

Fix lint errors

Browse source
This commit is contained in:
Erin 2023-12-06 14:37:21 -08:00
parent d55277cc65
commit 2db344f718
No known key found for this signature in database
GPG key ID: 1CAD275313C62460
1 changed files with 14 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

15
src/epplibwrapper/tests/test_pool.py
View file

@ -245,4 +245,17 @@ class TestConnectionPool(TestCase):
with self.assertRaises(RegistryError):
expected = "InfoDomain failed to execute due to a connection error."
result = registry.send(commands.InfoDomain(name="test.gov"), cleaned=True)
self.assertEqual(result, expected)
self.assertEqual(result, expected)
@patch.object(EPPLibWrapper, "_test_registry_connection_success", patch_success)
def test_retries_on_transport_error(self):
"""A .send is invoked on the pool, but transport error occurs and EPP
retries connection."""
with ExitStack() as stack:
stack.enter_context(patch.object(EPPConnectionPool, "_create_socket", self.fake_socket))
stack.enter_context(patch.object(Socket, "connect", self.fake_client))
# Pool should be running
self.assertEqual(registry.pool_status.connection_success, True)
self.assertEqual(registry.pool_status.pool_running, True)