From 4745d114be3123ff9aa1d0fb98d0e1fe41995562 Mon Sep 17 00:00:00 2001 From: Nicolas Date: Wed, 15 May 2024 12:42:14 -0700 Subject: [PATCH] Update crawl.test.ts --- apps/test-suite/tests/crawl.test.ts | 23 +++++++++++++++++++---- 1 file changed, 19 insertions(+), 4 deletions(-) diff --git a/apps/test-suite/tests/crawl.test.ts b/apps/test-suite/tests/crawl.test.ts index b56a76e..cdf0945 100644 --- a/apps/test-suite/tests/crawl.test.ts +++ b/apps/test-suite/tests/crawl.test.ts @@ -49,14 +49,29 @@ describe("Crawling Checkup (E2E)", () => { .set("Authorization", `Bearer ${process.env.TEST_API_KEY}`) .send({ url: websiteData.website, pageOptions: { onlyMainContent: true }, crawlerOptions: { limit: 100 }}); - await new Promise(resolve => setTimeout(resolve, 300000)); // wait for 300 seconds + const jobId = crawlResponse.body.jobId; + let completedResponse; + let isFinished = false; - const completedResponse = await request(TEST_URL) - .get(`/v0/crawl/status/${crawlResponse.body.jobId}`) - .set("Authorization", `Bearer ${process.env.TEST_API_KEY}`); + while (!isFinished) { + completedResponse = await request(TEST_URL) + .get(`/v0/crawl/status/${jobId}`) + .set("Authorization", `Bearer ${process.env.TEST_API_KEY}`); + + isFinished = completedResponse.body.status === "completed"; + + if (!isFinished) { + await new Promise(resolve => setTimeout(resolve, 1000)); // Wait for 1 second before checking again + } + } console.log('-------------------') console.log(websiteData.website); + if(!completedResponse) { + // fail the test + console.log('No response'); + return null; + } if (!completedResponse.body.data) { console.log(completedResponse.body.partial_data.length);