From 7656c0d76cf501c3178b01f5854d7f63519880a3 Mon Sep 17 00:00:00 2001
From: Gui Heise <gheise@ufl.edu>
Date: Tue, 23 Nov 2021 14:34:26 -0500
Subject: [PATCH] Remove children swaps

---
 mev_inspect/classifiers/specs/zero_ex.py |  1 +
 tests/test_0x.py                         | 20 ++++++++++----------
 2 files changed, 11 insertions(+), 10 deletions(-)

diff --git a/mev_inspect/classifiers/specs/zero_ex.py b/mev_inspect/classifiers/specs/zero_ex.py
index b84de26..7f2a8ff 100644
--- a/mev_inspect/classifiers/specs/zero_ex.py
+++ b/mev_inspect/classifiers/specs/zero_ex.py
@@ -163,6 +163,7 @@ ZEROX_GENERIC_SPECS = [
     ClassifierSpec(
         abi_name="INativeOrdersFeature",
         protocol=Protocol.zero_ex,
+        valid_contract_addresses=["0xdef1c0ded9bec7f1a1670819833240f027b25eff"],
         classifiers={
             "fillOrKillLimitOrder((address,address,uint128,uint128,uint128,address,address,address,address,bytes32,uint64,uint256),(uint8,uint8,bytes32,bytes32),uint128)": ZeroExSwapClassifier,
             "fillRfqOrder((address,address,uint128,uint128,address,address,address,bytes32,uint64,uint256),(uint8,uint8,bytes32,bytes32),uint128)": ZeroExSwapClassifier,
diff --git a/tests/test_0x.py b/tests/test_0x.py
index 1b62086..bd8d38c 100644
--- a/tests/test_0x.py
+++ b/tests/test_0x.py
@@ -48,10 +48,10 @@ def test__fillLimitOrder_swap():
         abi_name="INativeOrdersFeature",
         transaction_hash=transaction_hash,
         block_number=block_number,
-        trace_address=[0, 1, 0],
-        contract_address="0x1fcc3e6f76f7a96cd2b9d09f1d3c041ca1403c57",
+        trace_address=[0, 1],
+        contract_address="0xdef1c0ded9bec7f1a1670819833240f027b25eff",
         from_address="0xdef1c0ded9bec7f1a1670819833240f027b25eff",
-        to_address="0x1fcc3e6f76f7a96cd2b9d09f1d3c041ca1403c57",
+        to_address="0xdef1c0ded9bec7f1a1670819833240f027b25eff",
         token_in_address="0xa0b86991c6218b36c1d19d4a2e9eb0ce3606eb48",
         token_in_amount=30000000,
         token_out_address="0x9ff79c75ae2bcbe0ec63c0375a3ec90ff75bbe0f",
@@ -80,10 +80,10 @@ def test_RfqLimitOrder_swap():
         abi_name="INativeOrdersFeature",
         transaction_hash=transaction_hash,
         block_number=block_number,
-        trace_address=[0, 1, 13, 0, 1, 0],
-        contract_address="0x1fcc3e6f76f7a96cd2b9d09f1d3c041ca1403c57",
-        from_address="0xdef1c0ded9bec7f1a1670819833240f027b25eff",
-        to_address="0x1fcc3e6f76f7a96cd2b9d09f1d3c041ca1403c57",
+        trace_address=[0, 1, 13, 0, 1],
+        contract_address="0xdef1c0ded9bec7f1a1670819833240f027b25eff",
+        from_address="0xdef171fe48cf0115b1d80b88dc8eab59176fee57",
+        to_address="0xdef1c0ded9bec7f1a1670819833240f027b25eff",
         token_in_address="0xa0b86991c6218b36c1d19d4a2e9eb0ce3606eb48",
         token_in_amount=288948250430,
         token_out_address="0xc02aaa39b223fe8d0a0e5c4f27ead9083c756cc2",
@@ -112,10 +112,10 @@ def test__RfqLimitOrder_swap():
         abi_name="INativeOrdersFeature",
         transaction_hash=transaction_hash,
         block_number=block_number,
-        trace_address=[1, 0, 1, 0, 1, 0],
-        contract_address="0x1fcc3e6f76f7a96cd2b9d09f1d3c041ca1403c57",
+        trace_address=[1, 0, 1, 0, 1],
+        contract_address="0xdef1c0ded9bec7f1a1670819833240f027b25eff",
         from_address="0xdef1c0ded9bec7f1a1670819833240f027b25eff",
-        to_address="0x1fcc3e6f76f7a96cd2b9d09f1d3c041ca1403c57",
+        to_address="0xdef1c0ded9bec7f1a1670819833240f027b25eff",
         token_in_address="0xc02aaa39b223fe8d0a0e5c4f27ead9083c756cc2",
         token_in_amount=979486121594935552,
         token_out_address="0x95ad61b0a150d79219dcf64e1e6cc01f0b64c4ce",