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