diff --git a/test/upgrade/DefaultEmissionManager.1.2.0.t.sol b/test/upgrade/DefaultEmissionManager.1.2.0.t.sol index a8be54f..f8aa573 100644 --- a/test/upgrade/DefaultEmissionManager.1.2.0.t.sol +++ b/test/upgrade/DefaultEmissionManager.1.2.0.t.sol @@ -50,15 +50,16 @@ contract DefaultEmissionManagerTest is Test { vm.prank(POLYGON_PROTOCOL_COUNCIL); - admin.upgrade(ITransparentUpgradeableProxy(address(emProxy)), address(newEmImpl)); + admin.upgradeAndCall( + ITransparentUpgradeableProxy(address(emProxy)), + address(newEmImpl), + abi.encodeWithSelector(DefaultEmissionManager.reinitialize.selector) + ); // initialize can still not be called vm.expectRevert("Initializable: contract is already initialized"); emProxy.initialize(makeAddr("token"), msg.sender); - // reinitialize to reset startTimestamp and start supply - emProxy.reinitialize(); - assertEq(pol.totalSupply(), emProxy.START_SUPPLY_1_2_0()); assertEq(block.timestamp, emProxy.startTimestamp());