mirror of
https://github.com/cisagov/manage.get.gov.git
synced 2025-07-12 05:58:18 +02:00
Update test_pool.py
This commit is contained in:
parent
a744b9407f
commit
b8dfe0b8f0
1 changed files with 6 additions and 4 deletions
|
@ -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)
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue