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_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_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_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_add_non_deferred(self):
w = DeferWaiter()
w.add(2)
d = w.wait()
self.assertTrue(d.called)