fix: allow waitFor(async () => {}) to work

jcoreio · Dec 18, 2020 · e778c7f · e778c7f
1 parent 01570d3
commit e778c7f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/index.js b/src/index.js
@@ -16,7 +16,8 @@ class WaitFor {
       numAttempts++
       const thisAttemptStartTime = new Date().getTime()
       try {
-        await this.buildAssertion()
+        const assertion = await this.buildAssertion()
+        if (assertion) await assertion._obj
       } catch (error) {
         const now = new Date().getTime()
 

diff --git a/test/index.js b/test/index.js
@@ -52,7 +52,7 @@ describe('waitFor', function () {
       waitFor(async () => {
         await new Promise((resolve) => setTimeout(resolve, 200))
         if (i++ < 3) throw new Error('TEST')
-      }).to.be.fulfilled,
+      }),
       clock.tickAsync(1001),
     ])
     expect(i).to.equal(4)