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);
  });
});