Support prediction cancelation

Author: erbridge

lib/Prediction.js

@@ -26,6 +26,10 @@ export default class Prediction extends ReplicateObject {
     return `GET /v1/predictions/${this.id}`;
   }
 
+  async cancel() {
+    await this.client.request(`POST /v1/predictions/${this.id}/cancel`);
+  }
+
   hasTerminalStatus() {
     return [
       PredictionStatus.CANCELED,

lib/Prediction.test.js

@@ -55,3 +55,18 @@ describe("load()", () => {
     expect(returnedPrediction).toBe(prediction);
   });
 });
+
+describe("cancel()", () => {
+  it("makes request to cancel prediction", async () => {
+    jest.spyOn(client, "request").mockResolvedValue({
+      id: "testprediction",
+      status: PredictionStatus.PROCESSING,
+    });
+
+    await prediction.cancel();
+
+    expect(client.request).toHaveBeenCalledWith(
+      "POST /v1/predictions/testprediction/cancel"
+    );
+  });
+});