Files
openzeppelin-contracts/test/helpers/expectEvent.js
2018-11-22 16:54:58 -03:00

52 lines
1.3 KiB
JavaScript

const { decodeLogs } = require('./decodeLogs');
const BigNumber = web3.BigNumber;
const should = require('chai')
.use(require('chai-bignumber')(BigNumber))
.should();
function inLogs (logs, eventName, eventArgs = {}) {
const event = logs.find(function (e) {
if (e.event === eventName) {
for (const [k, v] of Object.entries(eventArgs)) {
contains(e.args, k, v);
}
return true;
}
});
should.exist(event);
return event;
}
async function inTransaction (tx, eventName, eventArgs = {}) {
const { logs } = await tx;
return inLogs(logs, eventName, eventArgs);
}
async function inConstruction (contract, eventName, eventArgs = {}) {
const receipt = await web3.eth.getTransactionReceipt(contract.transactionHash);
const logs = decodeLogs(receipt.logs, contract.constructor.events, contract.address);
return inLogs(logs, eventName, eventArgs);
}
function contains (args, key, value) {
if (isBigNumber(args[key])) {
args[key].should.be.bignumber.equal(value);
} else {
args[key].should.be.equal(value);
}
}
function isBigNumber (object) {
return object.isBigNumber ||
object instanceof BigNumber ||
(object.constructor && object.constructor.name === 'BigNumber');
}
module.exports = {
inLogs,
inTransaction,
inConstruction,
};