metamask-extension/app/scripts/migrations/037.test.js

import assert from 'assert';
import migration37 from './037';

describe('migration #37', function () {
  it('should update the version metadata', function (done) {
    const oldStorage = {
      meta: {
        version: 36,
      },
      data: {},
    };

    migration37
      .migrate(oldStorage)
      .then((newStorage) => {
        assert.deepEqual(newStorage.meta, {
          version: 37,
        });
        done();
      })
      .catch(done);
  });

  it('should transform old state to new format', function (done) {
    const oldStorage = {
      meta: {},
      data: {
        AddressBookController: {
          addressBook: {
            '0x1De7e54679bfF0c23856FbF547b2394e723FCA91': {
              address: '0x1De7e54679bfF0c23856FbF547b2394e723FCA91',
              chainId: '4',
              memo: '',
              name: 'account 3',
            },
            '0x32Be343B94f860124dC4fEe278FDCBD38C102D88': {
              address: '0x32Be343B94f860124dC4fEe278FDCBD38C102D88',
              chainId: '4',
              memo: '',
              name: 'account 2',
            },
            // there are no repeated addresses by the current implementation
            '0x1De7e54679bfF0c23856FbF547b2394e723FCA93': {
              address: '0x1De7e54679bfF0c23856FbF547b2394e723FCA93',
              chainId: '2',
              memo: '',
              name: 'account 2',
            },
          },
        },
      },
    };

    migration37
      .migrate(oldStorage)
      .then((newStorage) => {
        assert.deepEqual(newStorage.data.AddressBookController.addressBook, {
          4: {
            '0x1De7e54679bfF0c23856FbF547b2394e723FCA91': {
              address: '0x1De7e54679bfF0c23856FbF547b2394e723FCA91',
              chainId: '4',
              isEns: false,
              memo: '',
              name: 'account 3',
            },
            '0x32Be343B94f860124dC4fEe278FDCBD38C102D88': {
              address: '0x32Be343B94f860124dC4fEe278FDCBD38C102D88',
              chainId: '4',
              isEns: false,
              memo: '',
              name: 'account 2',
            },
          },
          2: {
            '0x1De7e54679bfF0c23856FbF547b2394e723FCA93': {
              address: '0x1De7e54679bfF0c23856FbF547b2394e723FCA93',
              chainId: '2',
              isEns: false,
              memo: '',
              name: 'account 2',
            },
          },
        });
        done();
      })
      .catch(done);
  });

  it('ens validation test', function (done) {
    const oldStorage = {
      meta: {},
      data: {
        AddressBookController: {
          addressBook: {
            '0x1De7e54679bfF0c23856FbF547b2394e723FCA91': {
              address: '0x1De7e54679bfF0c23856FbF547b2394e723FCA91',
              chainId: '4',
              memo: '',
              name: 'metamask.eth',
            },
          },
        },
      },
    };

    migration37
      .migrate(oldStorage)
      .then((newStorage) => {
        assert.deepEqual(newStorage.data.AddressBookController.addressBook, {
          4: {
            '0x1De7e54679bfF0c23856FbF547b2394e723FCA91': {
              address: '0x1De7e54679bfF0c23856FbF547b2394e723FCA91',
              chainId: '4',
              isEns: true,
              memo: '',
              name: 'metamask.eth',
            },
          },
        });
        done();
      })
      .catch(done);
  });
});