}) }) }) Describe("Cancel Task", func() { getTaskStatus := func(state string) task_examiner.TaskInfo { return task_examiner.TaskInfo{ TaskGuid: "task-guid-1", State: state, } } It("cancel task when task not in COMPLETED state", func() { fakeTaskExaminer.TaskStatusReturns(getTaskStatus(receptor.TaskStatePending), nil) err := taskRunner.CancelTask("task-guid-1") Expect(err).ToNot(HaveOccurred()) }) It("return error when task in COMPLETED state", func() { fakeTaskExaminer.TaskStatusReturns(getTaskStatus(receptor.TaskStateCompleted), nil) err := taskRunner.CancelTask("task-guid-1") Expect(err).To(MatchError("Unable to cancel COMPLETED task")) }) It("returns error when task not found", func() { fakeTaskExaminer.TaskStatusReturns(task_examiner.TaskInfo{}, errors.New("Task not found")) err := taskRunner.CancelTask("task-guid-1") Expect(err).To(MatchError("Task not found"))