remove mutation base

Author: godzillaba

test-foundry/libraries/vault/MasterVaultCore.t.sol

@@ -42,6 +42,11 @@ contract MasterVaultCoreTest is Test {
     string public symbol = "mTST";
     uint256 public constant DEAD_SHARES = 10 ** 6;
 
+    address public keeper = address(0xBBBB);
+    address public beneficiaryAddr = address(0x9999);
+    address public generalManager = address(0xAAAA);
+    address public pauser = address(0xCCCC);
+
     struct State {
         uint256 userShares;
         uint256 masterVaultTotalAssets;
@@ -70,6 +75,14 @@ contract MasterVaultCoreTest is Test {
         );
         token = new TestERC20();
         vault = MasterVault(factory.deployVault(address(token)));
+
+        vault.rolesRegistry().grantRole(vault.GENERAL_MANAGER_ROLE(), address(this));
+        vault.rolesRegistry().grantRole(vault.GENERAL_MANAGER_ROLE(), generalManager);
+        vault.rolesRegistry().grantRole(vault.KEEPER_ROLE(), keeper);
+        vault.rolesRegistry().grantRole(vault.FEE_MANAGER_ROLE(), address(this));
+        vault.rolesRegistry().grantRole(vault.PAUSER_ROLE(), pauser);
+        vault.setBeneficiary(beneficiaryAddr);
+        vault.setMinimumRebalanceAmount(1);
     }
 
     function _checkState(State memory expectedState) internal {
@@ -135,4 +148,22 @@ contract MasterVaultCoreTest is Test {
         console2.log(" subVaultTotalSupply:", state.subVaultTotalSupply);
         console2.log(" subVaultTokenBalance:", state.subVaultTokenBalance);
     }
+
+    function _depositAs(uint256 amount) internal returns (uint256) {
+        vm.prank(user);
+        token.mintAmount(amount);
+        vm.startPrank(user);
+        token.approve(address(vault), amount);
+        uint256 shares = vault.deposit(amount);
+        vm.stopPrank();
+        return shares;
+    }
+
+    function _setupWithAllocation(uint256 depositAmount, uint64 allocationWad) internal {
+        _depositAs(depositAmount);
+        vault.setTargetAllocationWad(allocationWad);
+        vm.warp(block.timestamp + 2);
+        vm.prank(keeper);
+        vault.rebalance(-1e18);
+    }
 }

test-foundry/libraries/vault/mutation/MasterVaultAccessControl.t.sol

@@ -1,10 +1,10 @@
 // SPDX-License-Identifier: UNLICENSED
 pragma solidity ^0.8.0;
 
-import {MasterVaultMutationBase} from "./MasterVaultMutationBase.t.sol";
+import {MasterVaultCoreTest} from "../MasterVaultCore.t.sol";
 import {MasterVault} from "../../../../contracts/tokenbridge/libraries/vault/MasterVault.sol";
 
-contract MasterVaultAccessControlTest is MasterVaultMutationBase {
+contract MasterVaultAccessControlTest is MasterVaultCoreTest {
     function test_onlyGateway_rejectsNonGateway() public {
         address notGateway = address(0x1234);
         vm.prank(notGateway);

test-foundry/libraries/vault/mutation/MasterVaultDefaultSubVault.t.sol

@@ -1,12 +1,12 @@
 // SPDX-License-Identifier: UNLICENSED
 pragma solidity ^0.8.0;
 
-import {MasterVaultMutationBase} from "./MasterVaultMutationBase.t.sol";
+import {MasterVaultCoreTest} from "../MasterVaultCore.t.sol";
 import {
     DefaultSubVault
 } from "../../../../contracts/tokenbridge/libraries/vault/MasterVaultFactory.sol";
 
-contract MasterVaultDefaultSubVaultTest is MasterVaultMutationBase {
+contract MasterVaultDefaultSubVaultTest is MasterVaultCoreTest {
     function test_defaultSubVault_withdraw_onlyMasterVault() public {
         DefaultSubVault dsv = DefaultSubVault(address(vault.subVault()));
         address attacker = address(0xdead);

test-foundry/libraries/vault/mutation/MasterVaultFees.t.sol

@@ -1,14 +1,14 @@
 // SPDX-License-Identifier: UNLICENSED
 pragma solidity ^0.8.0;
 
-import {MasterVaultMutationBase} from "./MasterVaultMutationBase.t.sol";
+import {MasterVaultCoreTest} from "../MasterVaultCore.t.sol";
 import {MasterVault} from "../../../../contracts/tokenbridge/libraries/vault/MasterVault.sol";
 import {TestERC20} from "../../../../contracts/tokenbridge/test/TestERC20.sol";
 import {IERC4626} from "@openzeppelin/contracts/interfaces/IERC4626.sol";
 import {IERC20} from "@openzeppelin/contracts/token/ERC20/IERC20.sol";
 import {Vm} from "forge-std/Test.sol";
 
-contract MasterVaultFeesTest is MasterVaultMutationBase {
+contract MasterVaultFeesTest is MasterVaultCoreTest {
     function test_totalProfit_correctMath() public {
         // this test was AI generated to target a mutant where subtract is replaced with modulo in profit calculation
 

test-foundry/libraries/vault/mutation/MasterVaultInit.t.sol

@@ -1,7 +1,7 @@
 // SPDX-License-Identifier: UNLICENSED
 pragma solidity ^0.8.0;
 
-import {MasterVaultMutationBase} from "./MasterVaultMutationBase.t.sol";
+import {MasterVaultCoreTest} from "../MasterVaultCore.t.sol";
 import {MasterVault} from "../../../../contracts/tokenbridge/libraries/vault/MasterVault.sol";
 import {TestERC20} from "../../../../contracts/tokenbridge/test/TestERC20.sol";
 import {IAccessControlUpgradeable} from "@openzeppelin/contracts-upgradeable/access/IAccessControlUpgradeable.sol";
@@ -26,7 +26,7 @@ contract OverflowDecimalsToken is BadTokenBase {
     function decimals() external pure returns (uint8) { return type(uint8).max - 5; }
 }
 
-contract MasterVaultInitTest is MasterVaultMutationBase {
+contract MasterVaultInitTest is MasterVaultCoreTest {
     function test_initialize_setsERC20Name() public {
         string memory n = vault.name();
         assertTrue(bytes(n).length > 0, "name should be set");

test-foundry/libraries/vault/mutation/MasterVaultMutationBase.t.sol

@@ -1,10 +1,10 @@
 // SPDX-License-Identifier: UNLICENSED
 pragma solidity ^0.8.0;
 
-import {MasterVaultMutationBase} from "./MasterVaultMutationBase.t.sol";
+import {MasterVaultCoreTest} from "../MasterVaultCore.t.sol";
 import {MasterVault} from "../../../../contracts/tokenbridge/libraries/vault/MasterVault.sol";
 
-contract MasterVaultPauseTest is MasterVaultMutationBase {
+contract MasterVaultPauseTest is MasterVaultCoreTest {
     function test_pause_works() public {
         vm.prank(pauser);
         vault.pause();

test-foundry/libraries/vault/mutation/MasterVaultPause.t.sol

@@ -1,10 +1,10 @@
 // SPDX-License-Identifier: UNLICENSED
 pragma solidity ^0.8.0;
 
-import {MasterVaultMutationBase} from "./MasterVaultMutationBase.t.sol";
+import {MasterVaultCoreTest} from "../MasterVaultCore.t.sol";
 import {MasterVault} from "../../../../contracts/tokenbridge/libraries/vault/MasterVault.sol";
 
-contract MasterVaultRebalanceTest is MasterVaultMutationBase {
+contract MasterVaultRebalanceTest is MasterVaultCoreTest {
     function test_rebalance_targetAllocationMet_reverts() public {
         _setupWithAllocation(1e18, 5e17);
         // Rebalance again with same allocation - should be met now

test-foundry/libraries/vault/mutation/MasterVaultRebalance.t.sol

@@ -1,10 +1,10 @@
 // SPDX-License-Identifier: UNLICENSED
 pragma solidity ^0.8.0;
 
-import {MasterVaultMutationBase} from "./MasterVaultMutationBase.t.sol";
+import {MasterVaultCoreTest} from "../MasterVaultCore.t.sol";
 import {MasterVault} from "../../../../contracts/tokenbridge/libraries/vault/MasterVault.sol";
 
-contract MasterVaultRebalanceCooldownTest is MasterVaultMutationBase {
+contract MasterVaultRebalanceCooldownTest is MasterVaultCoreTest {
     function test_rebalance_cooldownMath() public {
         _depositAs(10e18);
         vault.setTargetAllocationWad(3e17); // 30% allocation

test-foundry/libraries/vault/mutation/MasterVaultRebalanceCooldown.t.sol

@@ -1,10 +1,10 @@
 // SPDX-License-Identifier: UNLICENSED
 pragma solidity ^0.8.0;
 
-import {MasterVaultMutationBase} from "./MasterVaultMutationBase.t.sol";
+import {MasterVaultCoreTest} from "../MasterVaultCore.t.sol";
 import {MasterVault} from "../../../../contracts/tokenbridge/libraries/vault/MasterVault.sol";
 
-contract MasterVaultRedeemTest is MasterVaultMutationBase {
+contract MasterVaultRedeemTest is MasterVaultCoreTest {
     function test_redeem_minAssets_reverts() public {
         uint256 shares = _depositAs(1e18);
         vm.prank(user);
@@ -13,21 +13,4 @@ contract MasterVaultRedeemTest is MasterVaultMutationBase {
         );
         vault.redeem(shares, 2e18);
     }
-
-    // /// TARGETS MUTANT #38 in MasterVault.sol
-    // function test_redeem_minAssets_swapArgs_gt0() public {
-    //     uint256 shares = _depositAs(1e18);
-    //     vm.prank(user);
-    //     // minAssets=1 should pass since assets >= 1
-    //     vault.redeem(shares, 1);
-    // }
-
-    // /// TARGETS MUTANT #39 in MasterVault.sol
-    // function test_redeem_minAssets_swapedComparison() public {
-    //     uint256 shares = _depositAs(1e18);
-    //     // Request minAssets just above what we'd get
-    //     vm.prank(user);
-    //     vm.expectRevert();
-    //     vault.redeem(shares, 1e18 + 1);
-    // }
 }