def test_reset_state(self):
"""Test instantiation and simple resets."""
point = ISO8601Point('2020')
taskdef = TaskDef('who-cares', {}, 'live', point, False)
taskstate = TaskState(taskdef, point, TASK_STATUS_WAITING, None)
self.assertIsNone(
taskstate.reset_state(TASK_STATUS_WAITING),
'same status returns None',
)
self.assertEqual(
taskstate.reset_state(TASK_STATUS_SUCCEEDED),
(TASK_STATUS_WAITING, None),
'different status returns previous (status, hold_swap)',
)
self.assertEqual(
(taskstate.status, taskstate.hold_swap),
(TASK_STATUS_SUCCEEDED, None),
'reset status OK',
)
def test_reset_state(self):
"""Test instantiation and simple resets."""
point = ISO8601Point('2020')
taskdef = TaskDef('who-cares', {}, 'live', point, False)
taskstate = TaskState(taskdef, point, TASK_STATUS_WAITING, None)
self.assertIsNone(
taskstate.reset_state(TASK_STATUS_WAITING),
'same status returns None',
)
self.assertEqual(
taskstate.reset_state(TASK_STATUS_SUCCEEDED),
(TASK_STATUS_WAITING, None),
'different status returns previous (status, hold_swap)',
)
self.assertEqual(
(taskstate.status, taskstate.hold_swap),
(TASK_STATUS_SUCCEEDED, None),
'reset status OK',
)
def test_reset_state_respect_hold_swap(self):
point = ISO8601Point('2020')
taskdef = TaskDef('who-cares', {}, 'live', point, False)
taskstate = TaskState(
taskdef, point, TASK_STATUS_HELD, TASK_STATUS_RETRYING)
self.assertIsNone(
taskstate.reset_state(
TASK_STATUS_RETRYING, respect_hold_swap=True),
'same status returns None',
)
self.assertEqual(
taskstate.reset_state(
TASK_STATUS_SUCCEEDED, respect_hold_swap=True),
(TASK_STATUS_HELD, TASK_STATUS_RETRYING),
'different status returns previous (status, hold_swap)',
)
self.assertEqual(
(taskstate.status, taskstate.hold_swap),
(TASK_STATUS_SUCCEEDED, None),
'reset status OK',
)
def test_reset_state_respect_hold_swap(self):
point = ISO8601Point('2020')
taskdef = TaskDef('who-cares', {}, 'live', point, False)
taskstate = TaskState(taskdef, point, TASK_STATUS_HELD,
TASK_STATUS_RETRYING)
self.assertIsNone(
taskstate.reset_state(TASK_STATUS_RETRYING,
respect_hold_swap=True),
'same status returns None',
)
self.assertEqual(
taskstate.reset_state(TASK_STATUS_SUCCEEDED,
respect_hold_swap=True),
(TASK_STATUS_HELD, TASK_STATUS_RETRYING),
'different status returns previous (status, hold_swap)',
)
self.assertEqual(
(taskstate.status, taskstate.hold_swap),
(TASK_STATUS_SUCCEEDED, None),
'reset status OK',
)