More robust integration tests

jupyterlab · Nov 16, 2023 · 0589d8b · 0589d8b
1 parent 18cb50e
commit 0589d8b
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/ui-tests/tests/merge-commit.spec.ts b/ui-tests/tests/merge-commit.spec.ts
@@ -67,7 +67,7 @@ test.describe('Merge commit tests', () => {
       .getByRole('tab', { name: 'hello-world.py' })
       .waitFor({ state: 'visible' });
 
-    expect(page.waitForSelector('.jp-git-diff-root')).toBeTruthy();
+    await expect(page.locator('.jp-git-diff-root')).toBeVisible();
   });
 
   test('should revert merge commit', async ({ page }) => {
@@ -92,8 +92,6 @@ test.describe('Merge commit tests', () => {
       .locator('#jp-git-sessions')
       .getByText("Revert 'Merge branch 'sort-names''");
 
-    await revertMergeCommit.waitFor({ state: 'visible' });
-
-    expect(revertMergeCommit).toBeTruthy();
+    await expect(revertMergeCommit).toBeVisible();
   });
 });