From d7bb160d853d9d9aae7c6ec343d974d8257fcd1f Mon Sep 17 00:00:00 2001 From: Taarush Vemulapalli Date: Thu, 16 Dec 2021 14:33:10 -0500 Subject: [PATCH] Add received_token_address for Compound/CREAM (#168) --- mev_inspect/compound_liquidations.py | 2 ++ tests/test_compound.py | 5 +++++ 2 files changed, 7 insertions(+) diff --git a/mev_inspect/compound_liquidations.py b/mev_inspect/compound_liquidations.py index 5fcae87..df45420 100644 --- a/mev_inspect/compound_liquidations.py +++ b/mev_inspect/compound_liquidations.py @@ -44,6 +44,7 @@ def get_compound_liquidations( debt_purchase_amount=trace.value, protocol=trace.protocol, received_amount=seize_trace.inputs["seizeTokens"], + received_token_address=trace.to_address, transaction_hash=trace.transaction_hash, trace_address=trace.trace_address, block_number=trace.block_number, @@ -60,6 +61,7 @@ def get_compound_liquidations( debt_purchase_amount=trace.inputs["repayAmount"], protocol=trace.protocol, received_amount=seize_trace.inputs["seizeTokens"], + received_token_address=trace.to_address, transaction_hash=trace.transaction_hash, trace_address=trace.trace_address, block_number=trace.block_number, diff --git a/tests/test_compound.py b/tests/test_compound.py index f7f5206..1a1aa65 100644 --- a/tests/test_compound.py +++ b/tests/test_compound.py @@ -21,6 +21,7 @@ def test_c_ether_liquidations(trace_classifier: TraceClassifier): debt_token_address="0x39aa39c021dfbae8fac545936693ac917d5e7563", debt_purchase_amount=268066492249420078, received_amount=4747650169097, + received_token_address="0x4ddc2d193948926d02f9b1fe9e1daa0718270ed5", protocol=Protocol.compound_v2, transaction_hash=transaction_hash, trace_address=[1], @@ -44,6 +45,7 @@ def test_c_ether_liquidations(trace_classifier: TraceClassifier): debt_token_address="0x35a18000230da775cac24873d00ff85bccded550", debt_purchase_amount=414547860568297082, received_amount=321973320649, + received_token_address="0x4ddc2d193948926d02f9b1fe9e1daa0718270ed5", protocol=Protocol.compound_v2, transaction_hash=transaction_hash, trace_address=[1], @@ -68,6 +70,7 @@ def test_c_ether_liquidations(trace_classifier: TraceClassifier): debt_token_address="0x35a18000230da775cac24873d00ff85bccded550", debt_purchase_amount=1106497772527562662, received_amount=910895850496, + received_token_address="0x4ddc2d193948926d02f9b1fe9e1daa0718270ed5", protocol=Protocol.compound_v2, transaction_hash=transaction_hash, trace_address=[1], @@ -93,6 +96,7 @@ def test_c_token_liquidation(trace_classifier: TraceClassifier): debt_token_address="0x70e36f6bf80a52b3b46b3af8e106cc0ed743e8e4", debt_purchase_amount=1207055531, received_amount=21459623305, + received_token_address="0x39aa39c021dfbae8fac545936693ac917d5e7563", protocol=Protocol.compound_v2, transaction_hash=transaction_hash, trace_address=[1], @@ -118,6 +122,7 @@ def test_cream_token_liquidation(trace_classifier: TraceClassifier): debt_token_address="0x44fbebd2f576670a6c33f6fc0b00aa8c5753b322", debt_purchase_amount=14857434973806369550, received_amount=1547215810826, + received_token_address="0x697256caa3ccafd62bb6d3aa1c7c5671786a5fd9", protocol=Protocol.cream, transaction_hash=transaction_hash, trace_address=[],