def test_add_non_deferred(self):
w = DeferWaiter()
w.add(2)
self.assertFalse(w.has_waited())
d = w.wait()
self.assertTrue(d.called)
def test_waiter_waits_for_action_timer_starts(self, name,
timer_after_action):
w = DeferWaiter()
times = []
@defer.inlineCallbacks
def action():
times.append(round(self.reactor.seconds(), 1))
yield asyncSleep(0.5, reactor=self.reactor)
h = RepeatedActionHandler(
self.reactor,
w,
1,
action,
start_timer_after_action_completes=timer_after_action)
h.start()
self.reactor.pump([0.1] * 12)
self.assertEqual(times, [1.1])
d = w.wait()
self.assertFalse(d.called)
h.stop()
self.assertFalse(d.called)
self.reactor.pump([0.1] * 5) # action started on 1.1, will end at 1.6
self.assertTrue(d.called)
yield d
def test_add_deferred_called(self):
w = DeferWaiter()
w.add(defer.succeed(None))
self.assertFalse(w.has_waited())
d = w.wait()
self.assertTrue(d.called)
def test_runs_action_after_exception_with_timer(self, name,
timer_after_action):
w = DeferWaiter()
times = []
def action():
times.append(round(self.reactor.seconds(), 1))
if len(times) == 2:
raise TestException()
h = RepeatedActionHandler(
self.reactor,
w,
1,
action,
start_timer_after_action_completes=timer_after_action)
h.start()
self.reactor.pump([0.1] * 35)
self.assertEqual(times, [1.1, 2.1, 3.1])
h.stop()
d = w.wait()
self.assertTrue(d.called)
self.flushLoggedErrors(TestException)
yield d
def test_runs_action_with_timer_delay(self, name, timer_after_action,
expected_times):
w = DeferWaiter()
times = []
@defer.inlineCallbacks
def action():
times.append(round(self.reactor.seconds(), 1))
yield asyncSleep(0.5, reactor=self.reactor)
h = RepeatedActionHandler(
self.reactor,
w,
1,
action,
start_timer_after_action_completes=timer_after_action)
h.start()
self.reactor.pump([0.1] * 47)
self.assertEqual(times, expected_times)
h.stop()
d = w.wait()
self.assertTrue(d.called)
yield d
def test_add_deferred_not_called_and_call_later(self):
w = DeferWaiter()
d1 = defer.Deferred()
w.add(d1)
d = w.wait()
self.assertFalse(d.called)
d1.callback(None)
self.assertTrue(d.called)
def test_cancel_not_called(self):
w = DeferWaiter()
d1 = defer.Deferred()
w.add(d1)
w.cancel()
d = w.wait()
self.assertTrue(d.called)
with self.assertRaises(defer.CancelledError):
yield d1
def test_passes_result(self):
w = DeferWaiter()
d1 = defer.Deferred()
w.add(d1)
d1.callback(123)
res = yield d1
self.assertEqual(res, 123)
d = w.wait()
self.assertTrue(d.called)
def test_cancel_not_called(self):
w = DeferWaiter()
d1 = defer.Deferred()
w.add(d1)
self.assertTrue(w.has_waited())
w.cancel()
self.assertFalse(w.has_waited())
d = w.wait()
self.assertTrue(d.called)
with self.assertRaises(defer.CancelledError):
yield d1
self.flushLoggedErrors(defer.CancelledError)
def test_does_not_add_action_on_start(self):
w = DeferWaiter()
times = []
def action():
times.append(self.reactor.seconds())
h = RepeatedActionHandler(self.reactor, w, 1, action)
self.reactor.advance(2)
h.stop()
self.assertEqual(len(times), 0)
d = w.wait()
self.assertTrue(d.called)
yield d
def test_cancel_called(self):
w = DeferWaiter()
d1_waited = defer.Deferred()
d1 = defer.succeed(None)
d1.addCallback(lambda _: d1_waited)
w.add(d1)
w.cancel()
d = w.wait()
self.assertTrue(d.called)
self.assertTrue(d1.called)
self.assertTrue(d1_waited.called)
with self.assertRaises(defer.CancelledError):
yield d1
def test_can_update_interval(self):
w = DeferWaiter()
times = []
def action():
times.append(round(self.reactor.seconds(), 1))
h = RepeatedActionHandler(self.reactor, w, 1, action)
h.start()
self.reactor.pump([0.1] * 15)
h.setInterval(2)
self.reactor.pump([0.1] * 50)
self.assertEqual(times, [1.1, 2.1, 4.1, 6.2])
h.stop()
d = w.wait()
self.assertTrue(d.called)
yield d
def test_ignores_duplicate_start_or_stop(self):
w = DeferWaiter()
times = []
def action():
times.append(round(self.reactor.seconds(), 1))
h = RepeatedActionHandler(self.reactor, w, 1, action)
h.start()
h.start()
self.reactor.pump([0.1] * 35)
self.assertEqual(times, [1.1, 2.1, 3.1])
h.stop()
h.stop()
d = w.wait()
self.assertTrue(d.called)
yield d
def test_add_non_deferred(self):
w = DeferWaiter()
w.add(2)
d = w.wait()
self.assertTrue(d.called)
