These external modules now have their own test coverage and build enforcement. This allowed me to somewhat more easily add good tests around our personalSign strategy (held now in [eth-sig-util](https://github.com/flyswatter/eth-sig-util), and allow each of the keyrings to import that, etc.feature/default_network_editable
Dan Finlay
8 years ago
parent
29f07238f4
commit
0584988688
@ -1,125 +0,0 @@ |
||||
const EventEmitter = require('events').EventEmitter |
||||
const hdkey = require('ethereumjs-wallet/hdkey') |
||||
const bip39 = require('bip39') |
||||
const ethUtil = require('ethereumjs-util') |
||||
|
||||
// *Internal Deps
|
||||
const sigUtil = require('../lib/sig-util') |
||||
|
||||
// Options:
|
||||
const hdPathString = `m/44'/60'/0'/0` |
||||
const type = 'HD Key Tree' |
||||
|
||||
class HdKeyring extends EventEmitter { |
||||
|
||||
/* PUBLIC METHODS */ |
||||
|
||||
constructor (opts = {}) { |
||||
super() |
||||
this.type = type |
||||
this.deserialize(opts) |
||||
} |
||||
|
||||
serialize () { |
||||
return Promise.resolve({ |
||||
mnemonic: this.mnemonic, |
||||
numberOfAccounts: this.wallets.length, |
||||
}) |
||||
} |
||||
|
||||
deserialize (opts = {}) { |
||||
this.opts = opts || {} |
||||
this.wallets = [] |
||||
this.mnemonic = null |
||||
this.root = null |
||||
|
||||
if (opts.mnemonic) { |
||||
this._initFromMnemonic(opts.mnemonic) |
||||
} |
||||
|
||||
if (opts.numberOfAccounts) { |
||||
return this.addAccounts(opts.numberOfAccounts) |
||||
} |
||||
|
||||
return Promise.resolve([]) |
||||
} |
||||
|
||||
addAccounts (numberOfAccounts = 1) { |
||||
if (!this.root) { |
||||
this._initFromMnemonic(bip39.generateMnemonic()) |
||||
} |
||||
|
||||
const oldLen = this.wallets.length |
||||
const newWallets = [] |
||||
for (let i = oldLen; i < numberOfAccounts + oldLen; i++) { |
||||
const child = this.root.deriveChild(i) |
||||
const wallet = child.getWallet() |
||||
newWallets.push(wallet) |
||||
this.wallets.push(wallet) |
||||
} |
||||
const hexWallets = newWallets.map(w => w.getAddress().toString('hex')) |
||||
return Promise.resolve(hexWallets) |
||||
} |
||||
|
||||
getAccounts () { |
||||
return Promise.resolve(this.wallets.map(w => w.getAddress().toString('hex'))) |
||||
} |
||||
|
||||
// tx is an instance of the ethereumjs-transaction class.
|
||||
signTransaction (address, tx) { |
||||
const wallet = this._getWalletForAccount(address) |
||||
var privKey = wallet.getPrivateKey() |
||||
tx.sign(privKey) |
||||
return Promise.resolve(tx) |
||||
} |
||||
|
||||
// For eth_sign, we need to sign transactions:
|
||||
// hd
|
||||
signMessage (withAccount, data) { |
||||
const wallet = this._getWalletForAccount(withAccount) |
||||
const message = ethUtil.stripHexPrefix(data) |
||||
var privKey = wallet.getPrivateKey() |
||||
var msgSig = ethUtil.ecsign(new Buffer(message, 'hex'), privKey) |
||||
var rawMsgSig = ethUtil.bufferToHex(sigUtil.concatSig(msgSig.v, msgSig.r, msgSig.s)) |
||||
return Promise.resolve(rawMsgSig) |
||||
} |
||||
|
||||
// For eth_sign, we need to sign transactions:
|
||||
newGethSignMessage (withAccount, msgHex) { |
||||
const wallet = this._getWalletForAccount(withAccount) |
||||
const privKey = wallet.getPrivateKey() |
||||
const msgBuffer = ethUtil.toBuffer(msgHex) |
||||
const msgHash = ethUtil.hashPersonalMessage(msgBuffer) |
||||
const msgSig = ethUtil.ecsign(msgHash, privKey) |
||||
const rawMsgSig = ethUtil.bufferToHex(sigUtil.concatSig(msgSig.v, msgSig.r, msgSig.s)) |
||||
return Promise.resolve(rawMsgSig) |
||||
} |
||||
|
||||
exportAccount (address) { |
||||
const wallet = this._getWalletForAccount(address) |
||||
return Promise.resolve(wallet.getPrivateKey().toString('hex')) |
||||
} |
||||
|
||||
|
||||
/* PRIVATE METHODS */ |
||||
|
||||
_initFromMnemonic (mnemonic) { |
||||
this.mnemonic = mnemonic |
||||
const seed = bip39.mnemonicToSeed(mnemonic) |
||||
this.hdWallet = hdkey.fromMasterSeed(seed) |
||||
this.root = this.hdWallet.derivePath(hdPathString) |
||||
} |
||||
|
||||
|
||||
_getWalletForAccount (account) { |
||||
const targetAddress = sigUtil.normalize(account) |
||||
return this.wallets.find((w) => { |
||||
const address = w.getAddress().toString('hex') |
||||
return ((address === targetAddress) || |
||||
(sigUtil.normalize(address) === targetAddress)) |
||||
}) |
||||
} |
||||
} |
||||
|
||||
HdKeyring.type = type |
||||
module.exports = HdKeyring |
||||
@ -1,100 +0,0 @@ |
||||
const EventEmitter = require('events').EventEmitter |
||||
const Wallet = require('ethereumjs-wallet') |
||||
const ethUtil = require('ethereumjs-util') |
||||
const type = 'Simple Key Pair' |
||||
const sigUtil = require('../lib/sig-util') |
||||
|
||||
class SimpleKeyring extends EventEmitter { |
||||
|
||||
/* PUBLIC METHODS */ |
||||
|
||||
constructor (opts) { |
||||
super() |
||||
this.type = type |
||||
this.opts = opts || {} |
||||
this.wallets = [] |
||||
} |
||||
|
||||
serialize () { |
||||
return Promise.resolve(this.wallets.map(w => w.getPrivateKey().toString('hex'))) |
||||
} |
||||
|
||||
deserialize (privateKeys = []) { |
||||
return new Promise((resolve, reject) => { |
||||
try { |
||||
this.wallets = privateKeys.map((privateKey) => { |
||||
const stripped = ethUtil.stripHexPrefix(privateKey) |
||||
const buffer = new Buffer(stripped, 'hex') |
||||
const wallet = Wallet.fromPrivateKey(buffer) |
||||
return wallet |
||||
}) |
||||
} catch (e) { |
||||
reject(e) |
||||
} |
||||
resolve() |
||||
}) |
||||
} |
||||
|
||||
addAccounts (n = 1) { |
||||
var newWallets = [] |
||||
for (var i = 0; i < n; i++) { |
||||
newWallets.push(Wallet.generate()) |
||||
} |
||||
this.wallets = this.wallets.concat(newWallets) |
||||
const hexWallets = newWallets.map(w => ethUtil.bufferToHex(w.getAddress())) |
||||
return Promise.resolve(hexWallets) |
||||
} |
||||
|
||||
getAccounts () { |
||||
return Promise.resolve(this.wallets.map(w => ethUtil.bufferToHex(w.getAddress()))) |
||||
} |
||||
|
||||
// tx is an instance of the ethereumjs-transaction class.
|
||||
signTransaction (address, tx) { |
||||
const wallet = this._getWalletForAccount(address) |
||||
var privKey = wallet.getPrivateKey() |
||||
tx.sign(privKey) |
||||
return Promise.resolve(tx) |
||||
} |
||||
|
||||
// For eth_sign, we need to sign transactions:
|
||||
signMessage (withAccount, data) { |
||||
const wallet = this._getWalletForAccount(withAccount) |
||||
const message = ethUtil.stripHexPrefix(data) |
||||
var privKey = wallet.getPrivateKey() |
||||
var msgSig = ethUtil.ecsign(new Buffer(message, 'hex'), privKey) |
||||
var rawMsgSig = ethUtil.bufferToHex(sigUtil.concatSig(msgSig.v, msgSig.r, msgSig.s)) |
||||
return Promise.resolve(rawMsgSig) |
||||
} |
||||
|
||||
// For eth_sign, we need to sign transactions:
|
||||
|
||||
newGethSignMessage (withAccount, msgHex) { |
||||
const wallet = this._getWalletForAccount(withAccount) |
||||
const privKey = wallet.getPrivateKey() |
||||
const msgBuffer = ethUtil.toBuffer(msgHex) |
||||
const msgHash = ethUtil.hashPersonalMessage(msgBuffer) |
||||
const msgSig = ethUtil.ecsign(msgHash, privKey) |
||||
const rawMsgSig = ethUtil.bufferToHex(sigUtil.concatSig(msgSig.v, msgSig.r, msgSig.s)) |
||||
return Promise.resolve(rawMsgSig) |
||||
} |
||||
|
||||
exportAccount (address) { |
||||
const wallet = this._getWalletForAccount(address) |
||||
return Promise.resolve(wallet.getPrivateKey().toString('hex')) |
||||
} |
||||
|
||||
|
||||
/* PRIVATE METHODS */ |
||||
|
||||
_getWalletForAccount (account) { |
||||
const address = sigUtil.normalize(account) |
||||
let wallet = this.wallets.find(w => ethUtil.bufferToHex(w.getAddress()) === address) |
||||
if (!wallet) throw new Error('Simple Keyring - Unable to find matching address.') |
||||
return wallet |
||||
} |
||||
|
||||
} |
||||
|
||||
SimpleKeyring.type = type |
||||
module.exports = SimpleKeyring |
||||
@ -0,0 +1,118 @@ |
||||
const EventEmitter = require('events') |
||||
const ObservableStore = require('obs-store') |
||||
const ethUtil = require('ethereumjs-util') |
||||
const createId = require('./random-id') |
||||
|
||||
|
||||
module.exports = class MessageManager extends EventEmitter{ |
||||
constructor (opts) { |
||||
super() |
||||
this.memStore = new ObservableStore({ |
||||
unapprovedPersonalMsgs: {}, |
||||
unapprovedPersonalMsgCount: 0, |
||||
}) |
||||
this.messages = [] |
||||
} |
||||
|
||||
get unapprovedPersonalMsgCount () { |
||||
return Object.keys(this.getUnapprovedMsgs()).length |
||||
} |
||||
|
||||
getUnapprovedMsgs () { |
||||
return this.messages.filter(msg => msg.status === 'unapproved') |
||||
.reduce((result, msg) => { result[msg.id] = msg; return result }, {}) |
||||
} |
||||
|
||||
addUnapprovedMessage (msgParams) { |
||||
msgParams.data = normalizeMsgData(msgParams.data) |
||||
// create txData obj with parameters and meta data
|
||||
var time = (new Date()).getTime() |
||||
var msgId = createId() |
||||
var msgData = { |
||||
id: msgId, |
||||
msgParams: msgParams, |
||||
time: time, |
||||
status: 'unapproved', |
||||
} |
||||
this.addMsg(msgData) |
||||
|
||||
// signal update
|
||||
this.emit('update') |
||||
return msgId |
||||
} |
||||
|
||||
addMsg (msg) { |
||||
this.messages.push(msg) |
||||
this._saveMsgList() |
||||
} |
||||
|
||||
getMsg (msgId) { |
||||
return this.messages.find(msg => msg.id === msgId) |
||||
} |
||||
|
||||
approveMessage (msgParams) { |
||||
this.setMsgStatusApproved(msgParams.metamaskId) |
||||
return this.prepMsgForSigning(msgParams) |
||||
} |
||||
|
||||
setMsgStatusApproved (msgId) { |
||||
this._setMsgStatus(msgId, 'approved') |
||||
} |
||||
|
||||
setMsgStatusSigned (msgId, rawSig) { |
||||
const msg = this.getMsg(msgId) |
||||
msg.rawSig = rawSig |
||||
this._updateMsg(msg) |
||||
this._setMsgStatus(msgId, 'signed') |
||||
} |
||||
|
||||
prepMsgForSigning (msgParams) { |
||||
delete msgParams.metamaskId |
||||
return Promise.resolve(msgParams) |
||||
} |
||||
|
||||
rejectMsg (msgId) { |
||||
this._setMsgStatus(msgId, 'rejected') |
||||
} |
||||
|
||||
//
|
||||
// PRIVATE METHODS
|
||||
//
|
||||
|
||||
_setMsgStatus (msgId, status) { |
||||
const msg = this.getMsg(msgId) |
||||
if (!msg) throw new Error('MessageManager - Message not found for id: "${msgId}".') |
||||
msg.status = status |
||||
this._updateMsg(msg) |
||||
this.emit(`${msgId}:${status}`, msg) |
||||
if (status === 'rejected' || status === 'signed') { |
||||
this.emit(`${msgId}:finished`, msg) |
||||
} |
||||
} |
||||
|
||||
_updateMsg (msg) { |
||||
const index = this.messages.findIndex((message) => message.id === msg.id) |
||||
if (index !== -1) { |
||||
this.messages[index] = msg |
||||
} |
||||
this._saveMsgList() |
||||
} |
||||
|
||||
_saveMsgList () { |
||||
const unapprovedPersonalMsgs = this.getUnapprovedMsgs() |
||||
const unapprovedPersonalMsgCount = Object.keys(unapprovedPersonalMsgs).length |
||||
this.memStore.updateState({ unapprovedPersonalMsgs, unapprovedPersonalMsgCount }) |
||||
this.emit('updateBadge') |
||||
} |
||||
|
||||
} |
||||
|
||||
function normalizeMsgData(data) { |
||||
if (data.slice(0, 2) === '0x') { |
||||
// data is already hex
|
||||
return data |
||||
} else { |
||||
// data is unicode, convert to hex
|
||||
return ethUtil.bufferToHex(new Buffer(data, 'utf8')) |
||||
} |
||||
} |
||||
@ -1,28 +0,0 @@ |
||||
const ethUtil = require('ethereumjs-util') |
||||
|
||||
module.exports = { |
||||
|
||||
concatSig: function (v, r, s) { |
||||
const rSig = ethUtil.fromSigned(r) |
||||
const sSig = ethUtil.fromSigned(s) |
||||
const vSig = ethUtil.bufferToInt(v) |
||||
const rStr = padWithZeroes(ethUtil.toUnsigned(rSig).toString('hex'), 64) |
||||
const sStr = padWithZeroes(ethUtil.toUnsigned(sSig).toString('hex'), 64) |
||||
const vStr = ethUtil.stripHexPrefix(ethUtil.intToHex(vSig)) |
||||
return ethUtil.addHexPrefix(rStr.concat(sStr, vStr)).toString('hex') |
||||
}, |
||||
|
||||
normalize: function (address) { |
||||
if (!address) return |
||||
return ethUtil.addHexPrefix(address.toLowerCase()) |
||||
}, |
||||
|
||||
} |
||||
|
||||
function padWithZeroes (number, length) { |
||||
var myString = '' + number |
||||
while (myString.length < length) { |
||||
myString = '0' + myString |
||||
} |
||||
return myString |
||||
} |
||||
@ -1,127 +0,0 @@ |
||||
const assert = require('assert') |
||||
const extend = require('xtend') |
||||
const HdKeyring = require('../../../app/scripts/keyrings/hd') |
||||
|
||||
// Sample account:
|
||||
const privKeyHex = 'b8a9c05beeedb25df85f8d641538cbffedf67216048de9c678ee26260eb91952' |
||||
|
||||
const sampleMnemonic = 'finish oppose decorate face calm tragic certain desk hour urge dinosaur mango' |
||||
const firstAcct = '1c96099350f13d558464ec79b9be4445aa0ef579' |
||||
const secondAcct = '1b00aed43a693f3a957f9feb5cc08afa031e37a0' |
||||
|
||||
describe('hd-keyring', function() { |
||||
|
||||
let keyring |
||||
beforeEach(function() { |
||||
keyring = new HdKeyring() |
||||
}) |
||||
|
||||
describe('constructor', function(done) { |
||||
keyring = new HdKeyring({ |
||||
mnemonic: sampleMnemonic, |
||||
numberOfAccounts: 2, |
||||
}) |
||||
|
||||
const accounts = keyring.getAccounts() |
||||
.then((accounts) => { |
||||
assert.equal(accounts[0], firstAcct) |
||||
assert.equal(accounts[1], secondAcct) |
||||
done() |
||||
}) |
||||
}) |
||||
|
||||
describe('Keyring.type', function() { |
||||
it('is a class property that returns the type string.', function() { |
||||
const type = HdKeyring.type |
||||
assert.equal(typeof type, 'string') |
||||
}) |
||||
}) |
||||
|
||||
describe('#type', function() { |
||||
it('returns the correct value', function() { |
||||
const type = keyring.type |
||||
const correct = HdKeyring.type |
||||
assert.equal(type, correct) |
||||
}) |
||||
}) |
||||
|
||||
describe('#serialize empty wallets.', function() { |
||||
it('serializes a new mnemonic', function() { |
||||
keyring.serialize() |
||||
.then((output) => { |
||||
assert.equal(output.numberOfAccounts, 0) |
||||
assert.equal(output.mnemonic, null) |
||||
}) |
||||
}) |
||||
}) |
||||
|
||||
describe('#deserialize a private key', function() { |
||||
it('serializes what it deserializes', function(done) { |
||||
keyring.deserialize({ |
||||
mnemonic: sampleMnemonic, |
||||
numberOfAccounts: 1 |
||||
}) |
||||
.then(() => { |
||||
assert.equal(keyring.wallets.length, 1, 'restores two accounts') |
||||
return keyring.addAccounts(1) |
||||
}).then(() => { |
||||
return keyring.getAccounts() |
||||
}).then((accounts) => { |
||||
assert.equal(accounts[0], firstAcct) |
||||
assert.equal(accounts[1], secondAcct) |
||||
assert.equal(accounts.length, 2) |
||||
|
||||
return keyring.serialize() |
||||
}).then((serialized) => { |
||||
assert.equal(serialized.mnemonic, sampleMnemonic) |
||||
done() |
||||
}) |
||||
}) |
||||
}) |
||||
|
||||
describe('#addAccounts', function() { |
||||
describe('with no arguments', function() { |
||||
it('creates a single wallet', function(done) { |
||||
keyring.addAccounts() |
||||
.then(() => { |
||||
assert.equal(keyring.wallets.length, 1) |
||||
done() |
||||
}) |
||||
}) |
||||
}) |
||||
|
||||
describe('with a numeric argument', function() { |
||||
it('creates that number of wallets', function(done) { |
||||
keyring.addAccounts(3) |
||||
.then(() => { |
||||
assert.equal(keyring.wallets.length, 3) |
||||
done() |
||||
}) |
||||
}) |
||||
}) |
||||
}) |
||||
|
||||
describe('#getAccounts', function() { |
||||
it('calls getAddress on each wallet', function(done) { |
||||
|
||||
// Push a mock wallet
|
||||
const desiredOutput = 'foo' |
||||
keyring.wallets.push({ |
||||
getAddress() { |
||||
return { |
||||
toString() { |
||||
return desiredOutput |
||||
} |
||||
} |
||||
} |
||||
}) |
||||
|
||||
const output = keyring.getAccounts() |
||||
.then((output) => { |
||||
assert.equal(output[0], desiredOutput) |
||||
assert.equal(output.length, 1) |
||||
done() |
||||
}) |
||||
}) |
||||
}) |
||||
}) |
||||
@ -1,149 +0,0 @@ |
||||
const assert = require('assert') |
||||
const extend = require('xtend') |
||||
const Web3 = require('web3') |
||||
const web3 = new Web3() |
||||
const ethUtil = require('ethereumjs-util') |
||||
const SimpleKeyring = require('../../../app/scripts/keyrings/simple') |
||||
const TYPE_STR = 'Simple Key Pair' |
||||
|
||||
// Sample account:
|
||||
const privKeyHex = 'b8a9c05beeedb25df85f8d641538cbffedf67216048de9c678ee26260eb91952' |
||||
|
||||
describe('simple-keyring', function() { |
||||
|
||||
let keyring |
||||
beforeEach(function() { |
||||
keyring = new SimpleKeyring() |
||||
}) |
||||
|
||||
describe('Keyring.type', function() { |
||||
it('is a class property that returns the type string.', function() { |
||||
const type = SimpleKeyring.type |
||||
assert.equal(type, TYPE_STR) |
||||
}) |
||||
}) |
||||
|
||||
describe('#type', function() { |
||||
it('returns the correct value', function() { |
||||
const type = keyring.type |
||||
assert.equal(type, TYPE_STR) |
||||
}) |
||||
}) |
||||
|
||||
describe('#serialize empty wallets.', function() { |
||||
it('serializes an empty array', function(done) { |
||||
keyring.serialize() |
||||
.then((output) => { |
||||
assert.deepEqual(output, []) |
||||
done() |
||||
}) |
||||
}) |
||||
}) |
||||
|
||||
describe('#deserialize a private key', function() { |
||||
it('serializes what it deserializes', function() { |
||||
keyring.deserialize([privKeyHex]) |
||||
.then(() => { |
||||
assert.equal(keyring.wallets.length, 1, 'has one wallet') |
||||
const serialized = keyring.serialize() |
||||
assert.equal(serialized[0], privKeyHex) |
||||
}) |
||||
}) |
||||
}) |
||||
|
||||
describe('#signMessage', function() { |
||||
const address = '0x9858e7d8b79fc3e6d989636721584498926da38a' |
||||
const message = '0x879a053d4800c6354e76c7985a865d2922c82fb5b3f4577b2fe08b998954f2e0' |
||||
const privateKey = '0x7dd98753d7b4394095de7d176c58128e2ed6ee600abe97c9f6d9fd65015d9b18' |
||||
const expectedResult = '0x28fcb6768e5110144a55b2e6ce9d1ea5a58103033632d272d2b5cf506906f7941a00b539383fd872109633d8c71c404e13dba87bc84166ee31b0e36061a69e161c' |
||||
|
||||
it('passes the dennis test', function(done) { |
||||
keyring.deserialize([ privateKey ]) |
||||
.then(() => { |
||||
return keyring.signMessage(address, message) |
||||
}) |
||||
.then((result) => { |
||||
assert.equal(result, expectedResult) |
||||
done() |
||||
}) |
||||
}) |
||||
|
||||
it('reliably can decode messages it signs', function (done) { |
||||
|
||||
const message = 'hello there!' |
||||
const msgHashHex = web3.sha3(message) |
||||
let address |
||||
let addresses = [] |
||||
|
||||
keyring.deserialize([ privateKey ]) |
||||
.then(() => { |
||||
keyring.addAccounts(9) |
||||
}) |
||||
.then(() => { |
||||
return keyring.getAccounts() |
||||
}) |
||||
.then((addrs) => { |
||||
addresses = addrs |
||||
return Promise.all(addresses.map((address) => { |
||||
return keyring.signMessage(address, msgHashHex) |
||||
})) |
||||
}) |
||||
.then((signatures) => { |
||||
|
||||
signatures.forEach((sgn, index) => { |
||||
const address = addresses[index] |
||||
|
||||
var r = ethUtil.toBuffer(sgn.slice(0,66)) |
||||
var s = ethUtil.toBuffer('0x' + sgn.slice(66,130)) |
||||
var v = ethUtil.bufferToInt(ethUtil.toBuffer('0x' + sgn.slice(130,132))) |
||||
var m = ethUtil.toBuffer(msgHashHex) |
||||
var pub = ethUtil.ecrecover(m, v, r, s) |
||||
var adr = '0x' + ethUtil.pubToAddress(pub).toString('hex') |
||||
|
||||
assert.equal(adr, address, 'recovers address from signature correctly') |
||||
}) |
||||
done() |
||||
}) |
||||
}) |
||||
}) |
||||
|
||||
describe('#addAccounts', function() { |
||||
describe('with no arguments', function() { |
||||
it('creates a single wallet', function() { |
||||
keyring.addAccounts() |
||||
.then(() => { |
||||
assert.equal(keyring.wallets.length, 1) |
||||
}) |
||||
}) |
||||
}) |
||||
|
||||
describe('with a numeric argument', function() { |
||||
it('creates that number of wallets', function() { |
||||
keyring.addAccounts(3) |
||||
.then(() => { |
||||
assert.equal(keyring.wallets.length, 3) |
||||
}) |
||||
}) |
||||
}) |
||||
}) |
||||
|
||||
describe('#getAccounts', function() { |
||||
it('calls getAddress on each wallet', function(done) { |
||||
|
||||
// Push a mock wallet
|
||||
const desiredOutput = '0x18a3462427bcc9133bb46e88bcbe39cd7ef0e761' |
||||
keyring.wallets.push({ |
||||
getAddress() { |
||||
return ethUtil.toBuffer(desiredOutput) |
||||
} |
||||
}) |
||||
|
||||
keyring.getAccounts() |
||||
.then((output) => { |
||||
assert.equal(output[0], desiredOutput) |
||||
assert.equal(output.length, 1) |
||||
done() |
||||
}) |
||||
}) |
||||
}) |
||||
}) |
||||
@ -0,0 +1,89 @@ |
||||
const assert = require('assert') |
||||
const extend = require('xtend') |
||||
const EventEmitter = require('events') |
||||
|
||||
const PersonalMessageManager = require('../../app/scripts/lib/personal-message-manager') |
||||
|
||||
describe('Transaction 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') |
||||
}) |
||||
}) |
||||
}) |
||||
Loading…
Reference in new issue