aboutsummaryrefslogtreecommitdiffstats
path: root/test/unit/metamask-controller-test.js
blob: 9ec7cd0af7985b5036f5c326447d41b7a55b4f73 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
const assert = require('assert')
const sinon = require('sinon')
const clone = require('clone')
const MetaMaskController = require('../../app/scripts/metamask-controller')
const firstTimeState = require('../../app/scripts/first-time-state')
const BN = require('ethereumjs-util').BN
const GWEI_BN = new BN('1000000000')

describe('MetaMaskController', function () {
  const noop = () => {}
  const metamaskController = new MetaMaskController({
    showUnconfirmedMessage: noop,
    unlockAccountMessage: noop,
    showUnapprovedTx: noop,
    platform: {},
    encryptor: {
      encrypt: function(password, object) {
        this.object = object
        return Promise.resolve()
      },
      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()
  })

  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()
    })

    describe('#getGasPrice', function () {
      it('gives the 50th percentile lowest accepted gas price from recentBlocksController', async function () {
        const realRecentBlocksController = metamaskController.recentBlocksController
        metamaskController.recentBlocksController = {
          store: {
            getState: () => {
              return {
                recentBlocks: [
                  { gasPrices: [ '0x3b9aca00', '0x174876e800'] },
                  { gasPrices: [ '0x3b9aca00', '0x174876e800'] },
                  { gasPrices: [ '0x174876e800', '0x174876e800' ]},
                  { gasPrices: [ '0x174876e800', '0x174876e800' ]},
                ]
              }
            }
          }
        }

        const gasPrice = metamaskController.getGasPrice()
        assert.equal(gasPrice, '0x3b9aca00', 'accurately estimates 50th percentile accepted gas price')

        metamaskController.recentBlocksController = realRecentBlocksController
      })
    })

    describe('#createNewVaultAndKeychain', function () {
      it('can only create new vault on keyringController once', async function () {
        const selectStub = sinon.stub(metamaskController, 'selectFirstIdentity')


        const password = 'a-fake-password'

        const first = await metamaskController.createNewVaultAndKeychain(password)
        const second = await metamaskController.createNewVaultAndKeychain(password)

        assert(metamaskController.keyringController.createNewVaultAndKeychain.calledOnce)

        selectStub.reset()
      })
    })

    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)
            .catch((e) => {
              return
            })
          const second = await metamaskController.createNewVaultAndRestore(password, rightSeed)

        assert(metamaskController.keyringController.createNewVaultAndRestore.calledTwice)
      })
    })
  })
})