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()