import assert from 'assert'; import { TRANSACTION_STATUSES } from '../../../shared/constants/transaction'; import PersonalMessageManager from './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 () { const result = messageManager.messages; assert.ok(Array.isArray(result)); assert.equal(result.length, 0); }); }); describe('#addMsg', function () { it('adds a Msg returned in getMsgList', function () { const Msg = { id: 1, status: TRANSACTION_STATUSES.APPROVED, metamaskNetworkId: 'unit test', }; messageManager.addMsg(Msg); const 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 () { const Msg = { id: 1, status: TRANSACTION_STATUSES.UNAPPROVED, metamaskNetworkId: 'unit test', }; messageManager.addMsg(Msg); messageManager.setMsgStatusApproved(1); const result = messageManager.messages; assert.ok(Array.isArray(result)); assert.equal(result.length, 1); assert.equal(result[0].status, TRANSACTION_STATUSES.APPROVED); }); }); describe('#rejectMsg', function () { it('sets the Msg status to rejected', function () { const Msg = { id: 1, status: TRANSACTION_STATUSES.UNAPPROVED, metamaskNetworkId: 'unit test', }; messageManager.addMsg(Msg); messageManager.rejectMsg(1); const result = messageManager.messages; assert.ok(Array.isArray(result)); assert.equal(result.length, 1); assert.equal(result[0].status, TRANSACTION_STATUSES.REJECTED); }); }); describe('#_updateMsg', function () { it('replaces the Msg with the same id', function () { messageManager.addMsg({ id: '1', status: TRANSACTION_STATUSES.UNAPPROVED, metamaskNetworkId: 'unit test', }); messageManager.addMsg({ id: '2', status: TRANSACTION_STATUSES.APPROVED, metamaskNetworkId: 'unit test', }); messageManager._updateMsg({ id: '1', status: 'blah', hash: 'foo', metamaskNetworkId: 'unit test', }); const 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: TRANSACTION_STATUSES.UNAPPROVED, metamaskNetworkId: 'unit test', }); messageManager.addMsg({ id: '2', status: TRANSACTION_STATUSES.APPROVED, metamaskNetworkId: 'unit test', }); const result = messageManager.getUnapprovedMsgs(); assert.equal(typeof result, 'object'); assert.equal(result['1'].status, TRANSACTION_STATUSES.UNAPPROVED); assert.equal(result['2'], undefined); }); }); describe('#getMsg', function () { it('returns a Msg with the requested id', function () { messageManager.addMsg({ id: '1', status: TRANSACTION_STATUSES.UNAPPROVED, metamaskNetworkId: 'unit test', }); messageManager.addMsg({ id: '2', status: TRANSACTION_STATUSES.APPROVED, metamaskNetworkId: 'unit test', }); assert.equal( messageManager.getMsg('1').status, TRANSACTION_STATUSES.UNAPPROVED, ); assert.equal( messageManager.getMsg('2').status, TRANSACTION_STATUSES.APPROVED, ); }); }); describe('#normalizeMsgData', function () { it('converts text to a utf8 hex string', function () { const input = 'hello'; const output = messageManager.normalizeMsgData(input); assert.equal(output, '0x68656c6c6f', 'predictably hex encoded'); }); it('tolerates a hex prefix', function () { const input = '0x12'; const output = messageManager.normalizeMsgData(input); assert.equal(output, '0x12', 'un modified'); }); it('tolerates normal hex', function () { const input = '12'; const output = messageManager.normalizeMsgData(input); assert.equal(output, '0x12', 'adds prefix'); }); }); });