|
|
|
|
@ -1,18 +1,30 @@ |
|
|
|
|
const assert = require('assert') |
|
|
|
|
const sinon = require('sinon') |
|
|
|
|
const clone = require('clone') |
|
|
|
|
const nock = require('nock') |
|
|
|
|
const MetaMaskController = require('../../app/scripts/metamask-controller') |
|
|
|
|
const blacklistJSON = require('../stub/blacklist') |
|
|
|
|
const firstTimeState = require('../../app/scripts/first-time-state') |
|
|
|
|
const BN = require('ethereumjs-util').BN |
|
|
|
|
const GWEI_BN = new BN('1000000000') |
|
|
|
|
|
|
|
|
|
describe('MetaMaskController', function () { |
|
|
|
|
let metamaskController |
|
|
|
|
const sandbox = sinon.sandbox.create() |
|
|
|
|
const noop = () => { } |
|
|
|
|
const metamaskController = new MetaMaskController({ |
|
|
|
|
showUnconfirmedMessage: noop, |
|
|
|
|
unlockAccountMessage: noop, |
|
|
|
|
|
|
|
|
|
beforeEach(function () { |
|
|
|
|
|
|
|
|
|
nock('https://api.infura.io') |
|
|
|
|
.persist() |
|
|
|
|
.get('/v2/blacklist') |
|
|
|
|
.reply(200, blacklistJSON) |
|
|
|
|
|
|
|
|
|
nock('https://api.infura.io') |
|
|
|
|
.persist() |
|
|
|
|
.get(/.*/) |
|
|
|
|
.reply(200) |
|
|
|
|
|
|
|
|
|
metamaskController = new MetaMaskController({ |
|
|
|
|
showUnapprovedTx: noop, |
|
|
|
|
platform: {}, |
|
|
|
|
encryptor: { |
|
|
|
|
encrypt: function (password, object) { |
|
|
|
|
this.object = object |
|
|
|
|
@ -20,33 +32,17 @@ describe('MetaMaskController', function () { |
|
|
|
|
}, |
|
|
|
|
decrypt: function () { |
|
|
|
|
return Promise.resolve(this.object) |
|
|
|
|
} |
|
|
|
|
}, |
|
|
|
|
// initial state
|
|
|
|
|
}, |
|
|
|
|
initState: clone(firstTimeState), |
|
|
|
|
}) |
|
|
|
|
|
|
|
|
|
beforeEach(function () { |
|
|
|
|
// sinon allows stubbing methods that are easily verified
|
|
|
|
|
this.sinon = sinon.sandbox.create() |
|
|
|
|
sandbox.spy(metamaskController.keyringController, 'createNewVaultAndKeychain') |
|
|
|
|
sandbox.spy(metamaskController.keyringController, 'createNewVaultAndRestore') |
|
|
|
|
}) |
|
|
|
|
|
|
|
|
|
afterEach(function () { |
|
|
|
|
// sinon requires cleanup otherwise it will overwrite context
|
|
|
|
|
this.sinon.restore() |
|
|
|
|
}) |
|
|
|
|
|
|
|
|
|
describe('Metamask Controller', function () { |
|
|
|
|
assert(metamaskController) |
|
|
|
|
|
|
|
|
|
beforeEach(function () { |
|
|
|
|
sinon.spy(metamaskController.keyringController, 'createNewVaultAndKeychain') |
|
|
|
|
sinon.spy(metamaskController.keyringController, 'createNewVaultAndRestore') |
|
|
|
|
}) |
|
|
|
|
|
|
|
|
|
afterEach(function () { |
|
|
|
|
metamaskController.keyringController.createNewVaultAndKeychain.restore() |
|
|
|
|
metamaskController.keyringController.createNewVaultAndRestore.restore() |
|
|
|
|
nock.cleanAll() |
|
|
|
|
sandbox.restore() |
|
|
|
|
}) |
|
|
|
|
|
|
|
|
|
describe('#getGasPrice', function () { |
|
|
|
|
@ -61,10 +57,10 @@ describe('MetaMaskController', function () { |
|
|
|
|
{ gasPrices: [ '0x3b9aca00', '0x174876e800'] }, |
|
|
|
|
{ gasPrices: [ '0x174876e800', '0x174876e800' ]}, |
|
|
|
|
{ gasPrices: [ '0x174876e800', '0x174876e800' ]}, |
|
|
|
|
] |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
], |
|
|
|
|
} |
|
|
|
|
}, |
|
|
|
|
}, |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
const gasPrice = metamaskController.getGasPrice() |
|
|
|
|
@ -72,36 +68,16 @@ describe('MetaMaskController', function () { |
|
|
|
|
|
|
|
|
|
metamaskController.recentBlocksController = realRecentBlocksController |
|
|
|
|
}) |
|
|
|
|
|
|
|
|
|
it('gives the 1 gwei price if no blocks have been seen.', async function () { |
|
|
|
|
const realRecentBlocksController = metamaskController.recentBlocksController |
|
|
|
|
metamaskController.recentBlocksController = { |
|
|
|
|
store: { |
|
|
|
|
getState: () => { |
|
|
|
|
return { |
|
|
|
|
recentBlocks: [] |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
const gasPrice = metamaskController.getGasPrice() |
|
|
|
|
assert.equal(gasPrice, '0x' + GWEI_BN.toString(16), 'defaults to 1 gwei') |
|
|
|
|
|
|
|
|
|
metamaskController.recentBlocksController = realRecentBlocksController |
|
|
|
|
}) |
|
|
|
|
|
|
|
|
|
}) |
|
|
|
|
|
|
|
|
|
describe('#createNewVaultAndKeychain', function () { |
|
|
|
|
it('can only create new vault on keyringController once', async function () { |
|
|
|
|
const selectStub = sinon.stub(metamaskController, 'selectFirstIdentity') |
|
|
|
|
|
|
|
|
|
const selectStub = sandbox.stub(metamaskController, 'selectFirstIdentity') |
|
|
|
|
|
|
|
|
|
const password = 'a-fake-password' |
|
|
|
|
|
|
|
|
|
const first = await metamaskController.createNewVaultAndKeychain(password) |
|
|
|
|
const second = await metamaskController.createNewVaultAndKeychain(password) |
|
|
|
|
await metamaskController.createNewVaultAndKeychain(password) |
|
|
|
|
await metamaskController.createNewVaultAndKeychain(password) |
|
|
|
|
|
|
|
|
|
assert(metamaskController.keyringController.createNewVaultAndKeychain.calledOnce) |
|
|
|
|
|
|
|
|
|
@ -111,19 +87,17 @@ describe('MetaMaskController', function () { |
|
|
|
|
|
|
|
|
|
describe('#createNewVaultAndRestore', function () { |
|
|
|
|
it('should be able to call newVaultAndRestore despite a mistake.', async function () { |
|
|
|
|
// const selectStub = sinon.stub(metamaskController, 'selectFirstIdentity')
|
|
|
|
|
|
|
|
|
|
const password = 'what-what-what' |
|
|
|
|
const wrongSeed = 'debris dizzy just program just float decrease vacant alarm reduce speak stadiu' |
|
|
|
|
const rightSeed = 'debris dizzy just program just float decrease vacant alarm reduce speak stadium' |
|
|
|
|
const first = await metamaskController.createNewVaultAndRestore(password, wrongSeed) |
|
|
|
|
await metamaskController.createNewVaultAndRestore(password, wrongSeed) |
|
|
|
|
.catch((e) => { |
|
|
|
|
return |
|
|
|
|
}) |
|
|
|
|
const second = await metamaskController.createNewVaultAndRestore(password, rightSeed) |
|
|
|
|
await metamaskController.createNewVaultAndRestore(password, rightSeed) |
|
|
|
|
|
|
|
|
|
assert(metamaskController.keyringController.createNewVaultAndRestore.calledTwice) |
|
|
|
|
}) |
|
|
|
|
}) |
|
|
|
|
}) |
|
|
|
|
}) |
|
|
|
|
|
Dan Finlay
7 years ago