diff --git a/mev_inspect/aave_liquidations.py b/mev_inspect/aave_liquidations.py index 29ad1ba..9e6b0b0 100644 --- a/mev_inspect/aave_liquidations.py +++ b/mev_inspect/aave_liquidations.py @@ -33,8 +33,6 @@ AAVE_CONTRACT_ADDRESSES: List[str] = [ "0xbcca60bb61934080951369a648fb03df4f96263c", ] -ETH_TOKEN_ADDRESS = "0xeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeee" - def get_aave_liquidations( traces: List[ClassifiedTrace], diff --git a/mev_inspect/schemas/transfers.py b/mev_inspect/schemas/transfers.py index 9719c2b..174e2e0 100644 --- a/mev_inspect/schemas/transfers.py +++ b/mev_inspect/schemas/transfers.py @@ -3,7 +3,7 @@ from typing import List from pydantic import BaseModel -ETH_TOKEN_ADDRESS = "0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE" +ETH_TOKEN_ADDRESS = "0xeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeee" class Transfer(BaseModel): diff --git a/tests/liquidation_test.py b/tests/liquidation_test.py index eaff7dc..bc8860d 100644 --- a/tests/liquidation_test.py +++ b/tests/liquidation_test.py @@ -4,6 +4,7 @@ from mev_inspect.aave_liquidations import get_aave_liquidations from mev_inspect.schemas.liquidations import Liquidation from mev_inspect.schemas.classified_traces import Protocol from mev_inspect.classifiers.trace import TraceClassifier +from mev_inspect.transfers import ETH_TOKEN_ADDRESS from tests.utils import load_test_block @@ -168,11 +169,11 @@ def test_liquidations_with_eth_transfer(): liquidation1 = Liquidation( liquidated_user="0xad346c7762f74c78da86d2941c6eb546e316fbd0", liquidator_user="0x27239549dd40e1d60f5b80b0c4196923745b1fd2", - collateral_token_address="0xeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeee", + collateral_token_address=ETH_TOKEN_ADDRESS, debt_token_address="0x514910771af9ca656af840dff83e8264ecf986ca", debt_purchase_amount=1809152000000000000, received_amount=15636807387264000, - received_token_address="0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE", + received_token_address=ETH_TOKEN_ADDRESS, protocol=Protocol.aave, transaction_hash=transaction_hash, trace_address=[2, 3, 2], @@ -182,11 +183,11 @@ def test_liquidations_with_eth_transfer(): liquidation2 = Liquidation( liquidated_user="0xad346c7762f74c78da86d2941c6eb546e316fbd0", liquidator_user="0x27239549dd40e1d60f5b80b0c4196923745b1fd2", - collateral_token_address="0xeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeee", + collateral_token_address=ETH_TOKEN_ADDRESS, debt_token_address="0x514910771af9ca656af840dff83e8264ecf986ca", debt_purchase_amount=1809152000000000000, received_amount=8995273139160873, - received_token_address="0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE", + received_token_address=ETH_TOKEN_ADDRESS, protocol=Protocol.aave, transaction_hash=transaction_hash, trace_address=[2, 4, 2], diff --git a/tests/test_compound.py b/tests/test_compound.py index 44ca054..a05dece 100644 --- a/tests/test_compound.py +++ b/tests/test_compound.py @@ -2,6 +2,7 @@ from mev_inspect.compound_liquidations import get_compound_liquidations from mev_inspect.schemas.liquidations import Liquidation from mev_inspect.schemas.classified_traces import Protocol from mev_inspect.classifiers.trace import TraceClassifier +from mev_inspect.transfers import ETH_TOKEN_ADDRESS from tests.utils import load_test_block, load_comp_markets, load_cream_markets comp_markets = load_comp_markets() @@ -18,7 +19,7 @@ def test_c_ether_liquidations(): Liquidation( liquidated_user="0xb5535a3681cf8d5431b8acfd779e2f79677ecce9", liquidator_user="0xe0090ec6895c087a393f0e45f1f85098a6c33bef", - collateral_token_address="0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE", + collateral_token_address=ETH_TOKEN_ADDRESS, debt_token_address="0x39aa39c021dfbae8fac545936693ac917d5e7563", debt_purchase_amount=268066492249420078, received_amount=4747650169097, @@ -43,7 +44,7 @@ def test_c_ether_liquidations(): Liquidation( liquidated_user="0x45df6f00166c3fb77dc16b9e47ff57bc6694e898", liquidator_user="0xe0090ec6895c087a393f0e45f1f85098a6c33bef", - collateral_token_address="0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE", + collateral_token_address=ETH_TOKEN_ADDRESS, debt_token_address="0x35a18000230da775cac24873d00ff85bccded550", debt_purchase_amount=414547860568297082, received_amount=321973320649, @@ -69,7 +70,7 @@ def test_c_ether_liquidations(): Liquidation( liquidated_user="0xacbcf5d2970eef25f02a27e9d9cd31027b058b9b", liquidator_user="0xe0090ec6895c087a393f0e45f1f85098a6c33bef", - collateral_token_address="0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE", + collateral_token_address=ETH_TOKEN_ADDRESS, debt_token_address="0x35a18000230da775cac24873d00ff85bccded550", debt_purchase_amount=1106497772527562662, received_amount=910895850496,