def test_get_task(self):
# Setup
task_id = '123'
states = ['waiting', 'running', 'success']
sim = TaskSimulator()
sim.add_task_states(task_id, states)
# Test & Verify
for state in states:
task = sim.get_task(task_id).response_body
self.assertEqual(task.state, state)
self.assertEqual(0, len(sim.tasks_by_id[task_id]))
def test_get_task(self):
# Setup
task_id = "123"
states = ["waiting", "running", "success"]
sim = TaskSimulator()
sim.add_task_states(task_id, states)
# Test & Verify
for state in states:
task = sim.get_task(task_id).response_body
self.assertEqual(task.state, state)
self.assertEqual(0, len(sim.tasks_by_id[task_id]))
