fix: asset is erc20

yearn · Oct 12, 2023 · 2226996 · 2226996
1 parent 4c2c2e2
commit 2226996
Show file tree

Hide file tree

Showing 6 changed files with 7 additions and 8 deletions.
diff --git a/src/test/mocks/MockHealthCheck.sol b/src/test/mocks/MockHealthCheck.sol
@@ -24,7 +24,7 @@ contract MockHealthCheck is BaseHealthCheck {
     {
         require(_healthy(), "unhealthy");
 
-        _totalAssets = ERC20(asset).balanceOf(address(this));
+        _totalAssets = asset.balanceOf(address(this));
 
         _executeHealthCheck(_totalAssets);
     }

diff --git a/src/test/mocks/MockSolidlySwapper.sol b/src/test/mocks/MockSolidlySwapper.sol
@@ -16,7 +16,7 @@ contract MockSolidlySwapper is BaseStrategy, SolidlySwapper {
         override
         returns (uint256 _totalAssets)
     {
-        _totalAssets = ERC20(asset).balanceOf(address(this));
+        _totalAssets = asset.balanceOf(address(this));
     }
 
     function setMinAmountToSell(uint256 _minAmountToSell) external {

diff --git a/src/test/mocks/MockStrategy.sol b/src/test/mocks/MockStrategy.sol
@@ -17,7 +17,7 @@ contract MockStrategy is BaseStrategy {
         override
         returns (uint256 _totalAssets)
     {
-        _totalAssets = ERC20(asset).balanceOf(address(this));
+        _totalAssets = asset.balanceOf(address(this));
     }
 
     function _tendTrigger() internal view override returns (bool) {

diff --git a/src/test/mocks/MockUniswapV2Swapper.sol b/src/test/mocks/MockUniswapV2Swapper.sol
@@ -16,7 +16,7 @@ contract MockUniswapV2Swapper is BaseStrategy, UniswapV2Swapper {
         override
         returns (uint256 _totalAssets)
     {
-        _totalAssets = ERC20(asset).balanceOf(address(this));
+        _totalAssets = asset.balanceOf(address(this));
     }
 
     function setMinAmountToSell(uint256 _minAmountToSell) external {

diff --git a/src/test/mocks/MockUniswapV3Swapper.sol b/src/test/mocks/MockUniswapV3Swapper.sol
@@ -16,7 +16,7 @@ contract MockUniswapV3Swapper is BaseStrategy, UniswapV3Swapper {
         override
         returns (uint256 _totalAssets)
     {
-        _totalAssets = ERC20(asset).balanceOf(address(this));
+        _totalAssets = asset.balanceOf(address(this));
     }
 
     function setMinAmountToSell(uint256 _minAmountToSell) external {

diff --git a/src/test/utils/Setup.sol b/src/test/utils/Setup.sol
@@ -91,11 +91,10 @@ contract Setup is ExtendedTest {
 
         vm.prank(management);
         // Give the vault manager all the roles
-        vault.add_role(vaultManagement, 16383);
+        _vault.set_role(vaultManagement, 16383);
 
         vm.prank(vaultManagement);
-        vault.set_deposit_limit(type(uint256).max);
-
+        _vault.set_deposit_limit(type(uint256).max);
 
         return _vault;
     }