Merge pull request #958 from MetaMask/i897-ReplayProtection

Add replay protection

1 files changed, 7 insertions, 4 deletions

diff --git a/test/unit/idStore-migration-test.js b/test/unit/idStore-migration-test.js
index 7e91616bf..54f38fb2f 100644
--- a/test/unit/idStore-migration-test.js
+++ b/test/unit/idStore-migration-test.js
@@ -83,11 +83,14 @@ describe('IdentityStore to KeyringController migration', function() {
   })
 
   describe('entering a password', function() {
-    it('should identify an old wallet as an initialized keyring', function() {
+    it('should identify an old wallet as an initialized keyring', function(done) {
       keyringController.configManager.setWallet('something')
-      const state = keyringController.getState()
-      assert(state.isInitialized, 'old vault counted as initialized.')
-      assert(!state.lostAccounts, 'no lost accounts')
+      keyringController.getState()
+      .then((state) => {
+        assert(state.isInitialized, 'old vault counted as initialized.')
+        assert(!state.lostAccounts, 'no lost accounts')
+        done()
+      })
     })
   })
 })