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

changes for linter

Browse source
This commit is contained in:
David Kennedy 2024-01-30 16:42:59 -05:00
parent 9da326be16
commit 248dd4adfc
No known key found for this signature in database
GPG key ID: 6528A5386E66B96B
6 changed files with 33 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -135,7 +135,7 @@ class TestConnectionPool(TestCase):
stack.enter_context(patch.object(EPPConnectionPool, "kill_all_connections", do_nothing))
stack.enter_context(patch.object(SocketTransport, "send", self.fake_send))
stack.enter_context(patch.object(SocketTransport, "receive", fake_receive))
with less_console_noise():
with less_console_noise():
# Restart the connection pool
registry.start_connection_pool()
# Pool should be running, and be the right size
@ -211,7 +211,7 @@ class TestConnectionPool(TestCase):
stack.enter_context(patch.object(EPPConnectionPool, "kill_all_connections", do_nothing))
stack.enter_context(patch.object(SocketTransport, "send", self.fake_send))
stack.enter_context(patch.object(SocketTransport, "receive", fake_receive))
with less_console_noise():
with less_console_noise():
# Start the connection pool
registry.start_connection_pool()
# Kill the connection pool
@ -243,7 +243,7 @@ class TestConnectionPool(TestCase):
def test_raises_connection_error(self):
"""A .send is invoked on the pool, but registry connection is lost
right as we send a command."""
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))
@ -260,4 +260,3 @@ class TestConnectionPool(TestCase):
expected = "InfoDomain failed to execute due to a connection error."
result = registry.send(commands.InfoDomain(name="test.gov"), cleaned=True)
self.assertEqual(result, expected)

2
src/epplibwrapper/utility/pool.py
View file

@ -97,7 +97,7 @@ class EPPConnectionPool(ConnectionPool):
# Nothing to do, the pool will generate a new connection later
pass
gevent.sleep(delay)
def _create_socket(self, client, login) -> Socket:
"""Creates and returns a socket instance"""
socket = Socket(client, login)