From dc5e9fa81cda89f0baa168d80c4be2b53aeb9857 Mon Sep 17 00:00:00 2001
From: 0xSolDev <narayan@nuts.finance>
Date: Wed, 18 Dec 2024 00:36:34 +0530
Subject: [PATCH] feat: added test for exchange rate tokens

---
 .github/workflows/ci.yml |  1 -
 test/StableAsset.sol     | 75 ++++++++++++++++++++++++++++++++--------
 2 files changed, 61 insertions(+), 15 deletions(-)

diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 6cabe8c..914ce8e 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -1,7 +1,6 @@
 name: "CI"
 
 env:
-  API_KEY_ALCHEMY: ${{ secrets.API_KEY_ALCHEMY }}
   FOUNDRY_PROFILE: "ci"
 
 on:
diff --git a/test/StableAsset.sol b/test/StableAsset.sol
index 2d2883b..e7d23ba 100644
--- a/test/StableAsset.sol
+++ b/test/StableAsset.sol
@@ -11,23 +11,30 @@ import { LPToken } from "../src/LPToken.sol";
 import { WLPToken } from "../src/WLPToken.sol";
 import "@openzeppelin/contracts/proxy/beacon/UpgradeableBeacon.sol";
 import "../src/misc/ConstantExchangeRateProvider.sol";
+import "../src/mock/MockExchangeRateProvider.sol";
 
 contract StableAssetTest is Test {
   address governance = address(0x01);
   uint256 A = 100;
-  LPToken lpToken;
-  StableAsset stableAsset;
+  LPToken lpToken1;
+  StableAsset ethPool1;
   uint256 feeDenominator = 10000000000;
   uint256 mintFee = 10000000;
   uint256 swapFee = 20000000;
   uint256 redeemFee = 50000000;
+  MockToken WETH;
+  MockToken stETH;
+  MockToken rETH;
+  MockToken wstETH;
+  StableAsset ethPool2;
+  LPToken lpToken2;
 
   function setUp() public {
-    MockToken tokenA = new MockToken("test 1", "T1", 18);
-    MockToken tokenB = new MockToken("test 2", "T2", 18);
+    WETH = new MockToken("WETH", "WETH", 18);
+    stETH = new MockToken("stETH", "stETH", 18);
 
-    lpToken = new LPToken();
-    lpToken.initialize(
+    lpToken1 = new LPToken();
+    lpToken1.initialize(
       governance,
       "LP Token",
       "LPT"
@@ -35,11 +42,11 @@ contract StableAssetTest is Test {
 
     ConstantExchangeRateProvider exchangeRateProvider = new ConstantExchangeRateProvider();
 
-    stableAsset = new StableAsset();
+    ethPool1 = new StableAsset();
 
     address[] memory tokens = new address[](2);
-    tokens[0] = address(tokenA);
-    tokens[1] = address(tokenB);
+    tokens[0] = address(WETH);
+    tokens[1] = address(stETH);
 
     uint256[] memory precisions = new uint256[](2);
     precisions[0] = 1;
@@ -54,17 +61,52 @@ contract StableAssetTest is Test {
     exchangeRateProviders[0] = exchangeRateProvider;
     exchangeRateProviders[1] = exchangeRateProvider;
 
-    stableAsset.initialize(
+    ethPool1.initialize(
       tokens,
       precisions,
       fees,
-      lpToken,
+      lpToken1,
       A,
       exchangeRateProviders
     );
 
     vm.prank(governance);
-    lpToken.addPool(address(stableAsset));
+    lpToken1.addPool(address(ethPool1));
+
+    MockExchangeRateProvider rETHExchangeRateProvider = new MockExchangeRateProvider(1.1e18, 18);
+    MockExchangeRateProvider wstETHExchangeRateProvider = new MockExchangeRateProvider(1.2e18, 18);
+
+    rETH = new MockToken("rETH", "rETH", 18);
+    wstETH = new MockToken("wstETH", "wstETH", 18);
+
+    tokens = new address[](2);
+    tokens[0] = address(rETH);
+    tokens[1] = address(wstETH);
+
+    exchangeRateProviders = new IExchangeRateProvider[](2);
+    exchangeRateProviders[0] = IExchangeRateProvider(rETHExchangeRateProvider);
+    exchangeRateProviders[1] = IExchangeRateProvider(wstETHExchangeRateProvider);
+
+    ethPool2 = new StableAsset();
+
+    lpToken2 = new LPToken();
+    lpToken2.initialize(
+      governance,
+      "LP Token",
+      "LPT"
+    );
+
+    ethPool2.initialize(
+      tokens,
+      precisions,
+      fees,
+      lpToken2,
+      A,
+      exchangeRateProviders
+    );
+
+    vm.prank(governance);
+    lpToken2.addPool(address(ethPool2));
   }
 
   function test_CorrectMintAmount_UnequalTokenAmounts() external {
@@ -72,13 +114,18 @@ contract StableAssetTest is Test {
     amounts[0] = 110e18;
     amounts[1] = 90e18;
 
-    (uint256 lpTokensMinted, uint256 feesCharged) = stableAsset.getMintAmount(
+    (uint256 lpTokensMinted, uint256 feesCharged) = ethPool1.getMintAmount(
       amounts
     );
 
-    console.log(lpTokensMinted);
+    assertFee(lpTokensMinted+feesCharged, feesCharged, mintFee);
+
+    (lpTokensMinted, feesCharged) = ethPool2.getMintAmount(
+      amounts
+    );
 
     assertFee(lpTokensMinted+feesCharged, feesCharged, mintFee);
+
   }
 
   function assertFee(uint256 totalAmount, uint256 feeAmount, uint256 fee) internal {