fix: unstable test case

janhq · Oct 26, 2023 · 945ba83 · 945ba83
1 parent ca29f63
commit 945ba83
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 7 deletions.
diff --git a/electron/tests/my-models.e2e.spec.ts b/electron/tests/my-models.e2e.spec.ts
@@ -36,11 +36,6 @@ test.afterAll(async () => {
 
 test("shows my models", async () => {
   await page.getByTestId("My Models").first().click();
-  const header = await page
-    .getByRole("heading")
-    .filter({ hasText: "My Models" })
-    .first()
-    .isVisible();
-  expect(header).toBe(false);
+  await page.getByTestId("testid-mymodels-header").isVisible();
   //   More test cases here...
 });
diff --git a/web/screens/MyModels/index.tsx b/web/screens/MyModels/index.tsx
@@ -65,7 +65,7 @@ const MyModelsScreen = () => {
   return (
     <div className="flex h-full w-full overflow-y-auto">
       <div className="w-full p-5">
-        <h1 className="text-lg font-semibold">My Models</h1>
+        <h1 data-testid="testid-mymodels-header" className="text-lg font-semibold">My Models</h1>
         <p className="mt-2 text-gray-600 dark:text-gray-400">
           You have <span>{downloadedModels.length}</span> models downloaded
         </p>