def testWorkLeft(self): quests = [ quest_test.QuestCycle(quest_test.QuestPass(), quest_test.QuestSpin()) ] state = job_state.JobState(quests) state.AddChange(change_test.Change(123)) self.assertTrue(state.ScheduleWork())
def testAllAttemptsFail(self): quests = [quest_test.QuestCycle( quest_test.QuestFail, quest_test.QuestFail, quest_test.QuestFail2)] state = job_state.JobState(quests) state.AddChange(change_test.Change(123)) expected_regexp = '7/10.*\nException: Expected error for testing.$' self.assertTrue(state.ScheduleWork()) with self.assertRaisesRegexp(Exception, expected_regexp): self.assertFalse(state.ScheduleWork())