add test for using localNonce

1 files changed, 42 insertions, 4 deletions

diff --git a/test/unit/nonce-tracker-test.js b/test/unit/nonce-tracker-test.js
index 36025a360..2171d859d 100644
--- a/test/unit/nonce-tracker-test.js
+++ b/test/unit/nonce-tracker-test.js
@@ -2,8 +2,10 @@ const assert = require('assert')
 const NonceTracker = require('../../app/scripts/lib/nonce-tracker')
 
 describe('Nonce Tracker', function () {
-  let nonceTracker, provider, getPendingTransactions, pendingTxs
-
+  let nonceTracker, provider
+  let getPendingTransactions, pendingTxs
+  let getConfirmedTransactions, confirmedTxs
+  let providerResultStub = {}
 
   beforeEach(function () {
     pendingTxs = [{
@@ -12,14 +14,41 @@ describe('Nonce Tracker', function () {
         'from': '0x7d3517b0d011698406d6e0aed8453f0be2697926',
         'gas': '0x30d40',
         'value': '0x0',
+        'nonce': '0x3',
+      },
+    }]
+    confirmedTxs = [{
+      'status': 'confirmed',
+      'txParams': {
+        'from': '0x7d3517b0d011698406d6e0aed8453f0be2697926',
+        'gas': '0x30d40',
+        'value': '0x0',
         'nonce': '0x0',
       },
+    }, {
+      'status': 'confirmed',
+      'txParams': {
+        'from': '0x7d3517b0d011698406d6e0aed8453f0be2697926',
+        'gas': '0x30d40',
+        'value': '0x0',
+        'nonce': '0x1',
+      },
+    }, {
+      'status': 'confirmed',
+      'txParams': {
+        'from': '0x7d3517b0d011698406d6e0aed8453f0be2697926',
+        'gas': '0x30d40',
+        'value': '0x0',
+        'nonce': '0x2',
+      },
     }]
 
 
     getPendingTransactions = () => pendingTxs
+    getConfirmedTransactions = () => confirmedTxs
+    providerResultStub.result = '0x3'
     provider = {
-      sendAsync: (_, cb) => { cb(undefined, {result: '0x0'}) },
+      sendAsync: (_, cb) => { cb(undefined, providerResultStub) },
       _blockTracker: {
         getCurrentBlock: () => '0x11b568',
       },
@@ -27,6 +56,7 @@ describe('Nonce Tracker', function () {
     nonceTracker = new NonceTracker({
       provider,
       getPendingTransactions,
+      getConfirmedTransactions,
     })
   })
 
@@ -34,7 +64,15 @@ describe('Nonce Tracker', function () {
     it('should work', async function () {
       this.timeout(15000)
       const nonceLock = await nonceTracker.getNonceLock('0x7d3517b0d011698406d6e0aed8453f0be2697926')
-      assert.equal(nonceLock.nextNonce, '1', 'nonce should be 1')
+      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()
     })
   })