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