Expect(fakeTaskDB.ResolvingTaskCallCount()).To(Equal(1))
_, taskGuid := fakeTaskDB.ResolvingTaskArgsForCall(0)
Expect(taskGuid).To(Equal("task-guid"))
Expect(responseRecorder.Code).To(Equal(http.StatusOK))
response := &models.TaskLifecycleResponse{}
err := response.Unmarshal(responseRecorder.Body.Bytes())
Expect(err).NotTo(HaveOccurred())
Expect(response.Error).To(BeNil())
})
})
Context("when desiring the task fails", func() {
BeforeEach(func() {
fakeTaskDB.ResolvingTaskReturns(models.ErrUnknownError)
})
It("responds with an error", func() {
Expect(responseRecorder.Code).To(Equal(http.StatusOK))
response := &models.TaskLifecycleResponse{}
err := response.Unmarshal(responseRecorder.Body.Bytes())
Expect(err).NotTo(HaveOccurred())
Expect(response.Error).To(Equal(models.ErrUnknownError))
})
})
})
})
Describe("DeleteTask", func() {
httpClient *http.Client
)
BeforeEach(func() {
httpClient = cf_http.NewClient()
statusCodes = make(chan int)
reqCount = make(chan struct{})
fakeServer.RouteToHandler("POST", "/the-callback/url", func(w http.ResponseWriter, req *http.Request) {
w.WriteHeader(<-statusCodes)
})
callbackURL = fakeServer.URL() + "/the-callback/url"
taskDB = new(dbFakes.FakeTaskDB)
taskDB.ResolvingTaskReturns(nil)
taskDB.DeleteTaskReturns(nil)
})
simulateTaskCompleting := func(signals <-chan os.Signal, ready chan<- struct{}) error {
close(ready)
task = model_helpers.NewValidTask("the-task-guid")
task.CompletionCallbackUrl = callbackURL
taskworkpool.HandleCompletedTask(logger, httpClient, taskDB, task)
return nil
}
var process ifrit.Process
JustBeforeEach(func() {
process = ifrit.Invoke(ifrit.RunFunc(simulateTaskCompleting))
})