From a46e1e4a5737d5a28b02cd980c048ffa9c95d1e0 Mon Sep 17 00:00:00 2001
From: Florian Pellissier <111426680+flopell@users.noreply.github.com>
Date: Fri, 2 Aug 2024 17:12:19 +0200
Subject: [PATCH] fix(curve): fix sUSD pool

---
 .../integration_test.tycho.yaml               | 44 ++++++++++++++++++-
 substreams/ethereum-curve/src/consts.rs       |  2 +
 substreams/ethereum-curve/src/modules.rs      | 35 ++++++++++-----
 3 files changed, 70 insertions(+), 11 deletions(-)

diff --git a/substreams/ethereum-curve/integration_test.tycho.yaml b/substreams/ethereum-curve/integration_test.tycho.yaml
index e985105..8ff02ac 100644
--- a/substreams/ethereum-curve/integration_test.tycho.yaml
+++ b/substreams/ethereum-curve/integration_test.tycho.yaml
@@ -33,6 +33,48 @@ tests:
           creation_tx: "0xfac67ecbd423a5b915deff06045ec9343568edaec34ae95c43d35f2c018afdaa"
           skip_simulation: false
 
+  - name: test_tricrypto2
+    start_block: 12821148
+    stop_block: 12831387
+    expected_state:
+      protocol_components:
+        - id: "0xd51a44d3fae010294c616388b506acda1bfaae46"
+          tokens:
+            - "0xdac17f958d2ee523a2206206994597c13d831ec7"
+            - "0x2260fac5e5542a773aa44fbcfedf7c193bc2c599"
+            - "0xc02aaa39b223fe8d0a0e5c4f27ead9083c756cc2"
+          static_attributes:
+          creation_tx: "0xdafb6385ed988ce8aacecfe1d97b38ea5e60b1ebce74d2423f71ddd621680138"
+          skip_simulation: true #TODO: investigate this failure
+
+  - name: test_susd
+    start_block: 9906598
+    stop_block: 9907338
+    expected_state:
+      protocol_components:
+        - id: "0xa5407eae9ba41422680e2e00537571bcc53efbfd"
+          tokens:
+            - "0x6b175474e89094c44da98b954eedeac495271d0f"
+            - "0xa0b86991c6218b36c1d19d4a2e9eb0ce3606eb48"
+            - "0xdac17f958d2ee523a2206206994597c13d831ec7"
+            - "0x57ab1ec28d129707052df4df418d58a2d46d5f51"
+          static_attributes:
+          creation_tx: "0x51aca4a03a395de8855fa2ca59b7febe520c2a223e69c502066162f7c1a95ec2"
+          skip_simulation: false
+
+  - name: test_fraxusdc
+    start_block: 14939588
+    stop_block: 14939712
+    expected_state:
+      protocol_components:
+        - id: "0xdcef968d416a41cdac0ed8702fac8128a64241a2"
+          tokens:
+            - "0x853d955acef822db058eb8505911ed77f175b99e"
+            - "0xa0b86991c6218b36c1d19d4a2e9eb0ce3606eb48"
+          static_attributes:
+          creation_tx: "0x1f4254004ce9e19d4eb742ee5a69d30f29085902d976f73e97c44150225ef775"
+          skip_simulation: false
+
   - name: test_crypto_swap_ng_factory_plain_pool
     start_block: 18580701
     stop_block: 18614742
@@ -106,7 +148,7 @@ tests:
     stop_block: 19163633
     expected_state:
       protocol_components:
-        - id: "0x71db3764d6841d8b01dc27c0fd4a66a8a34b2be0"
+        - id: "0x71db3764d6841d8b01dc27c0fd4a66a8a34b2be0" #TODO: ADD TEST THAT USE WETH
           tokens:
             - "0x04c154b66cb340f3ae24111cc767e0184ed00cc6"
             - "0x4591dbff62656e7859afe5e45f6f47d3669fbb28"
diff --git a/substreams/ethereum-curve/src/consts.rs b/substreams/ethereum-curve/src/consts.rs
index 50181a8..7fc8ae8 100644
--- a/substreams/ethereum-curve/src/consts.rs
+++ b/substreams/ethereum-curve/src/consts.rs
@@ -15,3 +15,5 @@ pub const STABLESWAP_FACTORY: [u8; 20] = hex!("4F8846Ae9380B90d2E71D5e3D042dff3E
 // Important addresses
 pub const WETH_ADDRESS: [u8; 20] = hex!("C02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2");
 pub const ETH_ADDRESS: [u8; 20] = hex!("EeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE");
+pub const OLD_SUSD: [u8; 20] = hex!("57Ab1E02fEE23774580C119740129eAC7081e9D3");
+pub const NEW_SUSD: [u8; 20] = hex!("57ab1ec28d129707052df4df418d58a2d46d5f51");
diff --git a/substreams/ethereum-curve/src/modules.rs b/substreams/ethereum-curve/src/modules.rs
index 8c63055..b1d13e6 100644
--- a/substreams/ethereum-curve/src/modules.rs
+++ b/substreams/ethereum-curve/src/modules.rs
@@ -13,7 +13,7 @@ use substreams::{
 use substreams_ethereum::pb::eth;
 
 use crate::{
-    consts::{CRYPTO_SWAP_NG_FACTORY, TRICRYPTO_FACTORY},
+    consts::{CRYPTO_SWAP_NG_FACTORY, NEW_SUSD, OLD_SUSD, TRICRYPTO_FACTORY},
     pool_changes::emit_eth_deltas,
     pool_factories,
     pools::emit_specific_pools,
@@ -157,15 +157,30 @@ pub fn map_relative_balances(
                 .flat_map(|tx| {
                     emit_eth_deltas(tx, &tokens_store)
                         .into_iter()
-                        .chain(extract_balance_deltas_from_tx(tx, |token, transactor| {
-                            let pool_key = format!("pool:{}", hex::encode(transactor));
-                            if let Some(tokens) = tokens_store.get_last(pool_key) {
-                                let token_id = hex::encode(token);
-                                tokens.split(':').any(|t| t == token_id)
-                            } else {
-                                false
-                            }
-                        }))
+                        .chain(
+                            extract_balance_deltas_from_tx(tx, |token, transactor| {
+                                let pool_key = format!("pool:{}", hex::encode(transactor));
+                                if let Some(tokens) = tokens_store.get_last(pool_key) {
+                                    let token_id;
+                                    if token == OLD_SUSD {
+                                        token_id = hex::encode(NEW_SUSD);
+                                    } else {
+                                        token_id = hex::encode(token);
+                                    }
+                                    tokens.split(':').any(|t| t == token_id)
+                                } else {
+                                    false
+                                }
+                            })
+                            .into_iter()
+                            .map(|mut balance| {
+                                if balance.token == OLD_SUSD {
+                                    balance.token = NEW_SUSD.into();
+                                }
+                                balance
+                            })
+                            .collect::<Vec<_>>(),
+                        )
                 })
                 .collect();