test: fix flaky e2e test

Calvin-LL · Sep 26, 2023 · 81ad4a1 · 81ad4a1
1 parent 6f2dcd8
commit 81ad4a1
Showing 1 changed file with 50 additions and 40 deletions.
diff --git a/tests/e2e/main.test.ts b/tests/e2e/main.test.ts
@@ -107,27 +107,32 @@ test("encrypt then decrypt files", async ({ page, context }) => {
   const file1Content = "test1";
   const file2Content = "test2";
 
-  const [plainFileChooser] = await Promise.all([
-    encryptPage.waitForEvent("filechooser"),
-    encryptPage.locator("input[type=file]").click(),
-  ]);
-  await plainFileChooser.setFiles([
-    {
-      name: "test1.txt",
-      mimeType: "text/plain",
-      buffer: Buffer.from(file1Content),
-    },
-    {
-      name: "test2.txt",
-      mimeType: "text/plain",
-      buffer: Buffer.from(file2Content),
+  const [[encryptedFile1Download, encryptedFile2Download]] = await retry(
+    async () => {
+      const [plainFileChooser] = await Promise.all([
+        encryptPage.waitForEvent("filechooser"),
+        encryptPage.locator("input[type=file]").click(),
+      ]);
+      await plainFileChooser.setFiles([
+        {
+          name: "test1.txt",
+          mimeType: "text/plain",
+          buffer: Buffer.from(file1Content),
+        },
+        {
+          name: "test2.txt",
+          mimeType: "text/plain",
+          buffer: Buffer.from(file2Content),
+        },
+      ]);
+
+      return await Promise.all([
+        waitForDownloads(encryptPage, 2),
+        encryptPage.locator("button").filter({ hasText: "encrypt" }).click(),
+      ]);
     },
-  ]);
-
-  const [[encryptedFile1Download, encryptedFile2Download]] = await Promise.all([
-    waitForDownloads(encryptPage, 2),
-    encryptPage.locator("button").filter({ hasText: "encrypt" }).click(),
-  ]);
+    3
+  );
 
   const decryptLink = await page
     .locator("#decryption-link-card a")
@@ -136,27 +141,32 @@ test("encrypt then decrypt files", async ({ page, context }) => {
   decryptPage.on("console", console.log);
   await astroGoto(decryptPage, decryptLink!);
 
-  const [encryptedFileChooser] = await Promise.all([
-    decryptPage.waitForEvent("filechooser"),
-    decryptPage.locator("input[type=file]").click(),
-  ]);
-  await encryptedFileChooser.setFiles([
-    {
-      name: "test1.txt",
-      mimeType: "text/plain",
-      buffer: await getDownloadContentBuffer(encryptedFile1Download),
-    },
-    {
-      name: "test2.txt",
-      mimeType: "text/plain",
-      buffer: await getDownloadContentBuffer(encryptedFile2Download),
+  const [[decryptedFile1Download, decryptedFile2Download]] = await retry(
+    async () => {
+      const [encryptedFileChooser] = await Promise.all([
+        decryptPage.waitForEvent("filechooser"),
+        decryptPage.locator("input[type=file]").click(),
+      ]);
+      await encryptedFileChooser.setFiles([
+        {
+          name: "test1.txt",
+          mimeType: "text/plain",
+          buffer: await getDownloadContentBuffer(encryptedFile1Download),
+        },
+        {
+          name: "test2.txt",
+          mimeType: "text/plain",
+          buffer: await getDownloadContentBuffer(encryptedFile2Download),
+        },
+      ]);
+
+      return await Promise.all([
+        waitForDownloads(decryptPage, 2),
+        decryptPage.locator("button").filter({ hasText: "decrypt" }).click(),
+      ]);
     },
-  ]);
-
-  const [[decryptedFile1Download, decryptedFile2Download]] = await Promise.all([
-    waitForDownloads(decryptPage, 2),
-    decryptPage.locator("button").filter({ hasText: "decrypt" }).click(),
-  ]);
+    3
+  );
 
   expect(await getDownloadContentString(decryptedFile1Download)).toBe(
     file1Content