Add tests for task_step

Change-Id: I9f7ec4633e91264b6821ae3edb3a021e8843553d