metamask-extension/test/unit/app/util-test.js

const assert = require('assert')
const { getEnvironmentType, sufficientBalance } = require('../../../app/scripts/lib/util')
const {
  ENVIRONMENT_TYPE_POPUP,
  ENVIRONMENT_TYPE_NOTIFICATION,
  ENVIRONMENT_TYPE_FULLSCREEN,
  ENVIRONMENT_TYPE_BACKGROUND,
} = require('../../../app/scripts/lib/enums')

describe('getEnvironmentType', function () {
  it('should return popup type', function () {
    const environmentType = getEnvironmentType('http://extension-id/popup.html')
    assert.equal(environmentType, ENVIRONMENT_TYPE_POPUP)
  })

  it('should return notification type', function () {
    const environmentType = getEnvironmentType('http://extension-id/notification.html')
    assert.equal(environmentType, ENVIRONMENT_TYPE_NOTIFICATION)
  })

  it('should return fullscreen type for home.html', function () {
    const environmentType = getEnvironmentType('http://extension-id/home.html')
    assert.equal(environmentType, ENVIRONMENT_TYPE_FULLSCREEN)
  })

  it('should return fullscreen type for phishing.html', function () {
    const environmentType = getEnvironmentType('http://extension-id/phishing.html')
    assert.equal(environmentType, ENVIRONMENT_TYPE_FULLSCREEN)
  })

  it('should return background type', function () {
    const environmentType = getEnvironmentType('http://extension-id/_generated_background_page.html')
    assert.equal(environmentType, ENVIRONMENT_TYPE_BACKGROUND)
  })

  it('should return the correct type for a URL with a hash fragment', function () {
    const environmentType = getEnvironmentType('http://extension-id/popup.html#hash')
    assert.equal(environmentType, ENVIRONMENT_TYPE_POPUP)
  })

  it('should return the correct type for a URL with query parameters', function () {
    const environmentType = getEnvironmentType('http://extension-id/popup.html?param=foo')
    assert.equal(environmentType, ENVIRONMENT_TYPE_POPUP)
  })

  it('should return the correct type for a URL with query parameters and a hash fragment', function () {
    const environmentType = getEnvironmentType('http://extension-id/popup.html?param=foo#hash')
    assert.equal(environmentType, ENVIRONMENT_TYPE_POPUP)
  })
})

describe('SufficientBalance', function () {
  it('returns true if max tx cost is equal to balance.', function () {
    const tx = {
      'value': '0x1',
      'gas': '0x2',
      'gasPrice': '0x3',
    }
    const balance = '0x8'

    const result = sufficientBalance(tx, balance)
    assert.ok(result, 'sufficient balance found.')
  })

  it('returns true if max tx cost is less than balance.', function () {
    const tx = {
      'value': '0x1',
      'gas': '0x2',
      'gasPrice': '0x3',
    }
    const balance = '0x9'

    const result = sufficientBalance(tx, balance)
    assert.ok(result, 'sufficient balance found.')
  })

  it('returns false if max tx cost is more than balance.', function () {
    const tx = {
      'value': '0x1',
      'gas': '0x2',
      'gasPrice': '0x3',
    }
    const balance = '0x6'

    const result = sufficientBalance(tx, balance)
    assert.ok(!result, 'insufficient balance found.')
  })
})