Removed unnecessary arguments

pods-finance · Oct 17, 2022 · f100925 · f100925
1 parent 17136cc
commit f100925
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 11 deletions.
diff --git a/contracts/mocks/PrincipalProtectedMock.sol b/contracts/mocks/PrincipalProtectedMock.sol
@@ -43,11 +43,11 @@ contract PrincipalProtectedMock is BaseVault {
         sharePriceDecimals = _asset.decimals();
     }
 
-    function _afterRoundStart(uint256 assets) internal override {
+    function _afterRoundStart() internal override {
         lastRoundAssets = totalAssets();
-        if (assets > 0) {
-            IERC20Metadata(asset()).approve(address(yieldSource), assets);
-            yieldSource.deposit(assets, address(this));
+        if (processedDeposits > 0) {
+            IERC20Metadata(asset()).approve(address(yieldSource), processedDeposits);
+            yieldSource.deposit(processedDeposits, address(this));
         }
         uint256 supply = totalSupply();
 

diff --git a/contracts/mocks/YieldVaultMock.sol b/contracts/mocks/YieldVaultMock.sol
@@ -29,10 +29,10 @@ contract YieldVaultMock is BaseVault {
         yieldSource.withdraw(assets, address(this), address(this));
     }
 
-    function _afterRoundStart(uint256 assets) internal override {
-        if (yieldSource.previewDeposit(assets) > 0) {
-            IERC20Metadata(yieldSource.asset()).approve(address(yieldSource), assets);
-            yieldSource.deposit(assets, address(this));
+    function _afterRoundStart() internal override {
+        if (yieldSource.previewDeposit(processedDeposits) > 0) {
+            IERC20Metadata(yieldSource.asset()).approve(address(yieldSource), processedDeposits);
+            yieldSource.deposit(processedDeposits, address(this));
         }
     }
 }
diff --git a/contracts/vaults/BaseVault.sol b/contracts/vaults/BaseVault.sol
@@ -267,7 +267,7 @@ abstract contract BaseVault is IVault, ERC20Permit, ERC4626, Capped {
 
         isProcessingDeposits = false;
 
-        _afterRoundStart(processedDeposits);
+        _afterRoundStart();
         emit RoundStarted(currentRoundId, processedDeposits);
         processedDeposits = 0;
 
@@ -432,7 +432,7 @@ abstract contract BaseVault is IVault, ERC20Permit, ERC4626, Capped {
     /* This hook should be implemented in the contract implementation.
         It will trigger after setting isProcessingDeposits to false
     */
-    function _afterRoundStart(uint256 assets) internal virtual {}
+    function _afterRoundStart() internal virtual {}
 
     // solhint-disable-next-line no-empty-blocks
     /* This hook should be implemented in the contract implementation.

diff --git a/contracts/vaults/STETHVault.sol b/contracts/vaults/STETHVault.sol
@@ -48,7 +48,7 @@ contract STETHVault is BaseVault {
         sharePriceDecimals = _asset.decimals();
     }
 
-    function _afterRoundStart(uint256) internal override {
+    function _afterRoundStart() internal override {
         uint256 supply = totalSupply();
 
         lastRoundAssets = totalAssets();