Skip to content

1inch tests #9

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
decaf-addict wants to merge 10 commits into
base: main
Choose a base branch
from
Open

1inch tests #9

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
e169d04
test: 1inch tests
decaf-addict Aug 26, 2022
bb4cdf4
chaining 1inch wip
decaf-addict Aug 26, 2022
a8bec11
patching wip
decaf-addict Sep 27, 2022
734f6f5
nit: naming
decaf-addict Sep 27, 2022
f85a394
use real addresses
decaf-addict Sep 27, 2022
a85ce50
fix: mech exec
decaf-addict Sep 27, 2022
552704b
wip
decaf-addict Sep 27, 2022
7e67943
wip
decaf-addict Sep 27, 2022
5aefb3a
Update test_one_inch.py
decaf-addict Sep 27, 2022
80e14dd
wip
decaf-addict Sep 30, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
32 changes: 32 additions & 0 deletions contracts/Helpers/RouterLike.sol
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,32 @@
// SPDX-License-Identifier: MIT
pragma solidity ^0.8.11;


contract RouterLike {
SwapDescription public saved;
SwapDescription public savedSimple;
struct SwapDescription {
address srcToken;
address dstToken;
address payable srcReceiver;
address payable dstReceiver;
uint256 amount;
uint256 minReturnAmount;
uint256 flags;
bytes permit;
}

function swap(
address caller,
SwapDescription calldata desc,
bytes calldata data
) public {
saved = desc;
}

function swapSimple(
SwapDescription calldata desc
) public {
savedSimple = desc;
}
}
4 changes: 4 additions & 0 deletions tests/conftest.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -96,3 +96,7 @@ def tuple_helper_yul(alice, TupleHelperYul):
@pytest.fixture(scope="module")
def tuple_helper_vy(alice, TupleHelperVy):
yield alice.deploy(TupleHelperVy)

@pytest.fixture(scope="module")
def router_like(alice, RouterLike):
yield alice.deploy(RouterLike)
40 changes: 36 additions & 4 deletions tests/test_chaining_actions.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@ def test_chaining_action(weiroll_vm, tuple_helper):
w_crv_yfi_weth = WeirollContract.createContract(crv_yfi_weth)

# One inch section, eth->yfi
planner.add(w_weth.approve(one_inch.address, 2**256-1))
planner.add(w_weth.approve(one_inch.address, 2 ** 256 - 1))
swap_url = "https://api.1inch.io/v4.0/1/swap"
r = requests.get(
swap_url,
Expand Down Expand Up @@ -60,11 +60,12 @@ def test_chaining_action(weiroll_vm, tuple_helper):
yfi_int_amount = ReturnValue('uint256', one_inch_amount.command)

# Now that we have the yfi amount, let's do curve logic
planner.add(w_weth.approve(w_curve_swap.address, 2**256-1))
planner.add(w_yfi.approve(w_curve_swap.address, 2**256-1))
planner.add(w_weth.approve(w_curve_swap.address, 2 ** 256 - 1))
planner.add(w_yfi.approve(w_curve_swap.address, 2 ** 256 - 1))

params = [Wei("5 ether"), yfi_int_amount]
curve_ret = planner.add(
w_curve_swap.add_liquidity([Wei("5 ether"), yfi_int_amount], 0)
w_curve_swap.add_liquidity(params, 0)
)

planner.add(w_crv_yfi_weth.transfer(w_tuple_helper.address, curve_ret))
Expand All @@ -75,3 +76,34 @@ def test_chaining_action(weiroll_vm, tuple_helper):
)

assert crv_yfi_weth.balanceOf(w_tuple_helper.address) > 0


def test_return_value(weiroll_vm, tuple_helper):
yfi = Contract("0x0bc529c00C6401aEF6D220BE8C6Ea1667F6Ad93e")
crv_yfi_weth = Contract("0x29059568bB40344487d62f7450E78b8E6C74e0e5")
curve_swap = Contract("0xC26b89A667578ec7b3f11b2F98d6Fd15C07C54ba")
yfi_whale = accounts.at("0xfeb4acf3df3cdea7399794d0869ef76a6efaff52", force=True)

# Planner and all weiroll contracts
planner = WeirollPlanner(weiroll_vm)
w_yfi = WeirollContract.createContract(yfi)
w_curve_swap = WeirollContract.createContract(curve_swap)

yfi.transfer(weiroll_vm.address, 10 ** 18, {'from': yfi_whale})

w_yfi_bal = planner.call(yfi, "balanceOf", weiroll_vm.address)
# Now that we have the yfi amount, let's do curve logic
planner.add(w_yfi.approve(w_curve_swap.address, 2 ** 256 - 1))

params = [0, w_yfi_bal]
planner.add(
w_curve_swap.add_liquidity(params, 0)
)

cmds, state = planner.plan()
weiroll_tx = weiroll_vm.execute(
cmds, state, {"from": yfi_whale, "gas_limit": 8_000_000, "gas_price": 0}
)

assert crv_yfi_weth.balanceOf(weiroll_vm.address) > 0
assert False
Loading