metamask-extension/test/unit/migrations/023-test.js

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