Fix persistent warning bug

1 files changed, 51 insertions, 0 deletions

diff --git a/test/unit/reducers/unlock_vault_test.js b/test/unit/reducers/unlock_vault_test.js
new file mode 100644
index 000000000..b7540af08
--- /dev/null
+++ b/test/unit/reducers/unlock_vault_test.js
@@ -0,0 +1,51 @@
+var jsdom = require('mocha-jsdom')
+var assert = require('assert')
+var freeze = require('deep-freeze-strict')
+var path = require('path')
+var sinon = require('sinon')
+
+var actions = require(path.join(__dirname, '..', '..', '..', 'ui', 'app', 'actions.js'))
+var reducers = require(path.join(__dirname, '..', '..',  '..', 'ui', 'app', 'reducers.js'))
+
+describe('#unlockMetamask(selectedAccount)', function() {
+
+  beforeEach(function() {
+    // sinon allows stubbing methods that are easily verified
+    this.sinon = sinon.sandbox.create()
+  })
+
+  afterEach(function() {
+    // sinon requires cleanup otherwise it will overwrite context
+    this.sinon.restore()
+  })
+
+  describe('after an error', function() {
+    it('clears warning', function() {
+      const warning = 'this is the wrong warning'
+      const account = 'foo_account'
+      const initialState = {
+        appState: {
+          warning: warning,
+        }
+      }
+
+      const resultState = reducers(initialState, actions.unlockMetamask(account))
+      assert.equal(resultState.appState.warning, null, 'warning nullified')
+    })
+  })
+
+  describe('going home after an error', function() {
+    it('clears warning', function() {
+      const warning = 'this is the wrong warning'
+      const account = 'foo_account'
+      const initialState = {
+        appState: {
+          warning: warning,
+        }
+      }
+
+      const resultState = reducers(initialState, actions.goHome())
+      assert.equal(resultState.appState.warning, null, 'warning nullified')
+    })
+  })
+})