use listener to ensure expects are called

MetaMask · Mar 24, 2021 · 56c1c91 · 56c1c91
1 parent 41eb582
commit 56c1c91
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/transaction/TransactionController.test.ts b/src/transaction/TransactionController.test.ts
@@ -617,7 +617,7 @@ describe('TransactionController', () => {
       to: from,
     });
     controller.cancelTransaction('foo');
-    await new Promise(async (resolve) => {
+    const transactionListener = new Promise(async (resolve) => {
       controller.hub.once(`${controller.state.transactions[0].id}:finished`, () => {
         expect(controller.state.transactions[0].transaction.from).toBe(from);
         expect(controller.state.transactions[0].status).toBe(TransactionStatus.rejected);
@@ -626,6 +626,7 @@ describe('TransactionController', () => {
     });
     controller.cancelTransaction(controller.state.transactions[0].id);
     await expect(result).rejects.toThrow('User rejected the transaction');
+    await transactionListener;
   });
 
   it('should wipe transactions', async () => {