NoBey/openzeppelin-contracts
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add findMethod function in ERC827Token test

Browse Source
This commit is contained in:
AugustoL
2018-01-12 13:08:55 -03:00
parent 7bd95b1e31
commit e911b4d556
1 changed files with 83 additions and 69 deletions
Download Patch File Download Diff File Expand all files Collapse all files

152
test/ERC827Token.js
View File

@ -4,6 +4,7 @@ var Message = artifacts.require('./mock/MessageHelper.sol');
var ERC827TokenMock = artifacts.require('./mock/ERC827TokenMock.sol'); var ERC827TokenMock = artifacts.require('./mock/ERC827TokenMock.sol');
var BigNumber = web3.BigNumber; var BigNumber = web3.BigNumber;
var _ = require('lodash');
var ethjsABI = require('ethjs-abi'); var ethjsABI = require('ethjs-abi');
require('chai') require('chai')
.use(require('chai-as-promised')) .use(require('chai-as-promised'))
@ -13,6 +14,15 @@ require('chai')
contract('ERC827 Token', function (accounts) { contract('ERC827 Token', function (accounts) {
let token; let token;
function findMethod (abi, name, args) {
for (var i = 0; i < abi.length; i++) {
const methodArgs = _.map(abi[i].inputs, 'type').join(',');
if ((abi[i].name === name) && (methodArgs === args)) {
return abi[i];
}
}
}
beforeEach(async function () { beforeEach(async function () {
token = await ERC827TokenMock.new(accounts[0], 100); token = await ERC827TokenMock.new(accounts[0], 100);
}); });
@ -111,86 +121,90 @@ contract('ERC827 Token', function (accounts) {
}); });
describe('Test ERC827 methods', function () { describe('Test ERC827 methods', function () {
it(
'should return correct balances after transfer (with data) and show the event on receiver contract'
, async function () {
const message = await Message.new();
it('should return correct balances after transfer (with data) and show the event on receiver contract', async function () { const extraData = message.contract.showMessage.getData(
const message = await Message.new(); web3.toHex(123456), 666, 'Transfer Done'
);
const abiMethod = findMethod(token.abi, 'transfer', 'address,uint256,bytes');
const transferData = ethjsABI.encodeMethod(abiMethod,
[message.contract.address, 100, extraData]
);
const transaction = await token.sendTransaction(
{ from: accounts[0], data: transferData }
);
const extraData = message.contract.showMessage.getData( assert.equal(2, transaction.receipt.logs.length);
web3.toHex(123456), 666, 'Transfer Done'
);
// Use method #8 tranfer of the abi to encode the data tx new BigNumber(100).should.be.bignumber.equal(
const transferData = ethjsABI.encodeMethod(token.abi[8], await token.balanceOf(message.contract.address)
[message.contract.address, 100, extraData] );
); });
const transaction = await token.sendTransaction(
{ from: accounts[0], data: transferData }
);
assert.equal(2, transaction.receipt.logs.length); it(
'should return correct allowance after approve (with data) and show the event on receiver contract'
, async function () {
const message = await Message.new();
new BigNumber(100).should.be.bignumber.equal( const extraData = message.contract.showMessage.getData(
await token.balanceOf(message.contract.address) web3.toHex(123456), 666, 'Transfer Done'
); );
});
it('should return correct allowance after approve (with data) and show the event on receiver contract', async function () { const abiMethod = findMethod(token.abi, 'approve', 'address,uint256,bytes');
const message = await Message.new(); const approveData = ethjsABI.encodeMethod(abiMethod,
[message.contract.address, 100, extraData]
);
const transaction = await token.sendTransaction(
{ from: accounts[0], data: approveData }
);
const extraData = message.contract.showMessage.getData( assert.equal(2, transaction.receipt.logs.length);
web3.toHex(123456), 666, 'Transfer Done'
);
// Use method #3 approve of the abi to encode the data tx new BigNumber(100).should.be.bignumber.equal(
const approveData = ethjsABI.encodeMethod(token.abi[3], await token.allowance(accounts[0], message.contract.address)
[message.contract.address, 100, extraData] );
); });
const transaction = await token.sendTransaction(
{ from: accounts[0], data: approveData }
);
assert.equal(2, transaction.receipt.logs.length); it(
'should return correct balances after transferFrom (with data) and show the event on receiver contract'
, async function () {
const message = await Message.new();
new BigNumber(100).should.be.bignumber.equal( const extraData = message.contract.showMessage.getData(
await token.allowance(accounts[0], message.contract.address) web3.toHex(123456), 666, 'Transfer Done'
); );
});
it('should return correct balances after transferFrom (with data) and show the event on receiver contract', async function () { await token.approve(accounts[1], 100, { from: accounts[0] });
const message = await Message.new();
const extraData = message.contract.showMessage.getData( new BigNumber(100).should.be.bignumber.equal(
web3.toHex(123456), 666, 'Transfer Done' await token.allowance(accounts[0], accounts[1])
); );
await token.approve(accounts[1], 100, { from: accounts[0] }); const abiMethod = findMethod(token.abi, 'transferFrom', 'address,address,uint256,bytes');
const transferFromData = ethjsABI.encodeMethod(abiMethod,
[accounts[0], message.contract.address, 100, extraData]
);
const transaction = await token.sendTransaction(
{ from: accounts[1], data: transferFromData }
);
new BigNumber(100).should.be.bignumber.equal( assert.equal(2, transaction.receipt.logs.length);
await token.allowance(accounts[0], accounts[1])
);
// Use method #7 transferFrom of the abi to encode the data tx new BigNumber(100).should.be.bignumber.equal(
const transferFromData = ethjsABI.encodeMethod(token.abi[7], await token.balanceOf(message.contract.address)
[accounts[0], message.contract.address, 100, extraData] );
); });
const transaction = await token.sendTransaction(
{ from: accounts[1], data: transferFromData }
);
assert.equal(2, transaction.receipt.logs.length);
new BigNumber(100).should.be.bignumber.equal(
await token.balanceOf(message.contract.address)
);
});
it('should fail inside approve (with data)', async function () { it('should fail inside approve (with data)', async function () {
const message = await Message.new(); const message = await Message.new();
const extraData = message.contract.fail.getData(); const extraData = message.contract.fail.getData();
// Use method #3 approve of the abi to encode the data tx const abiMethod = findMethod(token.abi, 'approve', 'address,uint256,bytes');
const approveData = ethjsABI.encodeMethod(token.abi[3], const approveData = ethjsABI.encodeMethod(abiMethod,
[message.contract.address, 10, extraData] [message.contract.address, 10, extraData]
); );
await token.sendTransaction( await token.sendTransaction(
@ -207,8 +221,8 @@ contract('ERC827 Token', function (accounts) {
const extraData = message.contract.fail.getData(); const extraData = message.contract.fail.getData();
// Use method #8 tranfer of the abi to encode the data tx const abiMethod = findMethod(token.abi, 'transfer', 'address,uint256,bytes');
const transferData = ethjsABI.encodeMethod(token.abi[8], const transferData = ethjsABI.encodeMethod(abiMethod,
[message.contract.address, 10, extraData] [message.contract.address, 10, extraData]
); );
await token.sendTransaction( await token.sendTransaction(
@ -227,8 +241,8 @@ contract('ERC827 Token', function (accounts) {
await token.approve(accounts[1], 10, { from: accounts[2] }); await token.approve(accounts[1], 10, { from: accounts[2] });
// Use method #7 tranferFrom of the abi to encode the data tx const abiMethod = findMethod(token.abi, 'transferFrom', 'address,address,uint256,bytes');
const transferFromData = ethjsABI.encodeMethod(token.abi[7], const transferFromData = ethjsABI.encodeMethod(abiMethod,
[accounts[2], message.contract.address, 10, extraData] [accounts[2], message.contract.address, 10, extraData]
); );
await token.sendTransaction( await token.sendTransaction(
@ -249,8 +263,8 @@ contract('ERC827 Token', function (accounts) {
web3.toHex(123456), 666, 'Transfer Done' web3.toHex(123456), 666, 'Transfer Done'
); );
// Use method #3 approve of the abi to encode the data tx const abiMethod = findMethod(token.abi, 'approve', 'address,uint256,bytes');
const approveData = ethjsABI.encodeMethod(token.abi[3], const approveData = ethjsABI.encodeMethod(abiMethod,
[token.contract.address, 100, extraData] [token.contract.address, 100, extraData]
); );
await token.sendTransaction( await token.sendTransaction(
@ -265,8 +279,8 @@ contract('ERC827 Token', function (accounts) {
web3.toHex(123456), 666, 'Transfer Done' web3.toHex(123456), 666, 'Transfer Done'
); );
// Use method #8 tranfer of the abi to encode the data tx const abiMethod = findMethod(token.abi, 'transfer', 'address,uint256,bytes');
const transferData = ethjsABI.encodeMethod(token.abi[8], const transferData = ethjsABI.encodeMethod(abiMethod,
[token.contract.address, 100, extraData] [token.contract.address, 100, extraData]
); );
await token.sendTransaction( await token.sendTransaction(
@ -283,8 +297,8 @@ contract('ERC827 Token', function (accounts) {
await token.approve(accounts[1], 1, { from: accounts[0] }); await token.approve(accounts[1], 1, { from: accounts[0] });
// Use method #7 tranferFrom of the abi to encode the data tx const abiMethod = findMethod(token.abi, 'transferFrom', 'address,address,uint256,bytes');
const transferFromData = ethjsABI.encodeMethod(token.abi[7], const transferFromData = ethjsABI.encodeMethod(abiMethod,
[accounts[0], token.contract.address, 1, extraData] [accounts[0], token.contract.address, 1, extraData]
); );
await token.sendTransaction( await token.sendTransaction(