From 0c057bb649cd6b6497d9a79fbf8c2374ea928958 Mon Sep 17 00:00:00 2001
From: Nicolas <nicolascamara29@gmail.com>
Date: Fri, 16 Aug 2024 16:45:10 -0400
Subject: [PATCH] Update index.test.ts

---
 .../__tests__/e2e_v1_withAuth/index.test.ts   | 33 +++++++++++++++++++
 1 file changed, 33 insertions(+)

diff --git a/apps/api/src/__tests__/e2e_v1_withAuth/index.test.ts b/apps/api/src/__tests__/e2e_v1_withAuth/index.test.ts
index 2a8df20e..6d2a0a14 100644
--- a/apps/api/src/__tests__/e2e_v1_withAuth/index.test.ts
+++ b/apps/api/src/__tests__/e2e_v1_withAuth/index.test.ts
@@ -391,14 +391,47 @@ describe("E2E Tests for v1 API Routes", () => {
           }
           expect(response.body.data).toHaveProperty("markdown");
           expect(response.body.data).not.toHaveProperty("html");
+          expect(response.body.data).not.toHaveProperty("links");
           expect(response.body.data).not.toHaveProperty("rawHtml");
           expect(response.body.data).toHaveProperty("metadata");
           expect(response.body.data.markdown).toContain("PagerDuty");
           expect(response.body.data.metadata.statusCode).toBe(200);
           expect(response.body.data.metadata.error).toBeUndefined();
+
         },
         30000
       );
 
+      it.concurrent(
+        "should return a successful response with a valid links on page",
+        async () => {
+          const scrapeRequest: ScrapeRequest = {
+            url: "https://roastmywebsite.ai",
+            formats: ["links"],
+          };
+  
+          const response: ScrapeResponseRequestTest = await request(TEST_URL)
+            .post("/v1/scrape")
+            .set("Authorization", `Bearer ${process.env.TEST_API_KEY}`)
+            .set("Content-Type", "application/json")
+            .send(scrapeRequest);
+          
+          expect(response.statusCode).toBe(200);
+          expect(response.body).toHaveProperty("data");
+          if (!("data" in response.body)) {
+            throw new Error("Expected response body to have 'data' property");
+          }
+          expect(response.body.data).not.toHaveProperty("html");
+          expect(response.body.data).not.toHaveProperty("rawHtml");
+          expect(response.body.data).toHaveProperty("links");
+          expect(response.body.data).toHaveProperty("metadata");
+          expect(response.body.data.links).toContain("https://firecrawl.dev");
+          expect(response.body.data.metadata.statusCode).toBe(200);
+          expect(response.body.data.metadata.error).toBeUndefined();
+        },
+        30000
+      );
+      
+
   });
 });