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

import assert from 'assert'
import firstTimeState from '../../../app/scripts/first-time-state'
import migration28 from '../../../app/scripts/migrations/028'

const oldStorage = {
  meta: {},
  data: {
    PreferencesController: {
      tokens: [
        { address: '0xa', symbol: 'A', decimals: 4 },
        { address: '0xb', symbol: 'B', decimals: 4 },
      ],
      identities: {
        '0x6d14': {},
        '0x3695': {},
      },
    },
  },
}

describe('migration #28', function () {
  it('should add corresponding tokens to accountTokens', function (done) {
    migration28
      .migrate(oldStorage)
      .then((newStorage) => {
        const newTokens = newStorage.data.PreferencesController.tokens
        const newAccountTokens =
          newStorage.data.PreferencesController.accountTokens

        const testTokens = [
          { address: '0xa', symbol: 'A', decimals: 4 },
          { address: '0xb', symbol: 'B', decimals: 4 },
        ]
        assert.equal(
          newTokens.length,
          0,
          'tokens is expected to have the length of 0',
        )
        assert.equal(
          newAccountTokens['0x6d14'].mainnet.length,
          2,
          'tokens for address is expected to have the length of 2',
        )
        assert.equal(
          newAccountTokens['0x3695'].mainnet.length,
          2,
          'tokens for address is expected to have the length of 2',
        )
        assert.equal(
          Object.keys(newAccountTokens).length,
          2,
          'account tokens should be created for all identities',
        )
        assert.deepEqual(
          newAccountTokens['0x6d14'].mainnet,
          testTokens,
          'tokens for address should be the same than before',
        )
        assert.deepEqual(
          newAccountTokens['0x3695'].mainnet,
          testTokens,
          'tokens for address should be the same than before',
        )
        done()
      })
      .catch(done)
  })

  it('should successfully migrate first time state', function (done) {
    migration28
      .migrate({
        meta: {},
        data: firstTimeState,
      })
      .then((migratedData) => {
        assert.equal(migratedData.meta.version, migration28.version)
        done()
      })
      .catch(done)
  })
})