From 4dbcb59b4d8198fc2d6a8bfcb6f34f371a4faa5b Mon Sep 17 00:00:00 2001
From: Gui Heise <gheise@ufl.edu>
Date: Wed, 29 Sep 2021 10:52:58 -0400
Subject: [PATCH] Add trace address to liquidations

---
 mev_inspect/aave_liquidations.py    | 3 ++-
 mev_inspect/schemas/liquidations.py | 2 ++
 tests/liquidation_test.py           | 5 +++++
 3 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/mev_inspect/aave_liquidations.py b/mev_inspect/aave_liquidations.py
index 34d0b75..3ed5df5 100644
--- a/mev_inspect/aave_liquidations.py
+++ b/mev_inspect/aave_liquidations.py
@@ -63,10 +63,11 @@ def get_aave_liquidations(
                     protocol=Protocol.aave,
                     received_amount=received_amount,
                     transaction_hash=trace.transaction_hash,
+                    trace_address=trace.trace_address,
                     block_number=trace.block_number,
                 )
             )
-
+    print(liquidations)
     return liquidations
 
 
diff --git a/mev_inspect/schemas/liquidations.py b/mev_inspect/schemas/liquidations.py
index 0f8b183..fad090b 100644
--- a/mev_inspect/schemas/liquidations.py
+++ b/mev_inspect/schemas/liquidations.py
@@ -1,3 +1,4 @@
+from typing import List
 from pydantic import BaseModel
 from mev_inspect.schemas.classified_traces import Protocol
 
@@ -11,4 +12,5 @@ class Liquidation(BaseModel):
     received_amount: int
     protocol: Protocol
     transaction_hash: str
+    trace_address: List[int]
     block_number: str
diff --git a/tests/liquidation_test.py b/tests/liquidation_test.py
index 83b2458..62c58e4 100644
--- a/tests/liquidation_test.py
+++ b/tests/liquidation_test.py
@@ -24,6 +24,7 @@ def test_single_weth_liquidation():
             received_amount=8182733924513576561,
             protocol=Protocol.aave,
             transaction_hash=transaction_hash,
+            trace_address=[1, 1, 6],
             block_number=block_number,
         )
     ]
@@ -53,6 +54,7 @@ def test_single_liquidation():
             received_amount=2657946947610159065393,
             protocol=Protocol.aave,
             transaction_hash=transaction_hash,
+            trace_address=[0, 7, 1, 0, 6],
             block_number=block_number,
         )
     ]
@@ -81,6 +83,7 @@ def test_multiple_liquidations_in_block():
         received_amount=10111753901939162887,
         protocol=Protocol.aave,
         transaction_hash=transaction1,
+        trace_address=[],
         block_number=block_number,
     )
 
@@ -93,6 +96,7 @@ def test_multiple_liquidations_in_block():
         received_amount=21996356316098208090,
         protocol=Protocol.aave,
         transaction_hash=transaction2,
+        trace_address=[],
         block_number=block_number,
     )
 
@@ -105,6 +109,7 @@ def test_multiple_liquidations_in_block():
         received_amount=121531358145247546,
         protocol=Protocol.aave,
         transaction_hash=transaction3,
+        trace_address=[],
         block_number=block_number,
     )