You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
109 lines
3.7 KiB
109 lines
3.7 KiB
const ethSigUtil = require('eth-sig-util');
|
|
const Wallet = require('ethereumjs-wallet').default;
|
|
const { EIP712Domain } = require('../helpers/eip712');
|
|
|
|
const { expectEvent } = require('@openzeppelin/test-helpers');
|
|
const { expect } = require('chai');
|
|
|
|
const ERC2771ContextMock = artifacts.require('ERC2771ContextMock');
|
|
const MinimalForwarder = artifacts.require('MinimalForwarder');
|
|
const ContextMockCaller = artifacts.require('ContextMockCaller');
|
|
|
|
const { shouldBehaveLikeRegularContext } = require('../utils/Context.behavior');
|
|
|
|
const name = 'MinimalForwarder';
|
|
const version = '0.0.1';
|
|
|
|
contract('ERC2771Context', function (accounts) {
|
|
beforeEach(async function () {
|
|
this.forwarder = await MinimalForwarder.new();
|
|
this.recipient = await ERC2771ContextMock.new(this.forwarder.address);
|
|
|
|
this.domain = {
|
|
name,
|
|
version,
|
|
chainId: await web3.eth.getChainId(),
|
|
verifyingContract: this.forwarder.address,
|
|
};
|
|
this.types = {
|
|
EIP712Domain,
|
|
ForwardRequest: [
|
|
{ name: 'from', type: 'address' },
|
|
{ name: 'to', type: 'address' },
|
|
{ name: 'value', type: 'uint256' },
|
|
{ name: 'gas', type: 'uint256' },
|
|
{ name: 'nonce', type: 'uint256' },
|
|
{ name: 'data', type: 'bytes' },
|
|
],
|
|
};
|
|
});
|
|
|
|
it('recognize trusted forwarder', async function () {
|
|
expect(await this.recipient.isTrustedForwarder(this.forwarder.address));
|
|
});
|
|
|
|
context('when called directly', function () {
|
|
beforeEach(async function () {
|
|
this.context = this.recipient; // The Context behavior expects the contract in this.context
|
|
this.caller = await ContextMockCaller.new();
|
|
});
|
|
|
|
shouldBehaveLikeRegularContext(...accounts);
|
|
});
|
|
|
|
context('when receiving a relayed call', function () {
|
|
beforeEach(async function () {
|
|
this.wallet = Wallet.generate();
|
|
this.sender = web3.utils.toChecksumAddress(this.wallet.getAddressString());
|
|
this.data = {
|
|
types: this.types,
|
|
domain: this.domain,
|
|
primaryType: 'ForwardRequest',
|
|
};
|
|
});
|
|
|
|
describe('msgSender', function () {
|
|
it('returns the relayed transaction original sender', async function () {
|
|
const data = this.recipient.contract.methods.msgSender().encodeABI();
|
|
|
|
const req = {
|
|
from: this.sender,
|
|
to: this.recipient.address,
|
|
value: '0',
|
|
gas: '100000',
|
|
nonce: (await this.forwarder.getNonce(this.sender)).toString(),
|
|
data,
|
|
};
|
|
|
|
const sign = ethSigUtil.signTypedMessage(this.wallet.getPrivateKey(), { data: { ...this.data, message: req } });
|
|
expect(await this.forwarder.verify(req, sign)).to.equal(true);
|
|
|
|
const { tx } = await this.forwarder.execute(req, sign);
|
|
await expectEvent.inTransaction(tx, ERC2771ContextMock, 'Sender', { sender: this.sender });
|
|
});
|
|
});
|
|
|
|
describe('msgData', function () {
|
|
it('returns the relayed transaction original data', async function () {
|
|
const integerValue = '42';
|
|
const stringValue = 'OpenZeppelin';
|
|
const data = this.recipient.contract.methods.msgData(integerValue, stringValue).encodeABI();
|
|
|
|
const req = {
|
|
from: this.sender,
|
|
to: this.recipient.address,
|
|
value: '0',
|
|
gas: '100000',
|
|
nonce: (await this.forwarder.getNonce(this.sender)).toString(),
|
|
data,
|
|
};
|
|
|
|
const sign = ethSigUtil.signTypedMessage(this.wallet.getPrivateKey(), { data: { ...this.data, message: req } });
|
|
expect(await this.forwarder.verify(req, sign)).to.equal(true);
|
|
|
|
const { tx } = await this.forwarder.execute(req, sign);
|
|
await expectEvent.inTransaction(tx, ERC2771ContextMock, 'Data', { data, integerValue, stringValue });
|
|
});
|
|
});
|
|
});
|
|
});
|
|
|