parent
1ba74c1566
commit
7d243aacf9
@ -0,0 +1,61 @@ |
|||||||
|
// next version number
|
||||||
|
const version = 25 |
||||||
|
|
||||||
|
/* |
||||||
|
|
||||||
|
normalizes txParams on unconfirmed txs |
||||||
|
|
||||||
|
*/ |
||||||
|
const ethUtil = require('ethereumjs-util') |
||||||
|
const clone = require('clone') |
||||||
|
|
||||||
|
module.exports = { |
||||||
|
version, |
||||||
|
|
||||||
|
migrate: async function (originalVersionedData) { |
||||||
|
const versionedData = clone(originalVersionedData) |
||||||
|
versionedData.meta.version = version |
||||||
|
const state = versionedData.data |
||||||
|
const newState = transformState(state) |
||||||
|
versionedData.data = newState |
||||||
|
return versionedData |
||||||
|
}, |
||||||
|
} |
||||||
|
|
||||||
|
function transformState (state) { |
||||||
|
const newState = state |
||||||
|
|
||||||
|
if (newState.TransactionController) { |
||||||
|
if (newState.TransactionController.transactions) { |
||||||
|
const transactions = newState.TransactionController.transactions |
||||||
|
newState.TransactionController.transactions = transactions.map((txMeta) => { |
||||||
|
if (txMeta.status !== 'unapproved') return txMeta |
||||||
|
txMeta.txParams = normalizeTxParams(txMeta.txParams) |
||||||
|
return txMeta |
||||||
|
}) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
return newState |
||||||
|
} |
||||||
|
|
||||||
|
function normalizeTxParams (txParams) { |
||||||
|
// functions that handle normalizing of that key in txParams
|
||||||
|
const whiteList = { |
||||||
|
from: from => ethUtil.addHexPrefix(from).toLowerCase(), |
||||||
|
to: to => ethUtil.addHexPrefix(txParams.to).toLowerCase(), |
||||||
|
nonce: nonce => ethUtil.addHexPrefix(nonce), |
||||||
|
value: value => ethUtil.addHexPrefix(value), |
||||||
|
data: data => ethUtil.addHexPrefix(data), |
||||||
|
gas: gas => ethUtil.addHexPrefix(gas), |
||||||
|
gasPrice: gasPrice => ethUtil.addHexPrefix(gasPrice), |
||||||
|
} |
||||||
|
|
||||||
|
// apply only keys in the whiteList
|
||||||
|
const normalizedTxParams = {} |
||||||
|
Object.keys(whiteList).forEach((key) => { |
||||||
|
if (txParams[key]) normalizedTxParams[key] = whiteList[key](txParams[key]) |
||||||
|
}) |
||||||
|
|
||||||
|
return normalizedTxParams |
||||||
|
} |
@ -0,0 +1,49 @@ |
|||||||
|
const assert = require('assert') |
||||||
|
const migration25 = require('../../../app/scripts/migrations/025') |
||||||
|
const firstTimeState = { |
||||||
|
meta: {}, |
||||||
|
data: require('../../../app/scripts/first-time-state'), |
||||||
|
} |
||||||
|
|
||||||
|
const storage = { |
||||||
|
"meta": {}, |
||||||
|
"data": { |
||||||
|
"TransactionController": { |
||||||
|
"transactions": [ |
||||||
|
] |
||||||
|
}, |
||||||
|
}, |
||||||
|
} |
||||||
|
|
||||||
|
const transactions = [] |
||||||
|
|
||||||
|
|
||||||
|
while (transactions.length <= 10) { |
||||||
|
transactions.push({ txParams: { from: '0x8aCce2391c0d510a6c5E5d8f819a678f79b7e675', random: 'stuff', chainId: 2 }, status: 'unapproved' }) |
||||||
|
transactions.push({ txParams: { from: '0x8aCce2391c0d510a6c5E5d8f819a678f79b7e675' }, status: 'confirmed' }) |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
storage.data.TransactionController.transactions = transactions |
||||||
|
|
||||||
|
describe('storage is migrated successfully and the txParams.from are lowercase', () => { |
||||||
|
it('should lowercase the from for unapproved txs', (done) => { |
||||||
|
migration25.migrate(storage) |
||||||
|
.then((migratedData) => { |
||||||
|
const migratedTransactions = migratedData.data.TransactionController.transactions |
||||||
|
migratedTransactions.forEach((tx) => { |
||||||
|
if (tx.status === 'unapproved') assert(!tx.txParams.random) |
||||||
|
if (tx.status === 'unapproved') assert(!tx.txParams.chainId) |
||||||
|
}) |
||||||
|
done() |
||||||
|
}).catch(done) |
||||||
|
}) |
||||||
|
|
||||||
|
it('should migrate first time state', (done) => { |
||||||
|
migration25.migrate(firstTimeState) |
||||||
|
.then((migratedData) => { |
||||||
|
assert.equal(migratedData.meta.version, 25) |
||||||
|
done() |
||||||
|
}).catch(done) |
||||||
|
}) |
||||||
|
}) |
Loading…
Reference in new issue