test

rainlanguage · Aug 25, 2024 · c8437a2 · c8437a2
1 parent 6c5cc3c
commit c8437a2
Show file tree

Hide file tree

Showing 2 changed files with 70 additions and 2 deletions.
diff --git a/src/tranche/tranche-space.rain b/src/tranche/tranche-space.rain
@@ -186,7 +186,7 @@ scenarios:
                 # selling RED for it, then the amount is the INPUT.
                 amount-is-output: 0
                 io-ratio-expr: '''linear-growth'
-                io-ratio-base: 10
+                io-ratio-base: 9
                 io-ratio-growth: 0.5
                 tranche-size-expr: '''exponential-growth'
                 tranche-size-base: 1
@@ -201,7 +201,7 @@ scenarios:
                   # If the market is outside the chart then set this to 0.
                   initial-tranche-space: 0
                   post-action: '''post-action-constant'
-                  io-ratio-current: 0.1
+                  io-ratio-current: 10
                   post-action-tranche: '''post-action-tranche-linear'
                 scenarios:
                   prod:

diff --git a/test/TrancheSpaceTest.t.sol b/test/TrancheSpaceTest.t.sol
@@ -58,6 +58,74 @@ contract TrancheSpaceTest is StrategyTests {
         iOrderBook = IOrderBookV4(address(uint160(bytes20(orderBook))));
     }
 
+    function testSuccessiveInitTranches() public {
+
+        LibStrategyDeployment.StrategyDeploymentV4 memory strategy = LibStrategyDeployment.StrategyDeploymentV4(
+            "",
+            "",
+            0,
+            0,
+            10e18,
+            10e18,
+            0,
+            0,
+            "src/tranche/tranche-space.rain",
+            "src/settings.yml",
+            "",
+            "flare-sell",
+            "./lib/h20.test-std/lib/rain.orderbook",
+            "./lib/h20.test-std/lib/rain.orderbook/Cargo.toml"
+        );
+
+        OrderV3 memory order = addOrderDepositOutputTokens(strategy);
+
+
+        // Tranche 0
+        {
+            vm.recordLogs();
+            takeExternalOrder(order, strategy.inputTokenIndex, strategy.outputTokenIndex, new SignedContextV1[](0));
+
+            Vm.Log[] memory entries = vm.getRecordedLogs();
+            (uint256 strategyAmount, uint256 strategyRatio) = getCalculationContext(entries);
+
+            uint256 expectedTrancheAmount = 102009999999999997;
+            uint256 expectedTrancheRatio = 10e18;
+
+            assertEq(strategyAmount, expectedTrancheAmount);
+            assertEq(strategyRatio, expectedTrancheRatio);
+        }
+
+        // Tranche 1
+        {
+            vm.recordLogs();
+            takeExternalOrder(order, strategy.inputTokenIndex, strategy.outputTokenIndex, new SignedContextV1[](0));
+
+            Vm.Log[] memory entries = vm.getRecordedLogs();
+            (uint256 strategyAmount, uint256 strategyRatio) = getCalculationContext(entries);
+
+            uint256 expectedTrancheAmount = 98123904761904758;
+            uint256 expectedTrancheRatio = 10.5e18;
+
+            assertEq(strategyAmount, expectedTrancheAmount);
+            assertEq(strategyRatio, expectedTrancheRatio);
+        }
+
+        // Tranche 2
+        {
+            vm.recordLogs();
+            takeExternalOrder(order, strategy.inputTokenIndex, strategy.outputTokenIndex, new SignedContextV1[](0));
+
+            Vm.Log[] memory entries = vm.getRecordedLogs();
+            (uint256 strategyAmount, uint256 strategyRatio) = getCalculationContext(entries);
+
+            uint256 expectedTrancheAmount = 94600364545454541;
+            uint256 expectedTrancheRatio = 11e18;
+
+            assertEq(strategyAmount, expectedTrancheAmount);
+            assertEq(strategyRatio, expectedTrancheRatio);
+        }
+    }
+
     function testTrancheSpaceShynessExternalFlare() public {
 
         uint256 expectedRatio = 1e18;