def test_immediate_bad(self):
retries = MockRetry(self)
retries.next()
self.assertRaises(self.failureException, retries.immediate)
def test_immediate(self):
retries = MockRetry(self, expect_immediate=True)
retries.next()
retries.immediate()
self.assertRaises(BreakException, retries.next)
retries.done()
def test_later(self):
retries = MockRetry(self, expect_later=True)
retries.next()
retries.later()
self.assertRaises(BreakException, retries.next)
retries.done()
def test_later_bad(self):
retries = MockRetry(self)
retries.next()
self.assertRaises(self.failureException, retries.later)
def test_next(self):
retries = MockRetry(self)
self.assertEquals(retries.next(), retries)
self.assertRaises(StopIteration, retries.next)
retries.done()
