small adjustment

GoodDollar · Jun 4, 2020 · e6bb562 · e6bb562
1 parent 19ce573
commit e6bb562
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/src/server/db/__tests__/UserPrivate.js b/src/server/db/__tests__/UserPrivate.js
@@ -276,11 +276,9 @@ describe('UserPrivate', () => {
     await storage.completeDelayedTasks([_id])
 
     // no complete/failed tasks should be found despite we've called corresponding storage methods
-    await Promise.all(
-      [Complete, Failed].map(async status =>
-        expect(taskModel.find({ taskName: testTaskName, status })).resolves.toBeArrayOfSize(0)
-      )
-    )
+    await expect(
+      taskModel.find({ taskName: testTaskName, status: { $in: [Complete, Failed] } })
+    ).resolves.toBeArrayOfSize(0)
   })
 
   it('Should remove delayed tasks', async () => {