def test_transition(self): """Check _JobState.transition()""" valid = [['INCOMING', 'PREPROCESSING'], ['ARCHIVED', 'EXPIRED'], ['RUNNING', 'FAILED'], ['FAILED', 'INCOMING'], ['POSTPROCESSING', 'RUNNING']] invalid = [['INCOMING', 'RUNNING'], ['INCOMING', 'garbage']] for instate, outstate in valid: j = _JobState(instate) j.transition(outstate) self.assertEqual(j.get(), outstate) for instate, outstate in invalid: j = _JobState(instate) self.assertRaises(InvalidStateError, j.transition, outstate)
def test_create_state(self): """Check making new _JobState objects""" for state in ('INCOMING', 'FAILED', 'RUNNING'): j = _JobState(state) self.assertRaises(InvalidStateError, _JobState, 'garbage')
def test_str(self): """Check _JobState.__str__()""" for state in ('INCOMING', 'FAILED', 'RUNNING'): j = _JobState(state) self.assertEqual(str(j), "<_JobState %s>" % state)
def test_get(self): """Check _JobState.get()""" for state in ('INCOMING', 'FAILED', 'RUNNING'): j = _JobState(state) self.assertEqual(j.get(), state)