Add blockchainLifecycle management to the ExpirationWatcher test

1 files changed, 6 insertions, 2 deletions

diff --git a/packages/0x.js/test/expiration_watcher_test.ts b/packages/0x.js/test/expiration_watcher_test.ts
index 0d5ebb909..0f2470070 100644
--- a/packages/0x.js/test/expiration_watcher_test.ts
+++ b/packages/0x.js/test/expiration_watcher_test.ts
@@ -12,11 +12,13 @@ import {TokenUtils} from './utils/token_utils';
 import {ExpirationWatcher} from '../src/order_watcher/expiration_watcher';
 import {Token, DoneCallback} from '../src/types';
 import {ZeroEx} from '../src/0x';
+import {BlockchainLifecycle} from './utils/blockchain_lifecycle';
 import {FillScenarios} from './utils/fill_scenarios';
 import {reportCallbackErrors} from './utils/report_callback_errors';
 
 chaiSetup.configure();
 const expect = chai.expect;
+const blockchainLifecycle = new BlockchainLifecycle();
 
 describe('ExpirationWatcher', () => {
     let web3: Web3;
@@ -52,14 +54,16 @@ describe('ExpirationWatcher', () => {
         makerTokenAddress = makerToken.address;
         takerTokenAddress = takerToken.address;
     });
-    beforeEach(() => {
+    beforeEach(async () => {
+        await blockchainLifecycle.startAsync();
         const sinonTimerConfig = {shouldAdvanceTime: true} as any;
         // This constructor has incorrect types
         timer = Sinon.useFakeTimers(sinonTimerConfig);
         currentUnixTimestampSec = utils.getCurrentUnixTimestampSec();
         expirationWatcher = new ExpirationWatcher();
     });
-    afterEach(() => {
+    afterEach(async () => {
+        await blockchainLifecycle.revertAsync();
         timer.restore();
         expirationWatcher.unsubscribe();
     });