metamask-extension/test/unit/migrations/031-test.js

import assert from 'assert'
import migration31 from '../../../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)
  })
})