Remove approveTokens method, update test coverage report

brritoxyz · Jan 15, 2024 · 3157ddf · 3157ddf
1 parent c502d55
commit 3157ddf
Show file tree

Hide file tree

Showing 3 changed files with 114 additions and 148 deletions.
diff --git a/lcov.info b/lcov.info
@@ -28,136 +28,133 @@ BRH:0
 end_of_record
 TN:
 SF:src/BrrETH.sol
-FN:78,BrrETH.name
+FN:89,BrrETH.name
 FNDA:1,BrrETH.name
-DA:79,1
-FN:86,BrrETH.symbol
+DA:90,1
+FN:97,BrrETH.symbol
 FNDA:1,BrrETH.symbol
-DA:87,1
-FN:94,BrrETH.asset
+DA:98,1
+FN:105,BrrETH.asset
 FNDA:1,BrrETH.asset
-DA:95,2033
-FN:99,BrrETH.approveTokens
-FNDA:1,BrrETH.approveTokens
-DA:100,102
-DA:104,102
-DA:110,102
-FN:121,BrrETH.convertToShares
+DA:106,1647
+FN:117,BrrETH.convertToShares
 FNDA:334,BrrETH.convertToShares
-DA:129,818
-FN:140,BrrETH._deposit
-FNDA:483,BrrETH._deposit
-DA:146,483
-DA:148,483
-FN:157,BrrETH.deposit
+DA:125,721
+FN:136,BrrETH._deposit
+FNDA:386,BrrETH._deposit
+DA:142,386
+DA:144,386
+FN:153,BrrETH.deposit
 FNDA:272,BrrETH.deposit
+DA:157,272
+DA:159,272
 DA:161,272
 DA:163,272
-DA:165,272
-DA:167,272
-DA:168,272
-DA:170,272
-BRDA:170,0,0,1
-BRDA:170,0,1,271
-DA:172,271
-FN:184,BrrETH.deposit
-FNDA:313,BrrETH.deposit
-DA:189,313
-BRDA:189,1,0,101
-BRDA:189,1,1,212
-DA:190,101
-DA:192,212
-DA:194,212
-DA:196,212
-DA:203,212
-FN:207,BrrETH.harvest
-FNDA:100,BrrETH.harvest
-DA:208,158
-DA:210,158
-DA:212,158
-DA:214,158
-BRDA:214,2,0,-
-BRDA:214,2,1,158
-DA:218,158
-DA:224,158
-DA:235,158
-DA:238,158
-BRDA:238,3,0,-
-BRDA:238,3,1,158
-DA:241,158
-DA:243,158
-DA:245,158
-DA:249,158
-DA:256,158
-DA:258,158
-FN:270,BrrETH.setCometRewards
+DA:164,272
+DA:166,272
+BRDA:166,0,0,1
+BRDA:166,0,1,271
+DA:168,271
+FN:180,BrrETH.deposit
+FNDA:216,BrrETH.deposit
+DA:185,216
+BRDA:185,1,0,101
+BRDA:185,1,1,115
+DA:186,101
+DA:188,115
+DA:190,115
+DA:192,115
+DA:199,115
+FN:203,BrrETH.harvest
+FNDA:4,BrrETH.harvest
+DA:204,62
+DA:206,62
+DA:208,62
+DA:210,62
+BRDA:210,2,0,-
+BRDA:210,2,1,62
+DA:214,62
+DA:220,62
+DA:231,62
+DA:234,62
+BRDA:234,3,0,-
+BRDA:234,3,1,62
+DA:237,62
+DA:239,62
+DA:241,62
+DA:245,62
+DA:252,62
+DA:254,62
+FN:266,BrrETH.setCometRewards
 FNDA:104,BrrETH.setCometRewards
-DA:274,103
-BRDA:274,4,0,1
-BRDA:274,4,1,102
+DA:270,103
+BRDA:270,4,0,1
+BRDA:270,4,1,102
+DA:271,102
+BRDA:271,5,0,58
+BRDA:271,5,1,102
+DA:273,102
 DA:275,102
-BRDA:275,5,0,58
-BRDA:275,5,1,102
-DA:277,102
-DA:279,102
-FN:286,BrrETH.setRouter
+FN:282,BrrETH.setRouter
 FNDA:103,BrrETH.setRouter
-DA:287,102
-BRDA:287,6,0,1
-BRDA:287,6,1,101
+DA:283,102
+BRDA:283,6,0,1
+BRDA:283,6,1,101
+DA:285,101
 DA:289,101
-DA:292,101
+DA:291,101
 DA:294,101
-FN:301,BrrETH.setRewardFee
+DA:296,101
+FN:303,BrrETH.setRewardFee
 FNDA:203,BrrETH.setRewardFee
-DA:302,202
-BRDA:302,7,0,101
-BRDA:302,7,1,101
-DA:304,101
+DA:304,202
+BRDA:304,7,0,101
+BRDA:304,7,1,101
 DA:306,101
-FN:313,BrrETH.setProtocolFeeReceiver
+DA:308,101
+FN:315,BrrETH.setProtocolFeeReceiver
 FNDA:3,BrrETH.setProtocolFeeReceiver
-DA:316,2
-BRDA:316,8,0,1
-BRDA:316,8,1,1
-DA:317,1
+DA:318,2
+BRDA:318,8,0,1
+BRDA:318,8,1,1
 DA:319,1
 DA:321,1
-FN:328,BrrETH.setFeeDistributor
-FNDA:56,BrrETH.setFeeDistributor
-DA:329,55
-BRDA:329,9,0,1
-BRDA:329,9,1,54
-DA:331,54
-DA:333,54
-FN:340,BrrETH.transferOwnership
+DA:323,1
+FN:330,BrrETH.setFeeDistributor
+FNDA:6,BrrETH.setFeeDistributor
+DA:331,5
+BRDA:331,9,0,1
+BRDA:331,9,1,4
+DA:333,4
+DA:335,4
+FN:342,BrrETH.transferOwnership
 FNDA:1,BrrETH.transferOwnership
-DA:341,1
-FN:344,BrrETH.renounceOwnership
+DA:343,1
+FN:346,BrrETH.renounceOwnership
 FNDA:1,BrrETH.renounceOwnership
-DA:345,1
-FN:352,BrrETH.maxMint
+DA:347,1
+FN:354,BrrETH.maxMint
 FNDA:1,BrrETH.maxMint
-DA:353,1
-FN:356,BrrETH.maxWithdraw
+DA:355,1
+FN:358,BrrETH.maxWithdraw
 FNDA:1,BrrETH.maxWithdraw
-DA:357,1
-FN:360,BrrETH.previewMint
+DA:359,1
+FN:362,BrrETH.previewMint
 FNDA:1,BrrETH.previewMint
-DA:361,1
-FN:364,BrrETH.previewWithdraw
+DA:363,1
+FN:366,BrrETH.previewWithdraw
 FNDA:1,BrrETH.previewWithdraw
-DA:365,1
-FN:368,BrrETH.mint
+DA:367,1
+FN:370,BrrETH.mint
 FNDA:1,BrrETH.mint
-DA:369,1
-FN:372,BrrETH.withdraw
+DA:371,1
+FN:374,BrrETH.withdraw
 FNDA:1,BrrETH.withdraw
-DA:377,1
-FNF:22
-FNH:22
-LF:62
-LH:62
+DA:379,1
+FNF:21
+FNH:21
+LF:61
+LH:61
 BRF:20
 BRH:18
 end_of_record

diff --git a/src/BrrETH.sol b/src/BrrETH.sol
@@ -68,7 +68,18 @@ contract BrrETH is Ownable, ERC4626 {
         feeDistributor = initialOwner;
 
         _initializeOwner(initialOwner);
-        approveTokens();
+
+        ICometRewards.RewardConfig memory rewardConfig = cometRewards
+            .rewardConfig(_COMET);
+
+        // Enable the router to swap our Comet rewards for WETH.
+        rewardConfig.token.safeApproveWithRetry(
+            address(router),
+            type(uint256).max
+        );
+
+        // Enable Comet to transfer our WETH in exchange for cWETH.
+        _WETH.safeApproveWithRetry(_COMET, type(uint256).max);
     }
 
     /**
@@ -95,21 +106,6 @@ contract BrrETH is Ownable, ERC4626 {
         return _COMET;
     }
 
-    /// @notice Approve token allowances for vital contracts.
-    function approveTokens() public {
-        ICometRewards.RewardConfig memory rewardConfig = cometRewards
-            .rewardConfig(_COMET);
-
-        // Enable the router to swap our Comet rewards for WETH.
-        rewardConfig.token.safeApproveWithRetry(
-            address(router),
-            type(uint256).max
-        );
-
-        // Enable Comet to transfer our WETH in exchange for cWETH.
-        _WETH.safeApproveWithRetry(_COMET, type(uint256).max);
-    }
-
     /**
      * @notice Returns the amount of shares that the Vault will exchange for the amount of assets provided,
      *         in an ideal scenario where all conditions are met.

diff --git a/test/BrrETH.t.sol b/test/BrrETH.t.sol
@@ -117,33 +117,6 @@ contract BrrETHTest is Helper {
         assertEq(_COMET, vault.asset());
     }
 
-    /*//////////////////////////////////////////////////////////////
-                             approveTokens
-    //////////////////////////////////////////////////////////////*/
-
-    function testApproveTokens() external {
-        vm.startPrank(address(vault));
-
-        _WETH.safeApprove(_COMET, 0);
-        _COMP.safeApprove(_ROUTER, 0);
-
-        vm.stopPrank();
-
-        assertEq(ERC20(_WETH).allowance(address(vault), _COMET), 0);
-        assertEq(ERC20(_COMP).allowance(address(vault), _ROUTER), 0);
-
-        vault.approveTokens();
-
-        assertEq(
-            ERC20(_WETH).allowance(address(vault), _COMET),
-            type(uint256).max
-        );
-        assertEq(
-            ERC20(_COMP).allowance(address(vault), _ROUTER),
-            type(uint256).max
-        );
-    }
-
     /*//////////////////////////////////////////////////////////////
                              deposit (ETH)
     //////////////////////////////////////////////////////////////*/