org

Author: daejunpark

ERC4626.prop.sol

@@ -246,17 +246,6 @@ abstract contract ERC4626Prop is Test {
     // round trip properties
     //
 
-    modifier checkNoFreeProfit(address caller) {
-        uint256 assetsBefore = _getTotalAssets(caller);
-        _;
-        uint256 assetsAfter = _getTotalAssets(caller);
-        assertApproxLeAbs(assetsAfter, assetsBefore, _delta_);
-    }
-
-    function _getTotalAssets(address account) internal returns (uint256) {
-        return vault_convertToAssets(IERC20(_vault_).balanceOf(account)) + IERC20(_underlying_).balanceOf(account);
-    }
-
     // redeem(deposit(a)) <= a
     function prop_RT_deposit_redeem(address caller, uint assets) public checkNoFreeProfit(caller) {
         if (!_vaultMayBeEmpty) vm.assume(IERC20(_vault_).totalSupply() > 0);
@@ -329,6 +318,17 @@ abstract contract ERC4626Prop is Test {
         if (!_skipRoundTripShares) assertApproxLeAbs(shares2, shares1, _delta_);
     }
 
+    modifier checkNoFreeProfit(address caller) {
+        uint256 assetsBefore = _getTotalAssets(caller);
+        _;
+        uint256 assetsAfter = _getTotalAssets(caller);
+        assertApproxLeAbs(assetsAfter, assetsBefore, _delta_);
+    }
+
+    function _getTotalAssets(address account) internal returns (uint256) {
+        return vault_convertToAssets(IERC20(_vault_).balanceOf(account)) + IERC20(_underlying_).balanceOf(account);
+    }
+
     //
     // utils
     //