summaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
authorZara Gebru <zgebru@thoughtworks.com>2016-06-10 14:28:17 +0200
committerZara Gebru <zgebru@thoughtworks.com>2016-06-10 14:28:17 +0200
commit2d69975880214e59d493b5d30655295f77a5cb0c (patch)
treef37c7016e03171d55689ec40f3d2f6ba77cc2026 /src
parent06d0c7e8b4206aadb71281f607107ac7e0868945 (diff)
[bug] Add KeyNotFound import
Two tests where failing, because - Related: #7987
Diffstat (limited to 'src')
-rw-r--r--src/leap/keymanager/tests/test_keymanager.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/src/leap/keymanager/tests/test_keymanager.py b/src/leap/keymanager/tests/test_keymanager.py
index 867a13d..9deaff1 100644
--- a/src/leap/keymanager/tests/test_keymanager.py
+++ b/src/leap/keymanager/tests/test_keymanager.py
@@ -259,7 +259,7 @@ class KeyManagerKeyManagementTestCase(KeyManagerWithSoledadTestCase):
check_404_callback = used_kwargs['callback']
fake_response = Mock()
fake_response.code = NOT_FOUND
- with self.assertRaisesRegexp(KeyNotFound, '404: %s key not found.' % INVALID_MAIL_ADDRESS):
+ with self.assertRaisesRegexp(errors.KeyNotFound, '404: %s key not found.' % INVALID_MAIL_ADDRESS):
check_404_callback(fake_response)
d.addCallback(check_key_not_found_is_raised_if_404)
@@ -270,7 +270,7 @@ class KeyManagerKeyManagementTestCase(KeyManagerWithSoledadTestCase):
Test if key search requests throws KeyNotFound, the same error is raised.
"""
km = self._key_manager(url=NICKSERVER_URI)
- key_not_found_exception = KeyNotFound('some message')
+ key_not_found_exception = errors.KeyNotFound('some message')
km._async_client_pinned.request = Mock(side_effect=key_not_found_exception)
def assert_key_not_found_raised(error):