summaryrefslogtreecommitdiff
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
commit5409d64e20e353f9f0acb77ad2d89b80a8685ed0 (patch)
tree46156384e14b8b8c6a82bdfd3d0543102c5c88f7
parent5ebc52f361153fe2f9b19d2908f09ba5df5433d8 (diff)
[bug] Add KeyNotFound import
Two tests where failing, because - Related: #7987
-rw-r--r--keymanager/src/leap/keymanager/tests/test_keymanager.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/keymanager/src/leap/keymanager/tests/test_keymanager.py b/keymanager/src/leap/keymanager/tests/test_keymanager.py
index 867a13d..9deaff1 100644
--- a/keymanager/src/leap/keymanager/tests/test_keymanager.py
+++ b/keymanager/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):