diff options
| -rw-r--r-- | service/test/unit/config/test_sessions.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/service/test/unit/config/test_sessions.py b/service/test/unit/config/test_sessions.py index 8ae66015..ad4dd590 100644 --- a/service/test/unit/config/test_sessions.py +++ b/service/test/unit/config/test_sessions.py @@ -14,12 +14,10 @@ # You should have received a copy of the GNU Affero General Public License # along with Pixelated. If not, see <http://www.gnu.org/licenses/>. -import os from mock import patch from mock import MagicMock from mockito import when from twisted.internet import defer -from twisted.trial import unittest from pixelated.config.sessions import LeapSession, SessionCache from pixelated.bitmask_libraries.keymanager import UploadKeyError @@ -44,8 +42,9 @@ 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): + 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() |