Fix flaky tests and add utility to debug flaky tests

mrcnk · Feb 11, 2023 · 4ff6cc4 · 4ff6cc4
1 parent 0419b9d
commit 4ff6cc4
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 2 deletions.
diff --git a/background/services/chain/tests/db.integration.test.ts b/background/services/chain/tests/db.integration.test.ts
@@ -28,7 +28,9 @@ describe("Chain Database ", () => {
       expect(await db.getAccountsToTrack()).toHaveLength(1)
       await db.addAccountToTrack(account2)
       const accountsToTrack = await db.getAccountsToTrack()
-      expect(accountsToTrack).toEqual([account1, account2])
+      expect(accountsToTrack).toEqual(
+        expect.arrayContaining([account1, account2])
+      )
     })
 
     it("should not add the same account twice.", async () => {
@@ -127,7 +129,10 @@ describe("Chain Database ", () => {
       expect(await db.getAccountsToTrack()).toHaveLength(1)
       await db.addAccountToTrack(account2)
       const accountsToTrack = await db.getAccountsToTrack()
-      expect(accountsToTrack).toEqual([account1, account2])
+
+      expect(accountsToTrack).toEqual(
+        expect.arrayContaining([account1, account2])
+      )
     })
   })
   describe("getAllSavedTransactionHashes", () => {

diff --git a/setupJest.env.ts b/setupJest.env.ts
@@ -7,3 +7,24 @@ beforeEach(() => {
 afterEach(() => {
   global.indexedDB = new IDBFactory()
 })
+
+it.flaky = function checkFlaky(label: string, testCase: () => unknown): void {
+  // eslint-disable-next-line no-only-tests/no-only-tests
+  it.only(label, () => {
+    const results = []
+    for (let i = 0; i < 2000; i += 1) {
+      results.push(testCase())
+    }
+    return Promise.all(results)
+  })
+}
+
+// eslint-disable-next-line @typescript-eslint/no-namespace, @typescript-eslint/no-unused-vars
+declare namespace jest {
+  interface It {
+    /**
+     * Used for debugging flaky tests
+     */
+    flaky: (label: string, testCase: () => unknown) => void
+  }
+}