Wait for context menu

Author: code-asher

test/e2e/downloads.test.ts

@@ -18,7 +18,7 @@ describe("Downloads (enabled)", ["--disable-workspace-trust"], {}, async () => {
     // Action
     await codeServerPage.openContextMenu("text=unique-file.txt")
 
-    expect(await codeServerPage.page.isVisible("text=Download...")).toBe(true)
+    expect(await codeServerPage.page.waitForSelector("text=Download...")).toBe(true)
   })
 
   test("should see the 'Show Local' button on Save As", async ({ codeServerPage }) => {
@@ -72,7 +72,7 @@ describe("Downloads (disabled)", ["--disable-workspace-trust", "--disable-file-d
     // Action
     await codeServerPage.openContextMenu("text=unique-file.txt")
 
-    expect(await codeServerPage.page.isVisible("text=Download...")).toBe(false)
+    expect(await codeServerPage.page.waitForSelector("text=Download...")).toBe(false)
   })
 
   test("should not see the 'Show Local' button on Save as", async ({ codeServerPage }) => {