Mock register events to not complain about zmq certs

Related with: #815

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()