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

Linter + fix test

Browse source
This commit is contained in:
zandercymatics 2023-10-19 09:40:40 -06:00
parent b663ac7b71
commit e91ee079d0
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
1 changed files with 9 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -4,6 +4,7 @@ from unittest.mock import MagicMock, patch
from dateutil.tz import tzlocal from dateutil.tz import tzlocal
from django.test import TestCase from django.test import TestCase
from epplibwrapper.client import EPPLibWrapper from epplibwrapper.client import EPPLibWrapper
from epplibwrapper.errors import RegistryError
from epplibwrapper.socket import Socket from epplibwrapper.socket import Socket
from epplibwrapper.utility.pool import EPPConnectionPool from epplibwrapper.utility.pool import EPPConnectionPool
from registrar.models.domain import registry from registrar.models.domain import registry
@ -38,13 +39,15 @@ class TestConnectionPool(TestCase):
} }
def fake_socket(self, login, client): def fake_socket(self, login, client):
# Linter reasons
pw = "none"
# Create a fake client object # Create a fake client object
fake_client = Client( fake_client = Client(
SocketTransport( SocketTransport(
"none", "none",
cert_file="path/to/cert_file", cert_file="path/to/cert_file",
key_file="path/to/key_file", key_file="path/to/key_file",
password="none", password=pw,
) )
) )
@ -149,7 +152,7 @@ class TestConnectionPool(TestCase):
self.assertEqual(len(registry._pool.conn), self.pool_options["size"]) self.assertEqual(len(registry._pool.conn), self.pool_options["size"])
@patch.object(EPPLibWrapper, "_test_registry_connection_success", patch_success) @patch.object(EPPLibWrapper, "_test_registry_connection_success", patch_success)
def test_raises_transport_error(self): def test_raises_connection_error(self):
"""A .send is invoked on the pool, but registry connection is lost """A .send is invoked on the pool, but registry connection is lost
right as we send a command.""" right as we send a command."""
@ -176,5 +179,7 @@ class TestConnectionPool(TestCase):
self.assertEqual(registry.pool_status.pool_running, True) self.assertEqual(registry.pool_status.pool_running, True)
# Try to send a command out - should fail # Try to send a command out - should fail
with self.assertRaises(TransportError): with self.assertRaises(RegistryError):
registry.send(commands.InfoDomain(name="test.gov"), cleaned=True) expected_message = "InfoDomain failed to execute due to a connection error."
result = registry.send(commands.InfoDomain(name="test.gov"), cleaned=True)
self.assertEqual(result, expected_message)