metamask-extension/test/unit/personal-message-manager-test.js

const assert = require('assert')
const extend = require('xtend')
const EventEmitter = require('events')

const PersonalMessageManager = require('../../app/scripts/lib/personal-message-manager')

describe('Personal Message Manager', function() {
  let messageManager

  beforeEach(function() {
    messageManager = new PersonalMessageManager()
  })

  describe('#getMsgList', function() {
    it('when new should return empty array', function() {
      var result = messageManager.messages
      assert.ok(Array.isArray(result))
      assert.equal(result.length, 0)
    })
    it('should also return transactions from local storage if any', function() {

    })
  })

  describe('#addMsg', function() {
    it('adds a Msg returned in getMsgList', function() {
      var Msg = { id: 1, status: 'approved', metamaskNetworkId: 'unit test' }
      messageManager.addMsg(Msg)
      var result = messageManager.messages
      assert.ok(Array.isArray(result))
      assert.equal(result.length, 1)
      assert.equal(result[0].id, 1)
    })
  })

  describe('#setMsgStatusApproved', function() {
    it('sets the Msg status to approved', function() {
      var Msg = { id: 1, status: 'unapproved', metamaskNetworkId: 'unit test' }
      messageManager.addMsg(Msg)
      messageManager.setMsgStatusApproved(1)
      var result = messageManager.messages
      assert.ok(Array.isArray(result))
      assert.equal(result.length, 1)
      assert.equal(result[0].status, 'approved')
    })
  })

  describe('#rejectMsg', function() {
    it('sets the Msg status to rejected', function() {
      var Msg = { id: 1, status: 'unapproved', metamaskNetworkId: 'unit test' }
      messageManager.addMsg(Msg)
      messageManager.rejectMsg(1)
      var result = messageManager.messages
      assert.ok(Array.isArray(result))
      assert.equal(result.length, 1)
      assert.equal(result[0].status, 'rejected')
    })
  })

  describe('#_updateMsg', function() {
    it('replaces the Msg with the same id', function() {
      messageManager.addMsg({ id: '1', status: 'unapproved', metamaskNetworkId: 'unit test' })
      messageManager.addMsg({ id: '2', status: 'approved', metamaskNetworkId: 'unit test' })
      messageManager._updateMsg({ id: '1', status: 'blah', hash: 'foo', metamaskNetworkId: 'unit test' })
      var result = messageManager.getMsg('1')
      assert.equal(result.hash, 'foo')
    })
  })

  describe('#getUnapprovedMsgs', function() {
    it('returns unapproved Msgs in a hash', function() {
      messageManager.addMsg({ id: '1', status: 'unapproved', metamaskNetworkId: 'unit test' })
      messageManager.addMsg({ id: '2', status: 'approved', metamaskNetworkId: 'unit test' })
      let result = messageManager.getUnapprovedMsgs()
      assert.equal(typeof result, 'object')
      assert.equal(result['1'].status, 'unapproved')
      assert.equal(result['2'], undefined)
    })
  })

  describe('#getMsg', function() {
    it('returns a Msg with the requested id', function() {
      messageManager.addMsg({ id: '1', status: 'unapproved', metamaskNetworkId: 'unit test' })
      messageManager.addMsg({ id: '2', status: 'approved', metamaskNetworkId: 'unit test' })
      assert.equal(messageManager.getMsg('1').status, 'unapproved')
      assert.equal(messageManager.getMsg('2').status, 'approved')
    })
  })

  describe('#normalizeMsgData', function() {
    it('converts text to a utf8 buffer', function() {
      var input = 'hello'
      var output = messageManager.normalizeMsgData(input)
      assert.equal(output, '0x68656c6c6f', 'predictably hex encoded')
    })

    it('tolerates a hex prefix', function() {
      var input = '0x12'
      var output = messageManager.normalizeMsgData(input)
      assert.equal(output, '0x12', 'un modified')
    })

    it('tolerates normal hex', function() {
      var input = '12'
      var output = messageManager.normalizeMsgData(input)
      assert.equal(output, '0x12', 'adds prefix')
    })
  })

})