test: add swapExactSyForPt, swapExactYtForSy handlers

primitivefinance · Jun 21, 2024 · 8bd62ef · 8bd62ef
1 parent 8d27bd4
commit 8bd62ef
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 2 deletions.
diff --git a/test/invariant/RMMHandler.sol b/test/invariant/RMMHandler.sol
@@ -171,7 +171,19 @@ contract RMMHandler is CommonBase, StdUtils, StdCheats {
         vm.stopPrank();
     }
 
-    function swapExactSyForPt() public createActor countCall(this.swapExactSyForPt.selector) {}
+    function swapExactSyForPt() public createActor countCall(this.swapExactSyForPt.selector) {
+        deal(address(SY), currentActor, 1 ether);
+        vm.startPrank(currentActor);
+        SY.approve(address(rmm), 1 ether);
+        rmm.swapExactSyForPt(1 ether, 0, address(currentActor));
+        vm.stopPrank();
+    }
 
-    function swapExactYtForSy() public createActor countCall(this.swapExactYtForSy.selector) {}
+    function swapExactYtForSy() public createActor countCall(this.swapExactYtForSy.selector) {
+        deal(address(YT), currentActor, 1 ether);
+        vm.startPrank(currentActor);
+        YT.approve(address(rmm), 1 ether);
+        rmm.swapExactYtForSy(1 ether, 0, address(currentActor));
+        vm.stopPrank();
+    }
 }
diff --git a/test/invariant/RMMInvariants.t.sol b/test/invariant/RMMInvariants.t.sol
@@ -44,6 +44,9 @@ contract RMMInvariantsTest is SetUp {
         console.log("SwapExactPtForSy: ", handler.calls(RMMHandler.swapExactPtForSy.selector));
     }
 
+    /// forge-config: default.invariant.runs = 100
+    /// forge-config: default.invariant.depth = 2
+    /// forge-config: default.invariant.fail-on-revert = false
     function invariant_TradingFunction() public {
         IPYieldToken YT = handler.YT();
         PYIndex index = YT.newIndex();