feat: Migrate PositiveSlippage tests [LIT-804] (#650)
* feat: PositiveSlippageTransformer tests * Remove old tests * Affilite fee tests * Create mintable erc20 token for v06 * Update MintableERC20TokenV06 name
This commit is contained in:
Jacob Evans
GitHub
parent
c4b1f043c6
commit
80193215e3
117
contracts/erc20/contracts/src/v06/MintableERC20TokenV06.sol
Normal file
117
contracts/erc20/contracts/src/v06/MintableERC20TokenV06.sol
Normal file
@ -0,0 +1,117 @@
|
||||
/*
|
||||
|
||||
Copyright 2023 ZeroEx Intl.
|
||||
|
||||
Licensed under the Apache License, Version 2.0 (the "License");
|
||||
you may not use this file except in compliance with the License.
|
||||
You may obtain a copy of the License at
|
||||
|
||||
http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
||||
Unless required by applicable law or agreed to in writing, software
|
||||
distributed under the License is distributed on an "AS IS" BASIS,
|
||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
|
||||
*/
|
||||
|
||||
pragma solidity ^0.6.5;
|
||||
|
||||
import "@0x/contracts-utils/contracts/src/v06/LibSafeMathV06.sol";
|
||||
import "./IERC20TokenV06.sol";
|
||||
|
||||
contract MintableERC20TokenV06 is IERC20TokenV06 {
|
||||
using LibSafeMathV06 for uint256;
|
||||
|
||||
uint8 public override decimals = 18;
|
||||
mapping(address => uint256) internal balances;
|
||||
mapping(address => mapping(address => uint256)) internal allowed;
|
||||
|
||||
uint256 internal _totalSupply;
|
||||
|
||||
/// @dev send `value` token to `to` from `msg.sender`
|
||||
/// @param _to The address of the recipient
|
||||
/// @param _value The amount of token to be transferred
|
||||
/// @return True if transfer was successful
|
||||
function transfer(address _to, uint256 _value) external override returns (bool) {
|
||||
require(balances[msg.sender] >= _value, "ERC20_INSUFFICIENT_BALANCE");
|
||||
require(balances[_to].safeAdd(_value) >= balances[_to], "UINT256_OVERFLOW");
|
||||
|
||||
balances[msg.sender] = balances[msg.sender].safeSub(_value);
|
||||
balances[_to] = balances[_to].safeAdd(_value);
|
||||
|
||||
emit Transfer(msg.sender, _to, _value);
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
/// @dev send `value` token to `to` from `from` on the condition it is approved by `from`
|
||||
/// @param _from The address of the sender
|
||||
/// @param _to The address of the recipient
|
||||
/// @param _value The amount of token to be transferred
|
||||
/// @return True if transfer was successful
|
||||
function transferFrom(address _from, address _to, uint256 _value) external override returns (bool) {
|
||||
require(balances[_from] >= _value, "ERC20_INSUFFICIENT_BALANCE");
|
||||
require(allowed[_from][msg.sender] >= _value, "ERC20_INSUFFICIENT_ALLOWANCE");
|
||||
require(balances[_to].safeAdd(_value) >= balances[_to], "UINT256_OVERFLOW");
|
||||
|
||||
balances[_to] = balances[_to].safeAdd(_value);
|
||||
balances[_from] = balances[_from].safeSub(_value);
|
||||
allowed[_from][msg.sender] = allowed[_from][msg.sender].safeSub(_value);
|
||||
|
||||
emit Transfer(_from, _to, _value);
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
/// @dev `msg.sender` approves `_spender` to spend `_value` tokens
|
||||
/// @param _spender The address of the account able to transfer the tokens
|
||||
/// @param _value The amount of wei to be approved for transfer
|
||||
/// @return Always true if the call has enough gas to complete execution
|
||||
function approve(address _spender, uint256 _value) external override returns (bool) {
|
||||
allowed[msg.sender][_spender] = _value;
|
||||
emit Approval(msg.sender, _spender, _value);
|
||||
return true;
|
||||
}
|
||||
|
||||
/// @dev Query total supply of token
|
||||
/// @return Total supply of token
|
||||
function totalSupply() external view override returns (uint256) {
|
||||
return _totalSupply;
|
||||
}
|
||||
|
||||
/// @dev Query the balance of owner
|
||||
/// @param _owner The address from which the balance will be retrieved
|
||||
/// @return Balance of owner
|
||||
function balanceOf(address _owner) external view override returns (uint256) {
|
||||
return balances[_owner];
|
||||
}
|
||||
|
||||
/// @param _owner The address of the account owning tokens
|
||||
/// @param _spender The address of the account able to transfer the tokens
|
||||
/// @return Amount of remaining tokens allowed to spent
|
||||
function allowance(address _owner, address _spender) external view override returns (uint256) {
|
||||
return allowed[_owner][_spender];
|
||||
}
|
||||
|
||||
/// @dev Mints new tokens
|
||||
/// @param _to Address of the beneficiary that will own the minted token
|
||||
/// @param _value Amount of tokens to mint
|
||||
function _mint(address _to, uint256 _value) internal {
|
||||
balances[_to] = _value.safeAdd(balances[_to]);
|
||||
_totalSupply = _totalSupply.safeAdd(_value);
|
||||
|
||||
emit Transfer(address(0), _to, _value);
|
||||
}
|
||||
|
||||
/// @dev Mints new tokens
|
||||
/// @param _owner Owner of tokens that will be burned
|
||||
/// @param _value Amount of tokens to burn
|
||||
function _burn(address _owner, uint256 _value) internal {
|
||||
balances[_owner] = balances[_owner].safeSub(_value);
|
||||
_totalSupply = _totalSupply.safeSub(_value);
|
||||
|
||||
emit Transfer(_owner, address(0), _value);
|
||||
}
|
||||
}
|
||||
@ -1,168 +0,0 @@
|
||||
import { blockchainTests, constants, expect, getRandomInteger, randomAddress } from '@0x/contracts-test-utils';
|
||||
import { encodeAffiliateFeeTransformerData, ETH_TOKEN_ADDRESS } from '@0x/protocol-utils';
|
||||
import { BigNumber } from '@0x/utils';
|
||||
import * as _ from 'lodash';
|
||||
|
||||
import { artifacts } from '../artifacts';
|
||||
import {
|
||||
AffiliateFeeTransformerContract,
|
||||
TestMintableERC20TokenContract,
|
||||
TestTransformerHostContract,
|
||||
} from '../wrappers';
|
||||
|
||||
const { MAX_UINT256, ZERO_AMOUNT } = constants;
|
||||
|
||||
blockchainTests.resets('AffiliateFeeTransformer', env => {
|
||||
const recipients = new Array(2).fill(0).map(() => randomAddress());
|
||||
let caller: string;
|
||||
let token: TestMintableERC20TokenContract;
|
||||
let transformer: AffiliateFeeTransformerContract;
|
||||
let host: TestTransformerHostContract;
|
||||
|
||||
before(async () => {
|
||||
[caller] = await env.getAccountAddressesAsync();
|
||||
token = await TestMintableERC20TokenContract.deployFrom0xArtifactAsync(
|
||||
artifacts.TestMintableERC20Token,
|
||||
env.provider,
|
||||
env.txDefaults,
|
||||
artifacts,
|
||||
);
|
||||
transformer = await AffiliateFeeTransformerContract.deployFrom0xArtifactAsync(
|
||||
artifacts.AffiliateFeeTransformer,
|
||||
env.provider,
|
||||
env.txDefaults,
|
||||
artifacts,
|
||||
);
|
||||
host = await TestTransformerHostContract.deployFrom0xArtifactAsync(
|
||||
artifacts.TestTransformerHost,
|
||||
env.provider,
|
||||
{ ...env.txDefaults, from: caller },
|
||||
artifacts,
|
||||
);
|
||||
});
|
||||
|
||||
interface Balances {
|
||||
ethBalance: BigNumber;
|
||||
tokenBalance: BigNumber;
|
||||
}
|
||||
|
||||
const ZERO_BALANCES = {
|
||||
ethBalance: ZERO_AMOUNT,
|
||||
tokenBalance: ZERO_AMOUNT,
|
||||
};
|
||||
|
||||
async function getBalancesAsync(owner: string): Promise<Balances> {
|
||||
return {
|
||||
ethBalance: await env.web3Wrapper.getBalanceInWeiAsync(owner),
|
||||
tokenBalance: await token.balanceOf(owner).callAsync(),
|
||||
};
|
||||
}
|
||||
|
||||
async function mintHostTokensAsync(amount: BigNumber): Promise<void> {
|
||||
await token.mint(host.address, amount).awaitTransactionSuccessAsync();
|
||||
}
|
||||
|
||||
async function sendEtherAsync(to: string, amount: BigNumber): Promise<void> {
|
||||
await env.web3Wrapper.awaitTransactionSuccessAsync(
|
||||
await env.web3Wrapper.sendTransactionAsync({
|
||||
...env.txDefaults,
|
||||
to,
|
||||
from: caller,
|
||||
value: amount,
|
||||
}),
|
||||
);
|
||||
}
|
||||
|
||||
it('can transfer a token and ETH', async () => {
|
||||
const amounts = recipients.map(() => getRandomInteger(1, '1e18'));
|
||||
const tokens = [token.address, ETH_TOKEN_ADDRESS];
|
||||
const data = encodeAffiliateFeeTransformerData({
|
||||
fees: recipients.map((r, i) => ({
|
||||
token: tokens[i],
|
||||
amount: amounts[i],
|
||||
recipient: r,
|
||||
})),
|
||||
});
|
||||
await mintHostTokensAsync(amounts[0]);
|
||||
await sendEtherAsync(host.address, amounts[1]);
|
||||
await host
|
||||
.rawExecuteTransform(transformer.address, {
|
||||
data,
|
||||
sender: randomAddress(),
|
||||
recipient: randomAddress(),
|
||||
})
|
||||
.awaitTransactionSuccessAsync();
|
||||
expect(await getBalancesAsync(host.address)).to.deep.eq(ZERO_BALANCES);
|
||||
expect(await getBalancesAsync(recipients[0])).to.deep.eq({
|
||||
tokenBalance: amounts[0],
|
||||
ethBalance: ZERO_AMOUNT,
|
||||
});
|
||||
expect(await getBalancesAsync(recipients[1])).to.deep.eq({
|
||||
tokenBalance: ZERO_AMOUNT,
|
||||
ethBalance: amounts[1],
|
||||
});
|
||||
});
|
||||
|
||||
it('can transfer all of a token and ETH', async () => {
|
||||
const amounts = recipients.map(() => getRandomInteger(1, '1e18'));
|
||||
const tokens = [token.address, ETH_TOKEN_ADDRESS];
|
||||
const data = encodeAffiliateFeeTransformerData({
|
||||
fees: recipients.map((r, i) => ({
|
||||
token: tokens[i],
|
||||
amount: MAX_UINT256,
|
||||
recipient: r,
|
||||
})),
|
||||
});
|
||||
await mintHostTokensAsync(amounts[0]);
|
||||
await sendEtherAsync(host.address, amounts[1]);
|
||||
await host
|
||||
.rawExecuteTransform(transformer.address, {
|
||||
data,
|
||||
sender: randomAddress(),
|
||||
recipient: randomAddress(),
|
||||
})
|
||||
.awaitTransactionSuccessAsync();
|
||||
expect(await getBalancesAsync(host.address)).to.deep.eq(ZERO_BALANCES);
|
||||
expect(await getBalancesAsync(recipients[0])).to.deep.eq({
|
||||
tokenBalance: amounts[0],
|
||||
ethBalance: ZERO_AMOUNT,
|
||||
});
|
||||
expect(await getBalancesAsync(recipients[1])).to.deep.eq({
|
||||
tokenBalance: ZERO_AMOUNT,
|
||||
ethBalance: amounts[1],
|
||||
});
|
||||
});
|
||||
|
||||
it('can transfer less than the balance of a token and ETH', async () => {
|
||||
const amounts = recipients.map(() => getRandomInteger(1, '1e18'));
|
||||
const tokens = [token.address, ETH_TOKEN_ADDRESS];
|
||||
const data = encodeAffiliateFeeTransformerData({
|
||||
fees: recipients.map((r, i) => ({
|
||||
token: tokens[i],
|
||||
amount: amounts[i].minus(1),
|
||||
recipient: r,
|
||||
})),
|
||||
});
|
||||
await mintHostTokensAsync(amounts[0]);
|
||||
await sendEtherAsync(host.address, amounts[1]);
|
||||
await host
|
||||
.rawExecuteTransform(transformer.address, {
|
||||
data,
|
||||
sender: randomAddress(),
|
||||
recipient: randomAddress(),
|
||||
})
|
||||
.awaitTransactionSuccessAsync();
|
||||
expect(await getBalancesAsync(host.address)).to.deep.eq({
|
||||
tokenBalance: new BigNumber(1),
|
||||
ethBalance: new BigNumber(1),
|
||||
});
|
||||
expect(await getBalancesAsync(recipients[0])).to.deep.eq({
|
||||
tokenBalance: amounts[0].minus(1),
|
||||
ethBalance: ZERO_AMOUNT,
|
||||
});
|
||||
expect(await getBalancesAsync(recipients[1])).to.deep.eq({
|
||||
tokenBalance: ZERO_AMOUNT,
|
||||
ethBalance: amounts[1].minus(1),
|
||||
});
|
||||
});
|
||||
});
|
||||
@ -1,127 +0,0 @@
|
||||
import { blockchainTests, constants, expect, getRandomInteger, randomAddress } from '@0x/contracts-test-utils';
|
||||
import { encodePositiveSlippageFeeTransformerData } from '@0x/protocol-utils';
|
||||
import { BigNumber } from '@0x/utils';
|
||||
|
||||
import { artifacts } from '../artifacts';
|
||||
import {
|
||||
PositiveSlippageFeeTransformerContract,
|
||||
TestMintableERC20TokenContract,
|
||||
TestTransformerHostContract,
|
||||
} from '../wrappers';
|
||||
|
||||
const { ZERO_AMOUNT } = constants;
|
||||
|
||||
blockchainTests.resets('PositiveSlippageFeeTransformer', env => {
|
||||
const recipient = randomAddress();
|
||||
let caller: string;
|
||||
let token: TestMintableERC20TokenContract;
|
||||
let transformer: PositiveSlippageFeeTransformerContract;
|
||||
let host: TestTransformerHostContract;
|
||||
|
||||
before(async () => {
|
||||
[caller] = await env.getAccountAddressesAsync();
|
||||
token = await TestMintableERC20TokenContract.deployFrom0xArtifactAsync(
|
||||
artifacts.TestMintableERC20Token,
|
||||
env.provider,
|
||||
env.txDefaults,
|
||||
artifacts,
|
||||
);
|
||||
transformer = await PositiveSlippageFeeTransformerContract.deployFrom0xArtifactAsync(
|
||||
artifacts.PositiveSlippageFeeTransformer,
|
||||
env.provider,
|
||||
env.txDefaults,
|
||||
artifacts,
|
||||
);
|
||||
host = await TestTransformerHostContract.deployFrom0xArtifactAsync(
|
||||
artifacts.TestTransformerHost,
|
||||
env.provider,
|
||||
{ ...env.txDefaults, from: caller },
|
||||
artifacts,
|
||||
);
|
||||
});
|
||||
|
||||
interface Balances {
|
||||
ethBalance: BigNumber;
|
||||
tokenBalance: BigNumber;
|
||||
}
|
||||
|
||||
async function getBalancesAsync(owner: string): Promise<Balances> {
|
||||
return {
|
||||
ethBalance: await env.web3Wrapper.getBalanceInWeiAsync(owner),
|
||||
tokenBalance: await token.balanceOf(owner).callAsync(),
|
||||
};
|
||||
}
|
||||
|
||||
async function mintHostTokensAsync(amount: BigNumber): Promise<void> {
|
||||
await token.mint(host.address, amount).awaitTransactionSuccessAsync();
|
||||
}
|
||||
|
||||
it('does not transfer positive slippage fees when bestCaseAmount is equal to amount', async () => {
|
||||
const amount = getRandomInteger(1, '1e18');
|
||||
const data = encodePositiveSlippageFeeTransformerData({
|
||||
token: token.address,
|
||||
bestCaseAmount: amount,
|
||||
recipient,
|
||||
});
|
||||
await mintHostTokensAsync(amount);
|
||||
const beforeBalanceHost = await getBalancesAsync(host.address);
|
||||
const beforeBalanceRecipient = await getBalancesAsync(recipient);
|
||||
await host
|
||||
.rawExecuteTransform(transformer.address, {
|
||||
data,
|
||||
sender: randomAddress(),
|
||||
recipient: randomAddress(),
|
||||
})
|
||||
.awaitTransactionSuccessAsync();
|
||||
expect(await getBalancesAsync(host.address)).to.deep.eq(beforeBalanceHost);
|
||||
expect(await getBalancesAsync(recipient)).to.deep.eq(beforeBalanceRecipient);
|
||||
});
|
||||
|
||||
it('does not transfer positive slippage fees when bestCaseAmount is higher than amount', async () => {
|
||||
const amount = getRandomInteger(1, '1e18');
|
||||
const bestCaseAmount = amount.times(1.1).decimalPlaces(0, BigNumber.ROUND_FLOOR);
|
||||
const data = encodePositiveSlippageFeeTransformerData({
|
||||
token: token.address,
|
||||
bestCaseAmount,
|
||||
recipient,
|
||||
});
|
||||
await mintHostTokensAsync(amount);
|
||||
const beforeBalanceHost = await getBalancesAsync(host.address);
|
||||
const beforeBalanceRecipient = await getBalancesAsync(recipient);
|
||||
await host
|
||||
.rawExecuteTransform(transformer.address, {
|
||||
data,
|
||||
sender: randomAddress(),
|
||||
recipient: randomAddress(),
|
||||
})
|
||||
.awaitTransactionSuccessAsync();
|
||||
expect(await getBalancesAsync(host.address)).to.deep.eq(beforeBalanceHost);
|
||||
expect(await getBalancesAsync(recipient)).to.deep.eq(beforeBalanceRecipient);
|
||||
});
|
||||
|
||||
it('send positive slippage fee to recipient when bestCaseAmount is lower than amount', async () => {
|
||||
const amount = getRandomInteger(1, '1e18');
|
||||
const bestCaseAmount = amount.times(0.95).decimalPlaces(0, BigNumber.ROUND_FLOOR);
|
||||
const data = encodePositiveSlippageFeeTransformerData({
|
||||
token: token.address,
|
||||
bestCaseAmount,
|
||||
recipient,
|
||||
});
|
||||
await mintHostTokensAsync(amount);
|
||||
await host
|
||||
.rawExecuteTransform(transformer.address, {
|
||||
data,
|
||||
sender: randomAddress(),
|
||||
recipient: randomAddress(),
|
||||
})
|
||||
.awaitTransactionSuccessAsync();
|
||||
expect(await getBalancesAsync(host.address)).to.deep.eq({
|
||||
tokenBalance: bestCaseAmount,
|
||||
ethBalance: ZERO_AMOUNT,
|
||||
});
|
||||
expect(await getBalancesAsync(recipient)).to.deep.eq({
|
||||
tokenBalance: amount.minus(bestCaseAmount), // positive slippage
|
||||
ethBalance: ZERO_AMOUNT,
|
||||
});
|
||||
});
|
||||
});
|
||||
@ -0,0 +1,187 @@
|
||||
// SPDX-License-Identifier: Apache-2.0
|
||||
/*
|
||||
Copyright 2023 ZeroEx Intl.
|
||||
Licensed under the Apache License, Version 2.0 (the "License");
|
||||
you may not use this file except in compliance with the License.
|
||||
You may obtain a copy of the License at
|
||||
http://www.apache.org/licenses/LICENSE-2.0
|
||||
Unless required by applicable law or agreed to in writing, software
|
||||
distributed under the License is distributed on an "AS IS" BASIS,
|
||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
*/
|
||||
|
||||
pragma solidity ^0.6.5;
|
||||
pragma experimental ABIEncoderV2;
|
||||
|
||||
import "@0x/contracts-erc20/contracts/src/v06/WETH9V06.sol";
|
||||
import "@0x/contracts-erc20/contracts/src/v06/MintableERC20TokenV06.sol";
|
||||
|
||||
import "../BaseTest.sol";
|
||||
import "../../contracts/src/transformers/AffiliateFeeTransformer.sol";
|
||||
import "../../contracts/src/transformers/IERC20Transformer.sol";
|
||||
|
||||
contract AffiliateFeeTransformerTest is BaseTest {
|
||||
address public owner = account1;
|
||||
address public feeRecipient = account2;
|
||||
WETH9V06 weth = new WETH9V06();
|
||||
IERC20TokenV06 token1 = IERC20TokenV06(address(weth));
|
||||
|
||||
AffiliateFeeTransformer target = new AffiliateFeeTransformer();
|
||||
|
||||
function setUp() public {
|
||||
vm.deal(address(this), 1e19);
|
||||
weth.deposit{value: 10}();
|
||||
}
|
||||
|
||||
function test_affiliateFee() public {
|
||||
// Send positive slippage to contract which executes AffiliateFeeTransformer
|
||||
weth.transfer(address(target), 10);
|
||||
uint256 affiliateFeeAmount = 1;
|
||||
|
||||
AffiliateFeeTransformer.TokenFee[] memory fees = new AffiliateFeeTransformer.TokenFee[](1);
|
||||
fees[0] = AffiliateFeeTransformer.TokenFee({
|
||||
token: IERC20TokenV06(token1),
|
||||
amount: affiliateFeeAmount,
|
||||
recipient: payable(feeRecipient)
|
||||
});
|
||||
|
||||
bytes4 result = target.transform(
|
||||
IERC20Transformer.TransformContext({
|
||||
sender: payable(address(this)),
|
||||
recipient: payable(address(this)),
|
||||
data: abi.encode(fees)
|
||||
})
|
||||
);
|
||||
assertEq(token1.balanceOf(feeRecipient), affiliateFeeAmount);
|
||||
assertEq(result, LibERC20Transformer.TRANSFORMER_SUCCESS);
|
||||
}
|
||||
|
||||
function test_affiliateFee_entireBalance() public {
|
||||
// Send positive slippage to contract which executes AffiliateFeeTransformer
|
||||
weth.transfer(address(target), 10);
|
||||
|
||||
AffiliateFeeTransformer.TokenFee[] memory fees = new AffiliateFeeTransformer.TokenFee[](1);
|
||||
fees[0] = AffiliateFeeTransformer.TokenFee({
|
||||
token: IERC20TokenV06(token1),
|
||||
amount: uint256(-1),
|
||||
recipient: payable(feeRecipient)
|
||||
});
|
||||
|
||||
bytes4 result = target.transform(
|
||||
IERC20Transformer.TransformContext({
|
||||
sender: payable(address(this)),
|
||||
recipient: payable(address(this)),
|
||||
data: abi.encode(fees)
|
||||
})
|
||||
);
|
||||
assertEq(token1.balanceOf(feeRecipient), 10);
|
||||
assertEq(result, LibERC20Transformer.TRANSFORMER_SUCCESS);
|
||||
}
|
||||
|
||||
function test_affiliateFee_multipleFees() public {
|
||||
// Send positive slippage to contract which executes AffiliateFeeTransformer
|
||||
weth.transfer(address(target), 10);
|
||||
|
||||
AffiliateFeeTransformer.TokenFee[] memory fees = new AffiliateFeeTransformer.TokenFee[](2);
|
||||
fees[0] = AffiliateFeeTransformer.TokenFee({
|
||||
token: IERC20TokenV06(token1),
|
||||
amount: uint256(1),
|
||||
recipient: payable(feeRecipient)
|
||||
});
|
||||
|
||||
fees[1] = AffiliateFeeTransformer.TokenFee({
|
||||
token: IERC20TokenV06(token1),
|
||||
amount: uint256(1),
|
||||
recipient: payable(feeRecipient)
|
||||
});
|
||||
|
||||
bytes4 result = target.transform(
|
||||
IERC20Transformer.TransformContext({
|
||||
sender: payable(address(this)),
|
||||
recipient: payable(address(this)),
|
||||
data: abi.encode(fees)
|
||||
})
|
||||
);
|
||||
assertEq(token1.balanceOf(feeRecipient), 2);
|
||||
assertEq(result, LibERC20Transformer.TRANSFORMER_SUCCESS);
|
||||
}
|
||||
|
||||
function test_affiliateFee_ethFee() public {
|
||||
// Send positive slippage ETH to contract which executes AffiliateFeeTransformer
|
||||
vm.deal(address(target), 1);
|
||||
uint256 ethBalanceBefore = feeRecipient.balance;
|
||||
|
||||
AffiliateFeeTransformer.TokenFee[] memory fees = new AffiliateFeeTransformer.TokenFee[](1);
|
||||
fees[0] = AffiliateFeeTransformer.TokenFee({
|
||||
token: IERC20TokenV06(LibERC20Transformer.ETH_TOKEN_ADDRESS),
|
||||
amount: uint256(1),
|
||||
recipient: payable(feeRecipient)
|
||||
});
|
||||
|
||||
bytes4 result = target.transform(
|
||||
IERC20Transformer.TransformContext({
|
||||
sender: payable(address(this)),
|
||||
recipient: payable(address(this)),
|
||||
data: abi.encode(fees)
|
||||
})
|
||||
);
|
||||
assertEq(feeRecipient.balance, ethBalanceBefore + 1);
|
||||
assertEq(result, LibERC20Transformer.TRANSFORMER_SUCCESS);
|
||||
}
|
||||
|
||||
function test_affiliateFee_multipleFeesEthToken() public {
|
||||
// Send positive slippage to contract which executes AffiliateFeeTransformer
|
||||
weth.transfer(address(target), 10);
|
||||
vm.deal(address(target), 10);
|
||||
|
||||
uint256 ethBalanceBefore = feeRecipient.balance;
|
||||
|
||||
AffiliateFeeTransformer.TokenFee[] memory fees = new AffiliateFeeTransformer.TokenFee[](2);
|
||||
fees[0] = AffiliateFeeTransformer.TokenFee({
|
||||
token: IERC20TokenV06(LibERC20Transformer.ETH_TOKEN_ADDRESS),
|
||||
amount: uint256(1),
|
||||
recipient: payable(feeRecipient)
|
||||
});
|
||||
|
||||
fees[1] = AffiliateFeeTransformer.TokenFee({
|
||||
token: IERC20TokenV06(token1),
|
||||
amount: uint256(1),
|
||||
recipient: payable(feeRecipient)
|
||||
});
|
||||
|
||||
bytes4 result = target.transform(
|
||||
IERC20Transformer.TransformContext({
|
||||
sender: payable(address(this)),
|
||||
recipient: payable(address(this)),
|
||||
data: abi.encode(fees)
|
||||
})
|
||||
);
|
||||
assertEq(feeRecipient.balance, ethBalanceBefore + 1);
|
||||
assertEq(token1.balanceOf(feeRecipient), 1);
|
||||
assertEq(result, LibERC20Transformer.TRANSFORMER_SUCCESS);
|
||||
}
|
||||
|
||||
function test_affiliateFee_zeroAmount() public {
|
||||
// Send positive slippage to contract which executes AffiliateFeeTransformer
|
||||
weth.transfer(address(target), 10);
|
||||
|
||||
AffiliateFeeTransformer.TokenFee[] memory fees = new AffiliateFeeTransformer.TokenFee[](1);
|
||||
fees[0] = AffiliateFeeTransformer.TokenFee({
|
||||
token: IERC20TokenV06(token1),
|
||||
amount: 0,
|
||||
recipient: payable(feeRecipient)
|
||||
});
|
||||
|
||||
bytes4 result = target.transform(
|
||||
IERC20Transformer.TransformContext({
|
||||
sender: payable(address(this)),
|
||||
recipient: payable(address(this)),
|
||||
data: abi.encode(fees)
|
||||
})
|
||||
);
|
||||
assertEq(token1.balanceOf(feeRecipient), 0);
|
||||
assertEq(result, LibERC20Transformer.TRANSFORMER_SUCCESS);
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,100 @@
|
||||
// SPDX-License-Identifier: Apache-2.0
|
||||
/*
|
||||
Copyright 2023 ZeroEx Intl.
|
||||
Licensed under the Apache License, Version 2.0 (the "License");
|
||||
you may not use this file except in compliance with the License.
|
||||
You may obtain a copy of the License at
|
||||
http://www.apache.org/licenses/LICENSE-2.0
|
||||
Unless required by applicable law or agreed to in writing, software
|
||||
distributed under the License is distributed on an "AS IS" BASIS,
|
||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
*/
|
||||
|
||||
pragma solidity ^0.6.5;
|
||||
pragma experimental ABIEncoderV2;
|
||||
|
||||
import "@0x/contracts-erc20/contracts/src/v06/WETH9V06.sol";
|
||||
|
||||
import "../BaseTest.sol";
|
||||
import "../../contracts/src/transformers/PositiveSlippageFeeTransformer.sol";
|
||||
import "../../contracts/src/transformers/IERC20Transformer.sol";
|
||||
|
||||
contract PositiveSlippageFeeTransformerTest is BaseTest {
|
||||
address public owner = account1;
|
||||
address public feeRecipient = account2;
|
||||
WETH9V06 weth = new WETH9V06();
|
||||
IERC20TokenV06 token1 = IERC20TokenV06(address(weth));
|
||||
|
||||
PositiveSlippageFeeTransformer target = new PositiveSlippageFeeTransformer();
|
||||
|
||||
function setUp() public {
|
||||
vm.deal(address(this), 1e19);
|
||||
weth.deposit{value: 10}();
|
||||
}
|
||||
|
||||
function test_positiveSlippageFee() public {
|
||||
// Send positive slippage to contract which executes PositiveSlippageFeeTransformer
|
||||
weth.transfer(address(target), 10);
|
||||
uint256 bestCaseAmount = 1;
|
||||
|
||||
bytes4 result = target.transform(
|
||||
IERC20Transformer.TransformContext({
|
||||
sender: payable(address(this)),
|
||||
recipient: payable(address(this)),
|
||||
data: abi.encode(
|
||||
PositiveSlippageFeeTransformer.TokenFee({
|
||||
token: IERC20TokenV06(token1),
|
||||
bestCaseAmount: bestCaseAmount,
|
||||
recipient: payable(feeRecipient)
|
||||
})
|
||||
)
|
||||
})
|
||||
);
|
||||
assertEq(token1.balanceOf(feeRecipient), 9);
|
||||
assertEq(result, LibERC20Transformer.TRANSFORMER_SUCCESS);
|
||||
}
|
||||
|
||||
function test_positiveSlippageFee_bestCaseEqualsAmount() public {
|
||||
uint256 bestCaseAmount = 10;
|
||||
weth.transfer(address(target), 10);
|
||||
|
||||
bytes4 result = target.transform(
|
||||
IERC20Transformer.TransformContext({
|
||||
sender: payable(address(this)),
|
||||
recipient: payable(address(this)),
|
||||
data: abi.encode(
|
||||
PositiveSlippageFeeTransformer.TokenFee({
|
||||
token: IERC20TokenV06(token1),
|
||||
bestCaseAmount: bestCaseAmount,
|
||||
recipient: payable(feeRecipient)
|
||||
})
|
||||
)
|
||||
})
|
||||
);
|
||||
assertEq(token1.balanceOf(feeRecipient), 0);
|
||||
assertEq(result, LibERC20Transformer.TRANSFORMER_SUCCESS);
|
||||
}
|
||||
|
||||
function test_positiveSlippageFee_bestCaseGreaterThanAmount() public {
|
||||
uint256 bestCaseAmount = 10;
|
||||
weth.transfer(address(target), 1);
|
||||
|
||||
bytes4 result = target.transform(
|
||||
IERC20Transformer.TransformContext({
|
||||
sender: payable(address(this)),
|
||||
recipient: payable(address(this)),
|
||||
data: abi.encode(
|
||||
PositiveSlippageFeeTransformer.TokenFee({
|
||||
token: IERC20TokenV06(token1),
|
||||
bestCaseAmount: bestCaseAmount,
|
||||
recipient: payable(feeRecipient)
|
||||
})
|
||||
)
|
||||
})
|
||||
);
|
||||
assertEq(token1.balanceOf(feeRecipient), 0);
|
||||
assertEq(result, LibERC20Transformer.TRANSFORMER_SUCCESS);
|
||||
}
|
||||
}
|
||||
Loading…
x
Reference in New Issue
Block a user