Merge pull request #252 from flashbots/compound-eth

Fix Compound cETH liquidations
This commit is contained in:
Gui Heise 2022-01-27 22:04:23 -05:00 committed by GitHub
commit ad472d9d23
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 7 deletions

View File

@ -10,7 +10,7 @@ from mev_inspect.schemas.classifiers import (
SeizeClassifier,
)
from mev_inspect.schemas.liquidations import Liquidation
from mev_inspect.schemas.prices import CETH_TOKEN_ADDRESS
from mev_inspect.schemas.prices import CETH_TOKEN_ADDRESS, ETH_TOKEN_ADDRESS
from mev_inspect.schemas.traces import Protocol
from mev_inspect.schemas.transfers import Transfer
@ -32,10 +32,10 @@ class CompoundLiquidationClassifier(LiquidationClassifier):
debt_purchase_amount = None
received_amount = None
debt_purchase_amount = (
liquidation_trace.value
debt_purchase_amount, debt_token_address = (
(liquidation_trace.value, ETH_TOKEN_ADDRESS)
if debt_token_address == CETH_TOKEN_ADDRESS and liquidation_trace.value != 0
else liquidation_trace.inputs["repayAmount"]
else (liquidation_trace.inputs["repayAmount"], CETH_TOKEN_ADDRESS)
)
debt_transfer = get_debt_transfer(liquidator, child_transfers)

View File

@ -1,6 +1,7 @@
from mev_inspect.classifiers.trace import TraceClassifier
from mev_inspect.liquidations import get_liquidations
from mev_inspect.schemas.liquidations import Liquidation
from mev_inspect.schemas.prices import ETH_TOKEN_ADDRESS
from mev_inspect.schemas.traces import Protocol
from tests.utils import load_comp_markets, load_cream_markets, load_test_block
@ -18,7 +19,7 @@ def test_c_ether_liquidations(trace_classifier: TraceClassifier):
Liquidation(
liquidated_user="0xb5535a3681cf8d5431b8acfd779e2f79677ecce9",
liquidator_user="0xe0090ec6895c087a393f0e45f1f85098a6c33bef",
debt_token_address="0x4ddc2d193948926d02f9b1fe9e1daa0718270ed5",
debt_token_address=ETH_TOKEN_ADDRESS,
debt_purchase_amount=268066492249420078,
received_amount=4747650169097,
received_token_address="0x39aa39c021dfbae8fac545936693ac917d5e7563",
@ -44,7 +45,7 @@ def test_c_ether_liquidations(trace_classifier: TraceClassifier):
Liquidation(
liquidated_user="0x45df6f00166c3fb77dc16b9e47ff57bc6694e898",
liquidator_user="0xe0090ec6895c087a393f0e45f1f85098a6c33bef",
debt_token_address="0x4ddc2d193948926d02f9b1fe9e1daa0718270ed5",
debt_token_address=ETH_TOKEN_ADDRESS,
debt_purchase_amount=414547860568297082,
received_amount=321973320649,
received_token_address="0x35a18000230da775cac24873d00ff85bccded550",
@ -71,7 +72,7 @@ def test_c_ether_liquidations(trace_classifier: TraceClassifier):
Liquidation(
liquidated_user="0xacbcf5d2970eef25f02a27e9d9cd31027b058b9b",
liquidator_user="0xe0090ec6895c087a393f0e45f1f85098a6c33bef",
debt_token_address="0x4ddc2d193948926d02f9b1fe9e1daa0718270ed5",
debt_token_address=ETH_TOKEN_ADDRESS,
debt_purchase_amount=1106497772527562662,
received_amount=910895850496,
received_token_address="0x35a18000230da775cac24873d00ff85bccded550",