script: migrate auction + controller treasury

axieinfinity · Jul 5, 2024 · f0147d3 · f0147d3
1 parent 48d7e76
commit f0147d3
Showing 1 changed file with 15 additions and 3 deletions.
diff --git a/...05_MigrateAuctionAndControllerAdmin.s.sol → ...uctionAndControllerAdminAndTreasury.s.sol b/...05_MigrateAuctionAndControllerAdmin.s.sol → ...uctionAndControllerAdminAndTreasury.s.sol
@@ -7,21 +7,27 @@ import {
 } from "script/contracts/RONRegistrarControllerDeploy.s.sol";
 import { Contract } from "script/utils/Contract.sol";
 import { Migration } from "script/Migration.s.sol";
+import { RNSCommission, RNSCommissionDeploy } from "script/contracts/RNSCommissionDeploy.s.sol";
 
-contract Migration__20240507_MigrateAuctionAndControllerAdminMainnet is Migration {
+contract Migration__20240507_MigrateAuctionAndControllerAdminAndTreasuryMainnet is Migration {
   RONRegistrarController private _controller;
   RNSAuction private _auction;
+  RNSCommission private _commission;
   address private _currentAdmin;
   address private _nextAdmin;
   bytes32 public constant DEFAULT_ADMIN_ROLE = 0x00;
 
   function run() public {
     _auction = RNSAuction(loadContract(Contract.RNSAuction.key()));
     _controller = RONRegistrarController(loadContract(Contract.RONRegistrarController.key()));
+    _commission = RNSCommission(loadContract(Contract.RNSCommission.key()));
     _currentAdmin = 0x0F68eDBE14C8f68481771016d7E2871d6a35DE11;
     _nextAdmin = 0x9D05D1F5b0424F8fDE534BC196FFB6Dd211D902a;
 
     vm.startBroadcast(_currentAdmin);
+    _auction.setTreasury(payable(address(_commission)));
+    _controller.setTreasury(payable(address(_commission)));
+
     _auction.grantRole(DEFAULT_ADMIN_ROLE, _nextAdmin);
     _controller.grantRole(DEFAULT_ADMIN_ROLE, _nextAdmin);
 
@@ -31,9 +37,15 @@ contract Migration__20240507_MigrateAuctionAndControllerAdminMainnet is Migratio
   }
 
   function _postCheck() internal override {
+    _validateTreasuryAddress();
     _validateNewAdmin();
     _validatePrevAdminIsRevoked();
-    _validateNumberOfAdmin();
+    _validateAdminCount();
+  }
+
+  function _validateTreasuryAddress() internal logFn("_validateTreasuryAddress") {
+    assertEq(_auction.getTreasury(), address(_commission));
+    assertEq(_controller.getTreasury(), address(_commission));
   }
 
   function _validateNewAdmin() internal logFn("_validateNewAdmin") {
@@ -48,7 +60,7 @@ contract Migration__20240507_MigrateAuctionAndControllerAdminMainnet is Migratio
     require(!_controller.hasRole(DEFAULT_ADMIN_ROLE, prevAdmin));
   }
 
-  function _validateNumberOfAdmin() internal logFn("_validateNumberOfAdmin") {
+  function _validateAdminCount() internal logFn("_validateAdminCount") {
     assertEq(_auction.getRoleMemberCount(DEFAULT_ADMIN_ROLE), 1);
     assertEq(_controller.getRoleMemberCount(DEFAULT_ADMIN_ROLE), 1);
   }