diff options
Diffstat (limited to 'service/test/unit/bitmask_libraries/test_session.py')
| -rw-r--r-- | service/test/unit/bitmask_libraries/test_session.py | 8 | 
1 files changed, 6 insertions, 2 deletions
| diff --git a/service/test/unit/bitmask_libraries/test_session.py b/service/test/unit/bitmask_libraries/test_session.py index ea73af8c..3aa8c8c7 100644 --- a/service/test/unit/bitmask_libraries/test_session.py +++ b/service/test/unit/bitmask_libraries/test_session.py @@ -18,22 +18,26 @@ from mock import MagicMock  from pixelated.bitmask_libraries.session import LeapSession  from test_abstract_leap import AbstractLeapTest +from twisted.internet import defer  class SessionTest(AbstractLeapTest):      def setUp(self): +        super(SessionTest, self).setUp()          self.mail_fetcher_mock = MagicMock()          self.smtp_mock = MagicMock()      def tearDown(self):          self.mail_fetcher_mock = MagicMock() -    def test_background_jobs_are_started(self): +    @defer.inlineCallbacks +    def test_background_jobs_are_started_during_initial_sync(self):          self.config.start_background_jobs = True          with patch('pixelated.bitmask_libraries.session.reactor.callFromThread', new=_execute_func) as _: -            self._create_session() +            session = self._create_session() +            yield session.initial_sync()              self.mail_fetcher_mock.startService.assert_called_once_with() | 
