2 files changed, 59 insertions, 3 deletions
diff --git a/test/unit/migrations/031-test.js b/test/unit/migrations/031-test.js
new file mode 100644
index 000000000..c85fd7af4
--- /dev/null
+++ b/test/unit/migrations/031-test.js
@@ -0,0 +1,56 @@
+const assert = require('assert')
+const migration31 = require('../../../app/scripts/migrations/031')
+
+ describe('migration #31', () => {
+  it('should set completedOnboarding to true if vault exists', done => {
+    const oldStorage = {
+      'meta': {},
+      'data': {
+        'PreferencesController': {
+          'tokens': [{address: '0xa', symbol: 'A', decimals: 4}, {address: '0xb', symbol: 'B', decimals: 4}],
+          'identities': {
+            '0x6d14': {},
+            '0x3695': {},
+          },
+        },
+        'KeyringController': {
+          'vault': {
+            'data': 'test0',
+            'iv': 'test1',
+            'salt': 'test2',
+          },
+        },
+      },
+    }
+
+     migration31.migrate(oldStorage)
+      .then(newStorage => {
+        assert.equal(newStorage.data.PreferencesController.completedOnboarding, true)
+        done()
+      })
+      .catch(done)
+  })
+
+   it('should set completedOnboarding to false if vault does not exist', done => {
+    const oldStorage = {
+      'meta': {},
+      'data': {
+        'PreferencesController': {
+          'tokens': [{address: '0xa', symbol: 'A', decimals: 4}, {address: '0xb', symbol: 'B', decimals: 4}],
+          'identities': {
+            '0x6d14': {},
+            '0x3695': {},
+          },
+        },
+        'KeyringController': {},
+      },
+    }
+
+     migration31.migrate(oldStorage)
+      .then(newStorage => {
+        assert.equal(newStorage.data.PreferencesController.completedOnboarding, false)
+        done()
+      })
+      .catch(done)
+  })
+})
diff --git a/test/unit/ui/app/actions.spec.js b/test/unit/ui/app/actions.spec.js
index c7ac8b6cf..8d7de8b02 100644
--- a/test/unit/ui/app/actions.spec.js
+++ b/test/unit/ui/app/actions.spec.js
@@ -198,7 +198,7 @@ describe('Actions', () => {
       createNewVaultAndRestoreSpy = sinon.spy(background, 'createNewVaultAndRestore')
       clearSeedWordCacheSpy = sinon.spy(background, 'clearSeedWordCache')
       return store.dispatch(actions.createNewVaultAndRestore())
-        .then(() => {
+        .catch(() => {
           assert(clearSeedWordCacheSpy.calledOnce)
           assert(createNewVaultAndRestoreSpy.calledOnce)
         })
@@ -218,7 +218,7 @@ describe('Actions', () => {
       })
 
       return store.dispatch(actions.createNewVaultAndRestore())
-        .then(() => {
+        .catch(() => {
           assert.deepEqual(store.getActions(), expectedActions)
         })
     })
@@ -240,7 +240,7 @@ describe('Actions', () => {
       })
 
       return store.dispatch(actions.createNewVaultAndRestore())
-        .then(() => {
+        .catch(() => {
           assert.deepEqual(store.getActions(), expectedActions)
         })
     })