Merge pull request #2153 from MetaMask/BreakOutKeyringController
Break out keyring controller into separate modulefeature/default_network_editable
kumavis
7 years ago
committed by
GitHub
commit
ca0dff06f5
@ -1,164 +0,0 @@ |
||||
const assert = require('assert') |
||||
const KeyringController = require('../../app/scripts/keyring-controller') |
||||
const configManagerGen = require('../lib/mock-config-manager') |
||||
const ethUtil = require('ethereumjs-util') |
||||
const BN = ethUtil.BN |
||||
const mockEncryptor = require('../lib/mock-encryptor') |
||||
const sinon = require('sinon') |
||||
|
||||
describe('KeyringController', function () { |
||||
let keyringController |
||||
const password = 'password123' |
||||
const seedWords = 'puzzle seed penalty soldier say clay field arctic metal hen cage runway' |
||||
const addresses = ['eF35cA8EbB9669A35c31b5F6f249A9941a812AC1'.toLowerCase()] |
||||
const accounts = [] |
||||
// let originalKeystore
|
||||
|
||||
beforeEach(function (done) { |
||||
this.sinon = sinon.sandbox.create() |
||||
window.localStorage = {} // Hacking localStorage support into JSDom
|
||||
|
||||
keyringController = new KeyringController({ |
||||
configManager: configManagerGen(), |
||||
txManager: { |
||||
getTxList: () => [], |
||||
getUnapprovedTxList: () => [], |
||||
}, |
||||
accountTracker: { |
||||
addAccount (acct) { accounts.push(ethUtil.addHexPrefix(acct)) }, |
||||
}, |
||||
encryptor: mockEncryptor, |
||||
}) |
||||
|
||||
keyringController.createNewVaultAndKeychain(password) |
||||
.then(function (newState) { |
||||
newState |
||||
done() |
||||
}) |
||||
.catch((err) => { |
||||
done(err) |
||||
}) |
||||
}) |
||||
|
||||
afterEach(function () { |
||||
// Cleanup mocks
|
||||
this.sinon.restore() |
||||
}) |
||||
|
||||
describe('#createNewVaultAndKeychain', function () { |
||||
this.timeout(10000) |
||||
|
||||
it('should set a vault on the configManager', function (done) { |
||||
keyringController.store.updateState({ vault: null }) |
||||
assert(!keyringController.store.getState().vault, 'no previous vault') |
||||
keyringController.createNewVaultAndKeychain(password) |
||||
.then(() => { |
||||
const vault = keyringController.store.getState().vault |
||||
assert(vault, 'vault created') |
||||
done() |
||||
}) |
||||
.catch((reason) => { |
||||
done(reason) |
||||
}) |
||||
}) |
||||
}) |
||||
|
||||
describe('#restoreKeyring', function () { |
||||
it(`should pass a keyring's serialized data back to the correct type.`, function (done) { |
||||
const mockSerialized = { |
||||
type: 'HD Key Tree', |
||||
data: { |
||||
mnemonic: seedWords, |
||||
numberOfAccounts: 1, |
||||
}, |
||||
} |
||||
const mock = this.sinon.mock(keyringController) |
||||
|
||||
mock.expects('getBalanceAndNickname') |
||||
.exactly(1) |
||||
|
||||
keyringController.restoreKeyring(mockSerialized) |
||||
.then((keyring) => { |
||||
assert.equal(keyring.wallets.length, 1, 'one wallet restored') |
||||
return keyring.getAccounts() |
||||
}) |
||||
.then((accounts) => { |
||||
assert.equal(accounts[0], addresses[0]) |
||||
mock.verify() |
||||
done() |
||||
}) |
||||
.catch((reason) => { |
||||
done(reason) |
||||
}) |
||||
}) |
||||
}) |
||||
|
||||
describe('#createNickname', function () { |
||||
it('should add the address to the identities hash', function () { |
||||
const fakeAddress = '0x12345678' |
||||
keyringController.createNickname(fakeAddress) |
||||
const identities = keyringController.memStore.getState().identities |
||||
const identity = identities[fakeAddress] |
||||
assert.equal(identity.address, fakeAddress) |
||||
}) |
||||
}) |
||||
|
||||
describe('#saveAccountLabel', function () { |
||||
it('sets the nickname', function (done) { |
||||
const account = addresses[0] |
||||
var nick = 'Test nickname' |
||||
const identities = keyringController.memStore.getState().identities |
||||
identities[ethUtil.addHexPrefix(account)] = {} |
||||
keyringController.memStore.updateState({ identities }) |
||||
keyringController.saveAccountLabel(account, nick) |
||||
.then((label) => { |
||||
try { |
||||
assert.equal(label, nick) |
||||
const persisted = keyringController.store.getState().walletNicknames[account] |
||||
assert.equal(persisted, nick) |
||||
done() |
||||
} catch (err) { |
||||
done() |
||||
} |
||||
}) |
||||
.catch((reason) => { |
||||
done(reason) |
||||
}) |
||||
}) |
||||
}) |
||||
|
||||
describe('#getAccounts', function () { |
||||
it('returns the result of getAccounts for each keyring', function (done) { |
||||
keyringController.keyrings = [ |
||||
{ getAccounts () { return Promise.resolve([1, 2, 3]) } }, |
||||
{ getAccounts () { return Promise.resolve([4, 5, 6]) } }, |
||||
] |
||||
|
||||
keyringController.getAccounts() |
||||
.then((result) => { |
||||
assert.deepEqual(result, [1, 2, 3, 4, 5, 6]) |
||||
done() |
||||
}) |
||||
}) |
||||
}) |
||||
|
||||
describe('#addGasBuffer', function () { |
||||
it('adds 100k gas buffer to estimates', function () { |
||||
const gas = '0x04ee59' // Actual estimated gas example
|
||||
const tooBigOutput = '0x80674f9' // Actual bad output
|
||||
const bnGas = new BN(ethUtil.stripHexPrefix(gas), 16) |
||||
const correctBuffer = new BN('100000', 10) |
||||
const correct = bnGas.add(correctBuffer) |
||||
|
||||
// const tooBig = new BN(tooBigOutput, 16)
|
||||
const result = keyringController.addGasBuffer(gas) |
||||
const bnResult = new BN(ethUtil.stripHexPrefix(result), 16) |
||||
|
||||
assert.equal(result.indexOf('0x'), 0, 'included hex prefix') |
||||
assert(bnResult.gt(bnGas), 'Estimate increased in value.') |
||||
assert.equal(bnResult.sub(bnGas).toString(10), '100000', 'added 100k gas') |
||||
assert.equal(result, '0x' + correct.toString(16), 'Added the right amount') |
||||
assert.notEqual(result, tooBigOutput, 'not that bad estimate') |
||||
}) |
||||
}) |
||||
}) |
||||
Loading…
Reference in new issue