added cancel job

Author: ClaytonNorthey92

jobs.go

@@ -72,6 +72,7 @@ type GetJobOutputRequest GetJobRequest
 
 // these have the same structure
 type DeleteJobRequest GetJobRequest
+type CancelJobRequest GetJobRequest
 
 type DeleteJobResponseWithStatus struct {
 	Response DeleteJobResponse
@@ -141,6 +142,16 @@ type DeleteJobResponse struct {
 	Status string `json:"status"`
 }
 
+type CancelJobResponse struct {
+	ID     string `json:"id"`
+	Status string `json:"status"`
+}
+
+type CancelJobResponseWithStatus struct {
+	Response CancelJobResponse
+	Status   int
+}
+
 type DeleteManyJobsResponseWithStatus struct {
 	Response DeleteManyJobsResponse
 	Status   int
@@ -370,3 +381,35 @@ func (c *Client) DeleteJob(ctx context.Context, deleteJobRequest *DeleteJobReque
 		Status:   res.StatusCode,
 	}, nil
 }
+
+func (c *Client) CancelJob(ctx context.Context, cancelJobRequest *CancelJobRequest) (*CancelJobResponseWithStatus, error) {
+	url := c.makeURL(fmt.Sprintf("%s/%s/status/cancel", jobsPath, cancelJobRequest.ID))
+
+	req, err := http.NewRequestWithContext(ctx, http.MethodPut, url, http.NoBody)
+	if err != nil {
+		return nil, err
+	}
+
+	c.setHeaders(req)
+
+	res, err := c.client.Do(req)
+	if err != nil {
+		return nil, err
+	}
+	defer res.Body.Close()
+
+	body, err := io.ReadAll(res.Body)
+	if err != nil {
+		return nil, err
+	}
+
+	var cancelJobResponse CancelJobResponse
+	if err := json.Unmarshal(body, &cancelJobResponse); err != nil {
+		return nil, err
+	}
+
+	return &CancelJobResponseWithStatus{
+		Response: cancelJobResponse,
+		Status:   res.StatusCode,
+	}, nil
+}

jobs_test.go

@@ -529,3 +529,42 @@ func TestGetJobOutputSuccess(t *testing.T) {
 		t.Fatalf("unexpected diff: %s", diff)
 	}
 }
+
+func TestCancelJobSuccess(t *testing.T) {
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+
+	defer gock.Off()
+
+	jobResponseMock := CancelJobResponse{
+		ID:     "some-id",
+		Status: "canceled",
+	}
+
+	mockJson, err := json.Marshal(&jobResponseMock)
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Logf("will mock response as: %s", mockJson)
+
+	newGock().
+		Put(fmt.Sprintf("%s/some-id/status/cancel", jobsPath)).
+		Reply(200).
+		JSON(&jobResponseMock)
+
+	client := NewClient(myFakeEndpoint, myFakeAPIKey)
+	jobResponseWithStatus, err := client.CancelJob(ctx, &CancelJobRequest{
+		ID: "some-id",
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if jobResponseWithStatus.Status != http.StatusOK {
+		t.Fatalf("unexpected status: %d", jobResponseWithStatus.Status)
+	}
+
+	if diff := deep.Equal(jobResponseMock, jobResponseWithStatus.Response); len(diff) > 0 {
+		t.Fatalf("unexpected diff: %s", diff)
+	}
+}