Update test_pool.py

This commit is contained in:
zandercymatics 2023-10-19 10:25:51 -06:00
parent a744b9407f
commit b8dfe0b8f0
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7

View file

@ -1,7 +1,7 @@
import datetime import datetime
from pathlib import Path from pathlib import Path
from unittest.mock import MagicMock, patch from unittest.mock import MagicMock, patch
from dateutil.tz import tzlocal # type: ignore from dateutil.tz import tzlocal # type: ignore
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.errors import RegistryError
@ -182,6 +182,8 @@ class TestConnectionPool(TestCase):
# Try to send a command out - should fail # Try to send a command out - should fail
with self.assertRaises(RegistryError): with self.assertRaises(RegistryError):
expected_message = "InfoDomain failed to execute due to a connection error." expected = "InfoDomain failed to execute due to a connection error."
result = registry.send(commands.InfoDomain(name="test.gov"), cleaned=True) result = registry.send(
self.assertEqual(result, expected_message) commands.InfoDomain(name="test.gov"), cleaned=True
)
self.assertEqual(result, expected)