def test_get_status(self): action = Action() task_id = 'task_1' task = dict(cmd='reverse', string='test123', status=STATUS.queued) action.task[task_id].update(task) self.assertEqual(STATUS.queued.value, action.get_status(task_id)) task.update(dict(status=STATUS.progress)) action.task[task_id].update(task) self.assertEqual(STATUS.progress.value, action.get_status(task_id)) task.update(dict(status=STATUS.complete)) action.task[task_id].update(task) self.assertEqual(STATUS.complete.value, action.get_status(task_id))