You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
115 lines
3.3 KiB
115 lines
3.3 KiB
import assert from 'assert'
|
|
import migration23 from '../../../app/scripts/migrations/023'
|
|
import { TRANSACTION_STATUSES } from '../../../shared/constants/transaction'
|
|
|
|
const storage = {
|
|
meta: {},
|
|
data: {
|
|
TransactionController: {
|
|
transactions: [],
|
|
},
|
|
},
|
|
}
|
|
|
|
const transactions = []
|
|
const transactions40 = []
|
|
const transactions20 = []
|
|
|
|
const txStates = Object.values(TRANSACTION_STATUSES)
|
|
|
|
const deletableTxStates = [
|
|
TRANSACTION_STATUSES.CONFIRMED,
|
|
TRANSACTION_STATUSES.REJECTED,
|
|
TRANSACTION_STATUSES.FAILED,
|
|
TRANSACTION_STATUSES.DROPPED,
|
|
]
|
|
|
|
let nonDeletableCount = 0
|
|
|
|
let status
|
|
while (transactions.length <= 100) {
|
|
status = txStates[Math.floor(Math.random() * Math.floor(txStates.length - 1))]
|
|
// This is an old migration, let's allow it
|
|
// eslint-disable-next-line no-loop-func
|
|
if (!deletableTxStates.find((s) => s === status)) {
|
|
nonDeletableCount += 1
|
|
}
|
|
transactions.push({ status })
|
|
}
|
|
|
|
while (transactions40.length < 40) {
|
|
status = txStates[Math.floor(Math.random() * Math.floor(txStates.length - 1))]
|
|
transactions40.push({ status })
|
|
}
|
|
|
|
while (transactions20.length < 20) {
|
|
status = txStates[Math.floor(Math.random() * Math.floor(txStates.length - 1))]
|
|
transactions20.push({ status })
|
|
}
|
|
|
|
storage.data.TransactionController.transactions = transactions
|
|
|
|
describe('storage is migrated successfully and the proper transactions are remove from state', function () {
|
|
it('should remove transactions that are unneeded', function (done) {
|
|
migration23
|
|
.migrate(storage)
|
|
.then((migratedData) => {
|
|
let leftoverNonDeletableTxCount = 0
|
|
const migratedTransactions =
|
|
migratedData.data.TransactionController.transactions
|
|
migratedTransactions.forEach((tx) => {
|
|
if (!deletableTxStates.find((s) => s === tx.status)) {
|
|
leftoverNonDeletableTxCount += 1
|
|
}
|
|
})
|
|
assert.equal(
|
|
leftoverNonDeletableTxCount,
|
|
nonDeletableCount,
|
|
"migration shouldn't delete transactions we want to keep",
|
|
)
|
|
assert(
|
|
migratedTransactions.length >= 40,
|
|
`should be equal or greater to 40 if they are non deletable states got ${migratedTransactions.length} transactions`,
|
|
)
|
|
done()
|
|
})
|
|
.catch(done)
|
|
})
|
|
|
|
it('should not remove any transactions because 40 is the expected limit', function (done) {
|
|
storage.meta.version = 22
|
|
storage.data.TransactionController.transactions = transactions40
|
|
migration23
|
|
.migrate(storage)
|
|
.then((migratedData) => {
|
|
const migratedTransactions =
|
|
migratedData.data.TransactionController.transactions
|
|
|
|
assert.equal(
|
|
migratedTransactions.length,
|
|
40,
|
|
"migration shouldn't delete when at limit",
|
|
)
|
|
done()
|
|
})
|
|
.catch(done)
|
|
})
|
|
|
|
it('should not remove any transactions because 20 txs is under the expected limit', function (done) {
|
|
storage.meta.version = 22
|
|
storage.data.TransactionController.transactions = transactions20
|
|
migration23
|
|
.migrate(storage)
|
|
.then((migratedData) => {
|
|
const migratedTransactions =
|
|
migratedData.data.TransactionController.transactions
|
|
assert.equal(
|
|
migratedTransactions.length,
|
|
20,
|
|
"migration shouldn't delete when under limit",
|
|
)
|
|
done()
|
|
})
|
|
.catch(done)
|
|
})
|
|
})
|
|
|