248 lines
9.1 KiB
Python
248 lines
9.1 KiB
Python
from mev_inspect.classifiers.specs.balancer import BALANCER_V1_POOL_ABI_NAME
|
|
from mev_inspect.classifiers.specs.bancor import (
|
|
BANCOR_NETWORK_ABI_NAME,
|
|
BANCOR_NETWORK_CONTRACT_ADDRESS,
|
|
)
|
|
from mev_inspect.classifiers.specs.uniswap import (
|
|
UNISWAP_V2_PAIR_ABI_NAME,
|
|
UNISWAP_V3_POOL_ABI_NAME,
|
|
)
|
|
from mev_inspect.schemas.traces import Protocol
|
|
from mev_inspect.swaps import get_swaps
|
|
|
|
from .helpers import make_swap_trace, make_transfer_trace, make_unknown_trace
|
|
|
|
|
|
def test_swaps(
|
|
get_transaction_hashes,
|
|
get_addresses,
|
|
):
|
|
block_number = 123
|
|
|
|
[
|
|
first_transaction_hash,
|
|
second_transaction_hash,
|
|
third_transaction_hash,
|
|
fourth_transaction_hash,
|
|
] = get_transaction_hashes(4)
|
|
|
|
[
|
|
alice_address,
|
|
bob_address,
|
|
carl_address,
|
|
danielle_address,
|
|
first_token_in_address,
|
|
first_token_out_address,
|
|
first_pool_address,
|
|
second_token_in_address,
|
|
second_token_out_address,
|
|
second_pool_address,
|
|
third_token_in_address,
|
|
third_token_out_address,
|
|
third_pool_address,
|
|
fourth_token_in_address,
|
|
fourth_token_out_address,
|
|
first_converter_address,
|
|
] = get_addresses(16)
|
|
|
|
first_token_in_amount = 10
|
|
first_token_out_amount = 20
|
|
second_token_in_amount = 30
|
|
second_token_out_amount = 40
|
|
third_token_in_amount = 50
|
|
third_token_out_amount = 60
|
|
fourth_token_in_amount = 70
|
|
fourth_token_out_amount = 80
|
|
|
|
traces = [
|
|
make_unknown_trace(block_number, first_transaction_hash, []),
|
|
make_transfer_trace(
|
|
block_number,
|
|
first_transaction_hash,
|
|
trace_address=[0],
|
|
from_address=alice_address,
|
|
to_address=first_pool_address,
|
|
token_address=first_token_in_address,
|
|
amount=first_token_in_amount,
|
|
),
|
|
make_swap_trace(
|
|
block_number,
|
|
first_transaction_hash,
|
|
trace_address=[1],
|
|
from_address=alice_address,
|
|
contract_address=first_pool_address,
|
|
abi_name=UNISWAP_V2_PAIR_ABI_NAME,
|
|
protocol=Protocol.uniswap_v2,
|
|
function_signature="swap(uint256,uint256,address,bytes)",
|
|
recipient_address=bob_address,
|
|
recipient_input_key="to",
|
|
),
|
|
make_transfer_trace(
|
|
block_number,
|
|
first_transaction_hash,
|
|
trace_address=[1, 0],
|
|
from_address=first_pool_address,
|
|
to_address=bob_address,
|
|
token_address=first_token_out_address,
|
|
amount=first_token_out_amount,
|
|
),
|
|
make_swap_trace(
|
|
block_number,
|
|
second_transaction_hash,
|
|
trace_address=[],
|
|
from_address=bob_address,
|
|
contract_address=second_pool_address,
|
|
abi_name=UNISWAP_V3_POOL_ABI_NAME,
|
|
protocol=Protocol.uniswap_v3,
|
|
function_signature="swap(address,bool,int256,uint160,bytes)",
|
|
recipient_address=carl_address,
|
|
recipient_input_key="recipient",
|
|
),
|
|
make_transfer_trace(
|
|
block_number,
|
|
second_transaction_hash,
|
|
trace_address=[0],
|
|
from_address=second_pool_address,
|
|
to_address=carl_address,
|
|
token_address=second_token_out_address,
|
|
amount=second_token_out_amount,
|
|
),
|
|
make_transfer_trace(
|
|
block_number,
|
|
second_transaction_hash,
|
|
trace_address=[1],
|
|
from_address=bob_address,
|
|
to_address=second_pool_address,
|
|
token_address=second_token_in_address,
|
|
amount=second_token_in_amount,
|
|
),
|
|
make_transfer_trace(
|
|
block_number,
|
|
third_transaction_hash,
|
|
trace_address=[6, 0],
|
|
from_address=bob_address,
|
|
to_address=third_pool_address,
|
|
token_address=third_token_in_address,
|
|
amount=third_token_in_amount,
|
|
),
|
|
make_transfer_trace(
|
|
block_number,
|
|
third_transaction_hash,
|
|
trace_address=[6, 1],
|
|
from_address=third_pool_address,
|
|
to_address=bob_address,
|
|
token_address=third_token_out_address,
|
|
amount=third_token_out_amount,
|
|
),
|
|
make_swap_trace(
|
|
block_number,
|
|
third_transaction_hash,
|
|
trace_address=[6],
|
|
from_address=bob_address,
|
|
contract_address=third_pool_address,
|
|
abi_name=BALANCER_V1_POOL_ABI_NAME,
|
|
protocol=Protocol.balancer_v1,
|
|
function_signature="swapExactAmountIn(address,uint256,address,uint256,uint256)",
|
|
recipient_address=bob_address,
|
|
recipient_input_key="recipient",
|
|
),
|
|
make_transfer_trace(
|
|
block_number,
|
|
fourth_transaction_hash,
|
|
trace_address=[2],
|
|
from_address=danielle_address,
|
|
to_address=first_converter_address,
|
|
token_address=fourth_token_in_address,
|
|
amount=fourth_token_in_amount,
|
|
),
|
|
make_transfer_trace(
|
|
block_number,
|
|
fourth_transaction_hash,
|
|
trace_address=[1, 2],
|
|
from_address=first_converter_address,
|
|
to_address=danielle_address,
|
|
token_address=fourth_token_out_address,
|
|
amount=fourth_token_out_amount,
|
|
),
|
|
make_swap_trace(
|
|
block_number,
|
|
fourth_transaction_hash,
|
|
trace_address=[],
|
|
from_address=danielle_address,
|
|
contract_address=BANCOR_NETWORK_CONTRACT_ADDRESS,
|
|
abi_name=BANCOR_NETWORK_ABI_NAME,
|
|
protocol=Protocol.bancor,
|
|
function_signature="convertByPath(address[],uint256,uint256,address,address,uint256)",
|
|
recipient_address=danielle_address,
|
|
recipient_input_key="recipient",
|
|
),
|
|
]
|
|
|
|
swaps = get_swaps(traces)
|
|
|
|
assert len(swaps) == 4
|
|
|
|
for swap in swaps:
|
|
if swap.abi_name == UNISWAP_V2_PAIR_ABI_NAME:
|
|
uni_v2_swap = swap
|
|
elif swap.abi_name == UNISWAP_V3_POOL_ABI_NAME:
|
|
uni_v3_swap = swap
|
|
elif swap.abi_name == BALANCER_V1_POOL_ABI_NAME:
|
|
bal_v1_swap = swap
|
|
elif swap.abi_name == BANCOR_NETWORK_ABI_NAME:
|
|
bancor_swap = swap
|
|
else:
|
|
assert False
|
|
|
|
assert uni_v2_swap.abi_name == UNISWAP_V2_PAIR_ABI_NAME
|
|
assert uni_v2_swap.transaction_hash == first_transaction_hash
|
|
assert uni_v2_swap.block_number == block_number
|
|
assert uni_v2_swap.trace_address == [1]
|
|
assert uni_v2_swap.protocol == Protocol.uniswap_v2
|
|
assert uni_v2_swap.contract_address == first_pool_address
|
|
assert uni_v2_swap.from_address == alice_address
|
|
assert uni_v2_swap.to_address == bob_address
|
|
assert uni_v2_swap.token_in_address == first_token_in_address
|
|
assert uni_v2_swap.token_in_amount == first_token_in_amount
|
|
assert uni_v2_swap.token_out_address == first_token_out_address
|
|
assert uni_v2_swap.token_out_amount == first_token_out_amount
|
|
|
|
assert uni_v3_swap.abi_name == UNISWAP_V3_POOL_ABI_NAME
|
|
assert uni_v3_swap.transaction_hash == second_transaction_hash
|
|
assert uni_v3_swap.block_number == block_number
|
|
assert uni_v3_swap.trace_address == []
|
|
assert uni_v3_swap.protocol == Protocol.uniswap_v3
|
|
assert uni_v3_swap.contract_address == second_pool_address
|
|
assert uni_v3_swap.from_address == bob_address
|
|
assert uni_v3_swap.to_address == carl_address
|
|
assert uni_v3_swap.token_in_address == second_token_in_address
|
|
assert uni_v3_swap.token_in_amount == second_token_in_amount
|
|
assert uni_v3_swap.token_out_address == second_token_out_address
|
|
assert uni_v3_swap.token_out_amount == second_token_out_amount
|
|
|
|
assert bal_v1_swap.abi_name == BALANCER_V1_POOL_ABI_NAME
|
|
assert bal_v1_swap.transaction_hash == third_transaction_hash
|
|
assert bal_v1_swap.block_number == block_number
|
|
assert bal_v1_swap.trace_address == [6]
|
|
assert bal_v1_swap.protocol == Protocol.balancer_v1
|
|
assert bal_v1_swap.contract_address == third_pool_address
|
|
assert bal_v1_swap.from_address == bob_address
|
|
assert bal_v1_swap.to_address == bob_address
|
|
assert bal_v1_swap.token_in_address == third_token_in_address
|
|
assert bal_v1_swap.token_in_amount == third_token_in_amount
|
|
assert bal_v1_swap.token_out_address == third_token_out_address
|
|
assert bal_v1_swap.token_out_amount == third_token_out_amount
|
|
|
|
assert bancor_swap.abi_name == BANCOR_NETWORK_ABI_NAME
|
|
assert bancor_swap.transaction_hash == fourth_transaction_hash
|
|
assert bancor_swap.block_number == block_number
|
|
assert bancor_swap.trace_address == []
|
|
assert bancor_swap.protocol == Protocol.bancor
|
|
assert bancor_swap.contract_address == BANCOR_NETWORK_CONTRACT_ADDRESS
|
|
assert bancor_swap.from_address == danielle_address
|
|
assert bancor_swap.to_address == danielle_address
|
|
assert bancor_swap.token_in_address == fourth_token_in_address
|
|
assert bancor_swap.token_in_amount == fourth_token_in_amount
|
|
assert bancor_swap.token_out_address == fourth_token_out_address
|
|
assert bancor_swap.token_out_amount == fourth_token_out_amount
|