From 44f560887d48c1b0980613760453e9f8740a6fd6 Mon Sep 17 00:00:00 2001 From: NavaL Date: Mon, 8 Feb 2016 10:43:59 +0000 Subject: mocking register/unregister key generation event, on close - fixing build Issue #586 --- service/test/unit/bitmask_libraries/test_session.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) (limited to 'service/test/unit/bitmask_libraries') diff --git a/service/test/unit/bitmask_libraries/test_session.py b/service/test/unit/bitmask_libraries/test_session.py index bb1f55ca..9654e624 100644 --- a/service/test/unit/bitmask_libraries/test_session.py +++ b/service/test/unit/bitmask_libraries/test_session.py @@ -69,15 +69,19 @@ class SessionTest(AbstractLeapTest): unregister_mock.assert_called_once_with(KEYMANAGER_FINISHED_KEY_GENERATION, uid=email) - def test_close_stops_soledad(self): + @patch('pixelated.bitmask_libraries.session.register') + def test_close_stops_soledad(self, _): email = 'someone@somedomain.tld' self.provider.address_for.return_value = email session = self._create_session() - session.close() + with patch('pixelated.bitmask_libraries.session.unregister') as unregister_mock: + session.close() + self.soledad_session.close.assert_called_once_with() - def test_close_removes_session_from_cache(self): + @patch('pixelated.bitmask_libraries.session.register') + def test_close_removes_session_from_cache(self, _): email = 'someone@somedomain.tld' self.provider.address_for.return_value = email session = self._create_session() @@ -87,7 +91,8 @@ class SessionTest(AbstractLeapTest): self.assertEqual(session, SessionCache.lookup_session(key)) - session.close() + with patch('pixelated.bitmask_libraries.session.unregister') as unregister_mock: + session.close() self.assertIsNone(SessionCache.lookup_session(key)) -- cgit v1.2.3