api: Add test for segment-anything-2

livepeer · Sep 16, 2024 · a37bdad · a37bdad
1 parent 0d0ff68
commit a37bdad
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/packages/api/src/controllers/generate.test.ts b/packages/api/src/controllers/generate.test.ts
@@ -90,6 +90,7 @@ describe("controllers/generate", () => {
       "image-to-image",
       "image-to-video",
       "upscale",
+      "segment-anything-2",
     ];
     for (const api of apis) {
       aiGatewayServer.app.post(`/${api}`, async (req, res) => {
@@ -213,6 +214,19 @@ describe("controllers/generate", () => {
       });
       expect(aiGatewayCalls).toEqual({ upscale: 1 });
     });
+
+    it("should call the AI Gateway for generate API /segment-anything-2", async () => {
+      const res = await client.fetch("/beta/generate/segment-anything-2", {
+        method: "POST",
+        body: buildMultipartBody({}),
+      });
+      expect(res.status).toBe(200);
+      expect(await res.json()).toEqual({
+        message: "success",
+        reqContentType: expect.stringMatching("^multipart/form-data"),
+      });
+      expect(aiGatewayCalls).toEqual({ "segment-anything-2": 1 });
+    });
   });
 
   describe("validates multipart schema", () => {