metamask-extension/test/unit/app/controllers/permissions/restricted-methods-test.js

import { strict as assert } from 'assert'
import pify from 'pify'

import getRestrictedMethods from '../../../../../app/scripts/controllers/permissions/restrictedMethods'

describe('restricted methods', function () {
  describe('eth_accounts', function () {
    it('should handle getKeyringAccounts error', async function () {
      const restrictedMethods = getRestrictedMethods({
        getKeyringAccounts: async () => {
          throw new Error('foo')
        },
      })
      const ethAccountsMethod = pify(restrictedMethods.eth_accounts.method)

      const res = {}
      const fooError = new Error('foo')
      await assert.rejects(
        ethAccountsMethod(null, res, null),
        fooError,
        'Should reject with expected error',
      )

      assert.deepEqual(
        res,
        { error: fooError },
        'response should have expected error and no result',
      )
    })

    it('should handle missing identity for first account when sorting', async function () {
      const restrictedMethods = getRestrictedMethods({
        getIdentities: () => {
          return { '0x7e57e2': {} }
        },
        getKeyringAccounts: async () => ['0x7e57e2', '0x7e57e3'],
      })
      const ethAccountsMethod = pify(restrictedMethods.eth_accounts.method)

      const res = {}
      await assert.rejects(ethAccountsMethod(null, res, null))
      assert.ok(res.error instanceof Error, 'result should have error')
      assert.deepEqual(
        Object.keys(res),
        ['error'],
        'result should only contain error',
      )
    })

    it('should handle missing identity for second account when sorting', async function () {
      const restrictedMethods = getRestrictedMethods({
        getIdentities: () => {
          return { '0x7e57e3': {} }
        },
        getKeyringAccounts: async () => ['0x7e57e2', '0x7e57e3'],
      })
      const ethAccountsMethod = pify(restrictedMethods.eth_accounts.method)

      const res = {}
      await assert.rejects(ethAccountsMethod(null, res, null))
      assert.ok(res.error instanceof Error, 'result should have error')
      assert.deepEqual(
        Object.keys(res),
        ['error'],
        'result should only contain error',
      )
    })

    it('should return accounts in keyring order when none are selected', async function () {
      const keyringAccounts = ['0x7e57e2', '0x7e57e3', '0x7e57e4', '0x7e57e5']
      const restrictedMethods = getRestrictedMethods({
        getIdentities: () => {
          return keyringAccounts.reduce((identities, address) => {
            identities[address] = {}
            return identities
          }, {})
        },
        getKeyringAccounts: async () => [...keyringAccounts],
      })
      const ethAccountsMethod = pify(restrictedMethods.eth_accounts.method)

      const res = {}
      await ethAccountsMethod(null, res, null)
      assert.deepEqual(
        res,
        { result: keyringAccounts },
        'should return accounts in correct order',
      )
    })

    it('should return accounts in keyring order when all have same last selected time', async function () {
      const keyringAccounts = ['0x7e57e2', '0x7e57e3', '0x7e57e4', '0x7e57e5']
      const restrictedMethods = getRestrictedMethods({
        getIdentities: () => {
          return keyringAccounts.reduce((identities, address) => {
            identities[address] = { lastSelected: 1000 }
            return identities
          }, {})
        },
        getKeyringAccounts: async () => [...keyringAccounts],
      })
      const ethAccountsMethod = pify(restrictedMethods.eth_accounts.method)

      const res = {}
      await ethAccountsMethod(null, res, null)
      assert.deepEqual(
        res,
        { result: keyringAccounts },
        'should return accounts in correct order',
      )
    })

    it('should return accounts sorted by last selected (descending)', async function () {
      const keyringAccounts = ['0x7e57e2', '0x7e57e3', '0x7e57e4', '0x7e57e5']
      const expectedResult = keyringAccounts.slice().reverse()
      const restrictedMethods = getRestrictedMethods({
        getIdentities: () => {
          return keyringAccounts.reduce((identities, address, index) => {
            identities[address] = { lastSelected: index * 1000 }
            return identities
          }, {})
        },
        getKeyringAccounts: async () => [...keyringAccounts],
      })
      const ethAccountsMethod = pify(restrictedMethods.eth_accounts.method)

      const res = {}
      await ethAccountsMethod(null, res, null)
      assert.deepEqual(
        res,
        { result: expectedResult },
        'should return accounts in correct order',
      )
    })

    it('should return accounts sorted by last selected (descending) with unselected accounts last, in keyring order', async function () {
      const keyringAccounts = [
        '0x7e57e2',
        '0x7e57e3',
        '0x7e57e4',
        '0x7e57e5',
        '0x7e57e6',
      ]
      const expectedResult = [
        '0x7e57e4',
        '0x7e57e2',
        '0x7e57e3',
        '0x7e57e5',
        '0x7e57e6',
      ]
      const restrictedMethods = getRestrictedMethods({
        getIdentities: () => {
          return {
            '0x7e57e2': { lastSelected: 1000 },
            '0x7e57e3': {},
            '0x7e57e4': { lastSelected: 2000 },
            '0x7e57e5': {},
            '0x7e57e6': {},
          }
        },
        getKeyringAccounts: async () => [...keyringAccounts],
      })
      const ethAccountsMethod = pify(restrictedMethods.eth_accounts.method)

      const res = {}
      await ethAccountsMethod(null, res, null)
      assert.deepEqual(
        res,
        { result: expectedResult },
        'should return accounts in correct order',
      )
    })
  })
})