Refactor tests

Author: mattt

index.test.ts

@@ -229,42 +229,73 @@ describe("Replicate client", () => {
   });
 
   describe("predictions.create", () => {
-    test("Calls the correct API route with the correct payload", async () => {
-      nock(BASE_URL)
-        .post("/predictions")
-        .reply(200, {
-          id: "ufawqhfynnddngldkgtslldrkq",
-          model: "replicate/hello-world",
-          version:
-            "5c7d5dc6dd8bf75c1acaa8565735e7986bc5b66206b55cca93cb72c9bf15ccaa",
-          urls: {
-            get: "https://api.replicate.com/v1/predictions/ufawqhfynnddngldkgtslldrkq",
-            cancel:
-              "https://api.replicate.com/v1/predictions/ufawqhfynnddngldkgtslldrkq/cancel",
-          },
-          created_at: "2022-04-26T22:13:06.224088Z",
-          started_at: null,
-          completed_at: null,
-          status: "starting",
-          input: {
-            text: "Alice",
+    const predictionTestCases = [
+      {
+        description: "String input",
+        input: {
+          text: "Alice",
+        },
+      },
+      {},
+      {
+        description: "Array input",
+        input: {
+          text: ["Alice", "Bob", "Charlie"],
+        },
+      },
+      {
+        description: "Object input",
+        input: {
+          text: {
+            name: "Alice",
           },
-          output: null,
-          error: null,
-          logs: null,
-          metrics: {},
-        });
-      const prediction = await client.predictions.create({
+        },
+      },
+    ].map((testCase) => ({
+      ...testCase,
+      expectedResponse: {
+        id: "ufawqhfynnddngldkgtslldrkq",
+        model: "replicate/hello-world",
         version:
           "5c7d5dc6dd8bf75c1acaa8565735e7986bc5b66206b55cca93cb72c9bf15ccaa",
-        input: {
-          text: "Alice",
+        urls: {
+          get: "https://api.replicate.com/v1/predictions/ufawqhfynnddngldkgtslldrkq",
+          cancel:
+            "https://api.replicate.com/v1/predictions/ufawqhfynnddngldkgtslldrkq/cancel",
         },
-        webhook: "http://test.host/webhook",
-        webhook_events_filter: ["output", "completed"],
-      });
-      expect(prediction.id).toBe("ufawqhfynnddngldkgtslldrkq");
-    });
+        input: testCase.input,
+        created_at: "2022-04-26T22:13:06.224088Z",
+        started_at: null,
+        completed_at: null,
+        status: "starting",
+      },
+    }));
+
+    test.each(predictionTestCases)(
+      "$description",
+      async ({ input, expectedResponse }) => {
+        nock(BASE_URL)
+          .post("/predictions", {
+            version:
+              "5c7d5dc6dd8bf75c1acaa8565735e7986bc5b66206b55cca93cb72c9bf15ccaa",
+            input: input as Record<string, any>,
+            webhook: "http://test.host/webhook",
+            webhook_events_filter: ["output", "completed"],
+          })
+          .reply(200, expectedResponse);
+
+        const response = await client.predictions.create({
+          version:
+            "5c7d5dc6dd8bf75c1acaa8565735e7986bc5b66206b55cca93cb72c9bf15ccaa",
+          input: input as Record<string, any>,
+          webhook: "http://test.host/webhook",
+          webhook_events_filter: ["output", "completed"],
+        });
+
+        expect(response.input).toEqual(input);
+        expect(response.status).toBe(expectedResponse.status);
+      }
+    );
 
     const fileTestCases = [
       // Skip test case if File type is not available