diff --git a/app/scripts/metamask-controller.test.js b/app/scripts/metamask-controller.test.js index 67103aee5db5..53badde1d9ec 100644 --- a/app/scripts/metamask-controller.test.js +++ b/app/scripts/metamask-controller.test.js @@ -1248,6 +1248,8 @@ describe('MetaMaskController', () => { }; const { promise, resolve } = deferredPromise(); + const { promise: promiseStream, resolve: resolveStream } = + deferredPromise(); const streamTest = createThroughStream((chunk, _, cb) => { if (chunk.name !== 'metamask-phishing-safelist') { cb(); @@ -1270,10 +1272,12 @@ describe('MetaMaskController', () => { expect( metamaskController.backToSafetyPhishingWarning, ).toHaveBeenCalled(); + resolveStream(); }); await promise; streamTest.end(); + await promiseStream; }); }); @@ -1316,6 +1320,8 @@ describe('MetaMaskController', () => { }; const { promise, resolve } = deferredPromise(); + const { promise: promiseStream, resolve: resolveStream } = + deferredPromise(); const streamTest = createThroughStream((chunk, _, cb) => { if (chunk.name !== METAMASK_COOKIE_HANDLER) { cb(); @@ -1333,10 +1339,12 @@ describe('MetaMaskController', () => { expect( localMetaMaskController.getCookieFromMarketingPage, ).toHaveBeenCalledWith({ ga_client_id: 'XYZ.ABC' }); + resolveStream(); }); await promise; streamTest.end(); + await promiseStream; }); });