feat: further delay the model details test

instill-ai · Feb 19, 2023 · bb0e955 · bb0e955
1 parent 30f5308
commit bb0e955
Show file tree

Hide file tree

Showing 5 changed files with 13 additions and 5 deletions.
diff --git a/integration-test/common/model.ts b/integration-test/common/model.ts
@@ -132,7 +132,7 @@ export const expectCorrectModelDetails = async ({
   additionalRules,
 }: ExpectCorrectModelDetailsProps) => {
   // Mimic the behavior of long running operation
-  await delay(10000);
+  await delay(20000);
 
   await page.goto(`/models/${modelId}`, { waitUntil: "networkidle" });
 
@@ -168,16 +168,16 @@ export const expectCorrectModelDetails = async ({
   const modelStateLabel = page.locator("data-testid=state-label");
   const stateToggle = page.locator("#pipelineStateToggleButton");
   if (modelState === "STATE_ONLINE") {
-    await expect(modelStateLabel).toHaveText("Online");
+    await expect(modelStateLabel).toHaveText("Online", { timeout: 20000 });
     expect(await stateToggle.isChecked()).toBeTruthy();
   } else if (modelState === "STATE_OFFLINE") {
-    await expect(modelStateLabel).toHaveText("Offline");
+    await expect(modelStateLabel).toHaveText("Offline", { timeout: 20000 });
     expect(await stateToggle.isChecked()).not.toBeTruthy();
   } else if (modelState === "STATE_UNSPECIFIED") {
-    await expect(modelStateLabel).toHaveText("Unspecified");
+    await expect(modelStateLabel).toHaveText("Unspecified", { timeout: 20000 });
     expect(await stateToggle.isChecked()).not.toBeTruthy();
   } else {
-    await expect(modelStateLabel).toHaveText("Error");
+    await expect(modelStateLabel).toHaveText("Error", { timeout: 20000 });
     expect(await stateToggle.isChecked()).not.toBeTruthy();
   }
 

diff --git a/integration-test/model-artivc.spec.ts b/integration-test/model-artivc.spec.ts
@@ -24,6 +24,8 @@ test.use({
 });
 
 test.describe.serial("Artivc model", () => {
+  test.setTimeout(75000);
+
   test("should create artivc model", async ({ page }) => {
     await page.goto("/models/create");
 

diff --git a/integration-test/model-github.spec.ts b/integration-test/model-github.spec.ts
@@ -24,6 +24,8 @@ test.use({
 });
 
 test.describe.serial("GitHub model", () => {
+  test.setTimeout(75000);
+
   test("should create github model", async ({ page }) => {
     await page.goto("/models/create");
 

diff --git a/integration-test/model-huggingface.spec.ts b/integration-test/model-huggingface.spec.ts
@@ -24,6 +24,8 @@ test.use({
 });
 
 test.describe.serial("Hugging face model", () => {
+  test.setTimeout(75000);
+
   test("should create huggingface model", async ({ page }) => {
     await page.goto("/models/create", { waitUntil: "networkidle" });
 

diff --git a/integration-test/model-local.spec.ts b/integration-test/model-local.spec.ts
@@ -23,6 +23,8 @@ test.use({
 });
 
 test.describe.serial("Local model", () => {
+  test.setTimeout(75000);
+
   test("should create local model", async ({ page }) => {
     await page.goto("/models/create", { waitUntil: "networkidle" });