Refactor tests

Author: mattt

index.test.ts

@@ -114,50 +114,6 @@ describe("Replicate client", () => {
       const collections = await client.collections.list();
       expect(collections.results.length).toBe(2);
     });
-
-    describe("predictions.create", () => {
-      test("Handles array input correctly", async () => {
-        const inputArray = ["Alice", "Bob", "Charlie"];
-
-        nock(BASE_URL)
-          .post("/predictions", {
-            version:
-              "5c7d5dc6dd8bf75c1acaa8565735e7986bc5b66206b55cca93cb72c9bf15ccaa",
-            input: {
-              text: inputArray,
-            },
-          })
-          .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: inputArray,
-            },
-          });
-
-        const response = await client.predictions.create({
-          version:
-            "5c7d5dc6dd8bf75c1acaa8565735e7986bc5b66206b55cca93cb72c9bf15ccaa",
-          input: {
-            text: inputArray,
-          },
-        });
-
-        expect(response.input).toEqual({ text: inputArray });
-        expect(response.status).toBe("starting");
-      });
-    });
     // Add more tests for error handling, edge cases, etc.
   });
 
@@ -229,42 +185,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