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

import { strict as assert } from 'assert'
import {
  getEnvironmentType,
  sufficientBalance,
  isPrefixedFormattedHexString,
} from '../../../app/scripts/lib/util'

import {
  ENVIRONMENT_TYPE_POPUP,
  ENVIRONMENT_TYPE_NOTIFICATION,
  ENVIRONMENT_TYPE_FULLSCREEN,
  ENVIRONMENT_TYPE_BACKGROUND,
} from '../../../app/scripts/lib/enums'

describe('app utils', function () {
  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.')
    })
  })

  describe('isPrefixedFormattedHexString', function () {
    it('should return true for valid hex strings', function () {
      assert.equal(
        isPrefixedFormattedHexString('0x1'),
        true,
        'should return true',
      )

      assert.equal(
        isPrefixedFormattedHexString('0xa'),
        true,
        'should return true',
      )

      assert.equal(
        isPrefixedFormattedHexString('0xabcd1123fae909aad87452'),
        true,
        'should return true',
      )
    })

    it('should return false for invalid hex strings', function () {
      assert.equal(
        isPrefixedFormattedHexString('0x'),
        false,
        'should return false',
      )

      assert.equal(
        isPrefixedFormattedHexString('0x0'),
        false,
        'should return false',
      )

      assert.equal(
        isPrefixedFormattedHexString('0x01'),
        false,
        'should return false',
      )

      assert.equal(
        isPrefixedFormattedHexString(' 0x1'),
        false,
        'should return false',
      )

      assert.equal(
        isPrefixedFormattedHexString('0x1 '),
        false,
        'should return false',
      )

      assert.equal(
        isPrefixedFormattedHexString('0x1afz'),
        false,
        'should return false',
      )

      assert.equal(
        isPrefixedFormattedHexString('z'),
        false,
        'should return false',
      )

      assert.equal(
        isPrefixedFormattedHexString(2),
        false,
        'should return false',
      )

      assert.equal(
        isPrefixedFormattedHexString(['0x1']),
        false,
        'should return false',
      )

      assert.equal(isPrefixedFormattedHexString(), false, 'should return false')
    })
  })
})