tests: try to withdraw twice

AngleProtocol · Apr 15, 2024 · dae57c9 · dae57c9
1 parent f7680c9
commit dae57c9
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/test/fuzz/EmergencyWithdrawal.t.sol b/test/fuzz/EmergencyWithdrawal.t.sol
@@ -31,10 +31,12 @@ contract EmergencyWithdrawalFuzz is Fixture {
         for (uint i = 0; i < accounts.length; i++) {
             assertEq(Angle.balanceOf(accounts[i]), 0);
 
-            vm.startPrank(accounts[i]);
+            vm.prank(accounts[i]);
             veANGLE.withdraw_fast();
-            vm.stopPrank();
+            assertEq(Angle.balanceOf(accounts[i]), balances[i]);
 
+            vm.prank(accounts[i]);
+            veANGLE.withdraw_fast();
             assertEq(Angle.balanceOf(accounts[i]), balances[i]);
         }
     }
@@ -72,10 +74,12 @@ contract EmergencyWithdrawalFuzz is Fixture {
 
             vm.warp(block.timestamp + timeWraps[i]);
 
-            vm.startPrank(accounts[i]);
+            vm.prank(accounts[i]);
             veANGLE.withdraw_fast();
-            vm.stopPrank();
+            assertEq(Angle.balanceOf(accounts[i]), balances[i]);
 
+            vm.prank(accounts[i]);
+            veANGLE.withdraw_fast();
             assertEq(Angle.balanceOf(accounts[i]), balances[i]);
         }
     }