@@ -41,37 +41,49 @@ contract Integration_Upgrade_Redistribution_Base is UpgradeTest {
4141 // 3. Create an operator set and register an operator.
4242 operatorSet = avs.createOperatorSet (strategies);
4343
44- bool isOperatorSet = allocationManager.isOperatorSet (operatorSet);
45- console.log ("isOperatorSet " , isOperatorSet);
46-
4744 // 4. Operator allocates to operator set
48- // allocateParams = _genAllocation_AllAvailable(operator, operatorSet);
49- // operator.modifyAllocations(allocateParams);
45+ allocateParams = _genAllocation_AllAvailable (operator, operatorSet);
46+ operator.modifyAllocations (allocateParams);
5047
51- // _rollBlocksForCompleteAllocation(operator, operatorSet, strategies);
48+ _rollBlocksForCompleteAllocation (operator, operatorSet, strategies);
5249
5350 // 5. Operator registers for operator set
5451 operator.registerForOperatorSet (operatorSet);
5552
56- // // 6. Operator is randomly slashed by the operatorSet
57- // slashParams = _genSlashing_Rand(operator, operatorSet);
58- // avs.slashOperator(slashParams);
53+ // 6. Operator is randomly slashed by the operatorSet
54+ slashParams = _genSlashing_Rand (operator, operatorSet);
55+ avs.slashOperator (slashParams);
5956 }
6057}
6158
6259contract Integration_Upgrade_Redistribution is Integration_Upgrade_Redistribution_Base {
6360 function testFuzz_upgrade_burn (uint24 r ) public rand (r) {
64- // // 1. Upgrade contracts
65- // _upgradeEigenLayerContracts();
66-
67- // // 2. Burn shares
68- // (address[] memory strategiesWithBurnableShares,) = strategyManager.getStrategiesWithBurnableShares();
69- // for (uint i = 0; i < strategiesWithBurnableShares.length; i++) {
70- // strategyManager.burnShares(IStrategy(strategiesWithBurnableShares[i]));
71- // }
72-
73- // // Assert that there are no strategies with burned shares
74- // (address[] memory strategiesWithBurnedShares,) = strategyManager.getStrategiesWithBurnableShares();
75- // assertEq(strategiesWithBurnedShares.length, 0);
61+ // 1. Upgrade contracts
62+ _upgradeEigenLayerContracts ();
63+
64+ // 2. Burn shares
65+ (address [] memory strategiesWithBurnableShares ,) = strategyManager.getStrategiesWithBurnableShares ();
66+ for (uint i = 0 ; i < strategiesWithBurnableShares.length ; i++ ) {
67+ strategyManager.burnShares (IStrategy (strategiesWithBurnableShares[i]));
68+ }
69+
70+ // Assert that there are no strategies with burned shares
71+ (address [] memory strategiesWithBurnedShares ,) = strategyManager.getStrategiesWithBurnableShares ();
72+ assertEq (strategiesWithBurnedShares.length , 0 );
7673 }
74+
75+ // function testFuzz_burn_update_operatorSet(uint24 r) public rand(r) {
76+ // // 1. Burn shares
77+ // (address[] memory strategiesWithBurnableShares,) = strategyManager.getStrategiesWithBurnableShares();
78+ // for (uint i = 0; i < strategiesWithBurnableShares.length; i++) {
79+ // strategyManager.burnShares(IStrategy(strategiesWithBurnableShares[i]));
80+ // }
81+
82+ // // 2. Upgrade contracts
83+ // _upgradeEigenLayerContracts();
84+
85+ // // Assert that there are no strategies with burned shares
86+ // (address[] memory strategiesWithBurnedShares,) = strategyManager.getStrategiesWithBurnableShares();
87+ // assertEq(strategiesWithBurnedShares.length, 0);
88+ // }
7789}
0 commit comments