Added ability to nickname wallets locally

The changes are persisted to localstorage, so they cannot be restored on a new computer, but for right now it's a nice organizational feature.

2 files changed, 57 insertions, 0 deletions

diff --git a/test/unit/actions/save_account_label_test.js b/test/unit/actions/save_account_label_test.js
new file mode 100644
index 000000000..1df428b1d
--- /dev/null
+++ b/test/unit/actions/save_account_label_test.js
@@ -0,0 +1,36 @@
+var jsdom = require('mocha-jsdom')
+var assert = require('assert')
+var freeze = require('deep-freeze-strict')
+var path = require('path')
+
+var actions = require(path.join(__dirname, '..', '..', '..', 'ui', 'app', 'actions.js'))
+var reducers = require(path.join(__dirname, '..', '..',  '..', 'ui', 'app', 'reducers.js'))
+
+describe('SAVE_ACCOUNT_LABEL', function() {
+
+  it('updates the state.metamask.identities[:i].name property of the state to the action.value.label', function() {
+    var initialState = {
+      metamask: {
+        identities: {
+          foo: {
+            name: 'bar'
+          }
+        },
+      }
+    }
+    freeze(initialState)
+
+    const action = {
+      type: actions.SAVE_ACCOUNT_LABEL,
+      value: {
+        account: 'foo',
+        label: 'baz'
+      },
+    }
+    freeze(action)
+
+    var resultingState = reducers(initialState, action)
+    assert.equal(resultingState.metamask.identities.foo.name, action.value.label)
+  });
+});
+
diff --git a/test/unit/config-manager-test.js b/test/unit/config-manager-test.js
index e414ecb9e..aa94dc385 100644
--- a/test/unit/config-manager-test.js
+++ b/test/unit/config-manager-test.js
@@ -54,6 +54,27 @@ describe('config-manager', function() {
     })
   })
 
+  describe('wallet nicknames', function() {
+    it('should return null when no nicknames are saved', function() {
+      var nick = configManager.nicknameForWallet('0x0')
+      assert.equal(nick, null, 'no nickname returned')
+    })
+
+    it('should persist nicknames', function() {
+      var account = '0x0'
+      var nick1 = 'foo'
+      var nick2 = 'bar'
+      configManager.setNicknameForWallet(account, nick1)
+
+      var result1 = configManager.nicknameForWallet(account)
+      assert.equal(result1, nick1)
+
+      configManager.setNicknameForWallet(account, nick2)
+      var result2 = configManager.nicknameForWallet(account)
+      assert.equal(result2, nick2)
+    })
+  })
+
   describe('rpc manipulations', function() {
     it('changing rpc should return a different rpc', function() {
       var firstRpc = 'first'