Fixes QR and addresses after wallet recovery

Resolves brave#12170 Auditors: Test Plan:
NejcZdovc · Dec 16, 2017 · 97e7ee3 · 97e7ee3
1 parent 8272c80
commit 97e7ee3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/app/browser/api/ledger.js b/app/browser/api/ledger.js
@@ -1015,6 +1015,10 @@ const onWalletRecovery = (state, error, result) => {
       result = result.setIn(['properties', 'wallet', 'keyinfo', 'seed'], seed)
     }
 
+    // remove old QR codes and addresses
+    state = ledgerState.setInfoProp(state, 'walletQR', Immutable.Map())
+    state = ledgerState.setInfoProp(state, 'addresses', Immutable.Map())
+
     callback(error, result)
 
     if (balanceTimeoutId) {

diff --git a/test/unit/app/browser/api/ledgerTest.js b/test/unit/app/browser/api/ledgerTest.js
@@ -1539,6 +1539,10 @@ describe('ledger api unit tests', function () {
             updatedStamp: 0
           }
         }))
+        .setIn(['ledger', 'info'], Immutable.fromJS({
+          addresses: {},
+          walletQR: {}
+        }))
       assert(setRecoveryStatusSpy.withArgs(sinon.match.any, true))
       assert.deepEqual(result.toJS(), expectedSate.toJS())
       const callBack = onLedgerCallbackSpy.getCall(0).args[0]