metamask-extension/test/unit/nonce-tracker-test.js

const assert = require('assert')
const NonceTracker = require('../../app/scripts/lib/nonce-tracker')
const MockTxGen = require('../lib/mock-tx-gen')

describe('Nonce Tracker', function () {
  let nonceTracker, provider
  let getPendingTransactions, pendingTxs
  let getConfirmedTransactions, confirmedTxs
  let providerResultStub = {}

  describe('#getNonceLock', function () {

    describe('with 3 confirmed and 1 pending', function () {
      beforeEach(function () {
        const txGen = new MockTxGen()
        confirmedTxs = txGen.generate({ status: 'confirmed' }, { count: 3 })
        pendingTxs = txGen.generate({ status: 'pending' }, { count: 1 })

        getPendingTransactions = () => pendingTxs
        getConfirmedTransactions = () => confirmedTxs
        providerResultStub.result = '0x3'
        provider = {
          sendAsync: (_, cb) => { cb(undefined, providerResultStub) },
          _blockTracker: {
            getCurrentBlock: () => '0x11b568',
          },
        }
        nonceTracker = new NonceTracker({
          provider,
          getPendingTransactions,
          getConfirmedTransactions,
        })
      })

      it('should work', async function () {
        this.timeout(15000)
        const nonceLock = await nonceTracker.getNonceLock('0x7d3517b0d011698406d6e0aed8453f0be2697926')
        assert.equal(nonceLock.nextNonce, '4', 'nonce should be 4')
        await nonceLock.releaseLock()
      })

      it('should use localNonce if network returns a nonce lower then a confirmed tx in state', async function () {
        this.timeout(15000)
        providerResultStub.result = '0x1'
        const nonceLock = await nonceTracker.getNonceLock('0x7d3517b0d011698406d6e0aed8453f0be2697926')
        assert.equal(nonceLock.nextNonce, '4', 'nonce should be 4')
        await nonceLock.releaseLock()
      })
    })

    describe('with no previous txs', function () {
      beforeEach(function () {
        getPendingTransactions = () => []
        getConfirmedTransactions = () => []
        providerResultStub.result = '0x0'
        provider = {
          sendAsync: (_, cb) => { cb(undefined, providerResultStub) },
          _blockTracker: {
            getCurrentBlock: () => '0x11b568',
          },
        }
        nonceTracker = new NonceTracker({
          provider,
          getPendingTransactions,
          getConfirmedTransactions,
        })
      })

      it('should return 0', async function () {
        this.timeout(15000)
        const nonceLock = await nonceTracker.getNonceLock('0x7d3517b0d011698406d6e0aed8453f0be2697926')
        assert.equal(nonceLock.nextNonce, '0', 'nonce should be 0')
        await nonceLock.releaseLock()
      })
    })
  })
})