diff options
author | Zara Gebru <zgebru@thoughtworks.com> | 2016-06-10 14:28:17 +0200 |
---|---|---|
committer | Zara Gebru <zgebru@thoughtworks.com> | 2016-06-10 14:28:17 +0200 |
commit | 5409d64e20e353f9f0acb77ad2d89b80a8685ed0 (patch) | |
tree | 46156384e14b8b8c6a82bdfd3d0543102c5c88f7 | |
parent | 5ebc52f361153fe2f9b19d2908f09ba5df5433d8 (diff) |
[bug] Add KeyNotFound import
Two tests where failing, because
- Related: #7987
-rw-r--r-- | keymanager/src/leap/keymanager/tests/test_keymanager.py | 4 |
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): |