diff --git a/tests/test_arbitrages.py b/tests/test_arbitrages.py index 9920b8a..2248ff3 100644 --- a/tests/test_arbitrages.py +++ b/tests/test_arbitrages.py @@ -14,15 +14,16 @@ def test_two_pool_arbitrage(get_transaction_hashes, get_addresses): account_address, first_pool_address, second_pool_address, + unrelated_pool_address, first_token_address, second_token_address, - ] = get_addresses(5) + ] = get_addresses(6) first_token_in_amount = 10 first_token_out_amount = 10 second_token_amount = 15 - swaps = [ + arb_swaps = [ Swap( abi_name=UNISWAP_V2_PAIR_ABI_NAME, transaction_hash=transaction_hash, @@ -51,6 +52,34 @@ def test_two_pool_arbitrage(get_transaction_hashes, get_addresses): ), ] + unrelated_swap = Swap( + abi_name=UNISWAP_V3_POOL_ABI_NAME, + transaction_hash=transaction_hash, + block_number=block_number, + trace_address=[2, 0], + pool_address=unrelated_pool_address, + from_address=account_address, + to_address=account_address, + token_in_address=second_token_address, + token_in_amount=first_token_in_amount, + token_out_address=first_token_address, + token_out_amount=first_token_out_amount, + ) + + swaps = [ + unrelated_swap, + *arb_swaps, + ] + arbitrages = get_arbitrages(swaps) assert len(arbitrages) == 1 + + arbitrage = arbitrages[0] + + assert arbitrage.swaps == arb_swaps + assert arbitrage.account_address == account_address + assert arbitrage.profit_token_address == first_token_address + assert arbitrage.start_amount == first_token_in_amount + assert arbitrage.end_amount == first_token_out_amount + assert arbitrage.profit_amount == first_token_out_amount - first_token_in_amount