From 3d3c620ac53de1c407ff2ff222d923310770ed62 Mon Sep 17 00:00:00 2001 From: Tulio Casagrande Date: Fri, 20 Jan 2017 16:51:18 -0200 Subject: Re-raise the same exception to keep context --- service/pixelated/config/leap.py | 2 +- service/test/unit/config/test_leap.py | 5 ++--- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/service/pixelated/config/leap.py b/service/pixelated/config/leap.py index 52a70f3e..2b3a242a 100644 --- a/service/pixelated/config/leap.py +++ b/service/pixelated/config/leap.py @@ -99,7 +99,7 @@ class BootstrapUserServices(object): log.warn('{0}: {1}. Closing session for user: {2}'.format(e.__class__.__name__, e, user_auth.username)) if leap_session: leap_session.close() - raise e + raise @defer.inlineCallbacks def _setup_user_services(self, leap_session): diff --git a/service/test/unit/config/test_leap.py b/service/test/unit/config/test_leap.py index 29f19f2f..3cfa43f2 100644 --- a/service/test/unit/config/test_leap.py +++ b/service/test/unit/config/test_leap.py @@ -14,7 +14,6 @@ # You should have received a copy of the GNU Affero General Public License # along with Pixelated. If not, see . -from leap.soledad.common.errors import InvalidAuthTokenError from mock import MagicMock, patch, Mock from twisted.trial import unittest from twisted.internet import defer @@ -170,7 +169,7 @@ class TestUserBootstrap(unittest.TestCase): @patch('pixelated.config.leap.create_leap_session') @patch('pixelated.config.leap.log.warn') @defer.inlineCallbacks - def test__setup__should_close_leap_sesson_on_error_from__setup_user_services(self, mock_logger_warn, mock_create_leap_session, mock_setup_user_services): + def test__setup__should_close_leap_session_on_error_from__setup_user_services(self, mock_logger_warn, mock_create_leap_session, mock_setup_user_services): leap_session = Mock() leap_session.close = Mock() mock_create_leap_session.return_value = leap_session @@ -184,7 +183,7 @@ class TestUserBootstrap(unittest.TestCase): @patch('pixelated.config.leap.create_leap_session') @patch('pixelated.config.leap.log.warn') @defer.inlineCallbacks - def test__setup__should_close_leap_sesson_on_error_from__add_welcome_email(self, mock_logger_warn, mock_create_leap_session, _, mock_add_welcome_email): + def test__setup__should_close_leap_session_on_error_from__add_welcome_email(self, mock_logger_warn, mock_create_leap_session, _, mock_add_welcome_email): leap_session = Mock() leap_session.close = Mock() mock_create_leap_session.return_value = leap_session -- cgit v1.2.3 From a966e35f5d3eda2cdcd39899e9bce6db6b4da73a Mon Sep 17 00:00:00 2001 From: Tulio Casagrande Date: Sat, 21 Jan 2017 11:48:08 -0200 Subject: Remove old UploadKeyError test I was surprised this test was not working correctly. --- service/pixelated/config/sessions.py | 5 +---- service/test/unit/config/test_sessions.py | 13 ------------- 2 files changed, 1 insertion(+), 17 deletions(-) diff --git a/service/pixelated/config/sessions.py b/service/pixelated/config/sessions.py index b4d69d29..eb233a9f 100644 --- a/service/pixelated/config/sessions.py +++ b/service/pixelated/config/sessions.py @@ -18,9 +18,7 @@ from __future__ import absolute_import import os import errno -import traceback import requests -import sys from twisted.internet import defer, threads, reactor from twisted.logger import Logger @@ -35,10 +33,9 @@ from leap.common.events import ( catalog as events ) -from pixelated.bitmask_libraries.keymanager import Keymanager, UploadKeyError +from pixelated.bitmask_libraries.keymanager import Keymanager from pixelated.adapter.mailstore import LeapMailStore from pixelated.config import leap_config -from pixelated.bitmask_libraries.certs import LeapCertificate from pixelated.bitmask_libraries.smtp import LeapSMTPConfig logger = Logger() diff --git a/service/test/unit/config/test_sessions.py b/service/test/unit/config/test_sessions.py index 06bffc97..b5e4029d 100644 --- a/service/test/unit/config/test_sessions.py +++ b/service/test/unit/config/test_sessions.py @@ -17,10 +17,8 @@ from mock import patch from mock import MagicMock -from mockito import when from twisted.internet import defer from pixelated.config.sessions import LeapSession, SessionCache, LeapSessionFactory, SoledadWrongPassphraseException -from pixelated.bitmask_libraries.keymanager import UploadKeyError from test.unit.bitmask_libraries.test_abstract_leap import AbstractLeapTest from leap.common.events.catalog import KEYMANAGER_FINISHED_KEY_GENERATION from leap.soledad.common.crypto import WrongMacError, UnknownMacMethodError @@ -43,17 +41,6 @@ class SessionTest(AbstractLeapTest): yield session.first_required_sync() mail_fetcher_mock.startService.assert_called_once() - @patch('pixelated.config.sessions.register') - @defer.inlineCallbacks - def test_upload_key_error_closes_the_session(self, _): - when(self.keymanager).generate_openpgp_key().thenRaise(UploadKeyError('Could not upload key')) - session = self._create_session() - session.close = MagicMock() - - with self.assertRaises(UploadKeyError): - yield session.finish_bootstrap() - session.close.assert_called_once() - @patch('pixelated.config.sessions.register') @patch('pixelated.config.sessions.unregister') @patch('pixelated.config.sessions.Account') -- cgit v1.2.3