1 files changed, 56 insertions, 0 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)
+  })
+})