From 2d69975880214e59d493b5d30655295f77a5cb0c Mon Sep 17 00:00:00 2001 From: Zara Gebru Date: Fri, 10 Jun 2016 14:28:17 +0200 Subject: [bug] Add KeyNotFound import Two tests where failing, because - Related: #7987 --- src/leap/keymanager/tests/test_keymanager.py | 4 ++-- 1 file 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): -- cgit v1.2.3