def test_install_timers(self): x = Polaroid(self.state, app=self.app) x.timer = timer x.__exit__() x.__enter__() self.assertIn(x.capture, MockTimer.installed) self.assertIn(x.cleanup, MockTimer.installed) self.assertTrue(x._tref.active) self.assertTrue(x._ctref.active) x.__exit__() self.assertFalse(x._tref.active) self.assertFalse(x._ctref.active) self.assertTrue(x._tref.called) self.assertFalse(x._ctref.called)
def test_install_timers(self): x = Polaroid(self.state, app=self.app) x.timer = timer x.__exit__() x.__enter__() assert x.capture in MockTimer.installed assert x.cleanup in MockTimer.installed x._tref.cancel.assert_not_called() x._ctref.cancel.assert_not_called() x.__exit__() x._tref.cancel.assert_called() x._ctref.cancel.assert_called() x._tref.assert_called() x._ctref.assert_not_called()