diff --git a/integration-test/common/model.ts b/integration-test/common/model.ts index fae35a075..3e7648bd4 100644 --- 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 index 36a7ed810..461616446 100644 --- 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 index 632cd6a0a..254ccd7e8 100644 --- 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 index 258816eed..75daee25a 100644 --- 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 index 99a690059..c51c281b4 100644 --- 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" });