Add registrar controller switch-over integration test

Author: stevieraykatz

test/Integration/IntegrationTestBase.t.sol

@@ -118,10 +118,18 @@ contract IntegrationTestBase is Test {
         vm.stopPrank();
     }
 
-    function test_integration_register() public {
-        vm.stopPrank();
-        vm.startPrank(alice);
+    function _getBasePrices() internal pure returns (uint256[] memory) {
+        uint256[] memory rentPrices = new uint256[](6);
+        rentPrices[0] = 316_808_781_402;
+        rentPrices[1] = 31_680_878_140;
+        rentPrices[2] = 3_168_087_814;
+        rentPrices[3] = 316_808_781;
+        rentPrices[4] = 31_680_878;
+        rentPrices[5] = 3_168_087; // 3,168,808.781402895 = 1e14 / (365.25 * 24 * 3600)
+        return rentPrices;
+    }
 
+    function _registerAlice() internal {
         string memory name = "alice";
         uint256 duration = 365.25 days;
 
@@ -139,15 +147,4 @@ contract IntegrationTestBase is Test {
 
         registrarController.register{value: registerPrice}(request);
     }
-
-    function _getBasePrices() internal pure returns (uint256[] memory) {
-        uint256[] memory rentPrices = new uint256[](6);
-        rentPrices[0] = 316_808_781_402;
-        rentPrices[1] = 31_680_878_140;
-        rentPrices[2] = 3_168_087_814;
-        rentPrices[3] = 316_808_781;
-        rentPrices[4] = 31_680_878;
-        rentPrices[5] = 3_168_087; // 3,168,808.781402895 = 1e14 / (365.25 * 24 * 3600)
-        return rentPrices;
-    }
 }

test/Integration/SwitchToUpgradeableRegistrarController.t.sol

@@ -0,0 +1,113 @@
+//SPDX-License-Identifier: MIT
+pragma solidity ^0.8.23;
+
+import {Test} from "forge-std/Test.sol";
+import {console} from "forge-std/console.sol";
+
+import {IntegrationTestBase} from "./IntegrationTestBase.t.sol";
+import {MockL2ReverseRegistrar} from "test/mocks/MockL2ReverseRegistrar.sol";
+
+import {ExponentialPremiumPriceOracle} from "src/L2/ExponentialPremiumPriceOracle.sol";
+import {IPriceOracle} from "src/L2/interface/IPriceOracle.sol";
+import {UpgradeableRegistrarController} from "src/L2/UpgradeableRegistrarController.sol";
+import {TransparentUpgradeableProxy} from
+    "openzeppelin-contracts/contracts/proxy/transparent/TransparentUpgradeableProxy.sol";
+
+import {BASE_ETH_NODE} from "src/util/Constants.sol";
+
+contract SwitchToUpgradeableRegistrarController is IntegrationTestBase {
+    UpgradeableRegistrarController public controllerImpl;
+    UpgradeableRegistrarController public controller;
+    TransparentUpgradeableProxy public proxy;
+    MockL2ReverseRegistrar public l2ReverseRegistrar;
+
+    address admin;
+
+    uint256 constant UPGRADE_TIMESTAMP = 1746057600; // May 1 2025
+
+    function setUp() public override {
+        super.setUp();
+        _registerAlice();
+
+        vm.warp(UPGRADE_TIMESTAMP);
+
+        admin = makeAddr("admin");
+
+        l2ReverseRegistrar = new MockL2ReverseRegistrar();
+
+        exponentialPremiumPriceOracle = new ExponentialPremiumPriceOracle(
+            _getBasePrices(), EXPIRY_AUCTION_START_PRICE, EXPIRY_AUCTION_DURATION_DAYS
+        );
+
+        bytes memory controllerInitData = abi.encodeWithSelector(
+            UpgradeableRegistrarController.initialize.selector,
+            baseRegistrar,
+            exponentialPremiumPriceOracle,
+            reverseRegistrar,
+            owner,
+            BASE_ETH_NODE,
+            ".base.eth",
+            payments,
+            address(registrarController),
+            address(l2ReverseRegistrar)
+        );
+
+        controllerImpl = new UpgradeableRegistrarController();
+        proxy = new TransparentUpgradeableProxy(address(controllerImpl), admin, controllerInitData);
+        controller = UpgradeableRegistrarController(address(proxy));
+
+        _postDeployConfig();
+    }
+
+    function _postDeployConfig() internal {
+        vm.startPrank(owner);
+        baseRegistrar.addController(address(proxy));
+        reverseRegistrar.setControllerApproval(address(proxy), true);
+        defaultL2Resolver.setRegistrarController(address(proxy));
+        vm.stopPrank();
+    }
+
+    function test_canRegisterANewName() public {
+        string memory name = "new-name";
+        uint256 duration = 365.25 days;
+        uint256[] memory cointypes = new uint256[](1);
+        cointypes[0] = 0x80000000 | 0x00002105;
+
+        console.log("fetching price");
+
+        uint256 registerPrice = controller.registerPrice(name, duration);
+
+        uint256 expectedPrice = _getBasePrices()[4] * duration;
+
+        vm.assertEq(registerPrice, expectedPrice);
+
+        console.log("Fetched Price");
+
+        UpgradeableRegistrarController.RegisterRequest memory request = UpgradeableRegistrarController.RegisterRequest({
+            name: name,
+            owner: alice,
+            duration: duration,
+            resolver: address(defaultL2Resolver),
+            data: new bytes[](0),
+            reverseRecord: true,
+            cointypes: cointypes,
+            signatureExpiry: block.timestamp,
+            signature: ""
+        });
+
+        vm.deal(alice, 1 ether);
+        vm.prank(alice);
+        controller.register{value: registerPrice}(request);
+    }
+
+    function test_canRenewExistingName() public {
+        string memory name = "alice";
+        uint256 duration = 365.25 days;
+
+        IPriceOracle.Price memory prices = controller.rentPrice(name, duration);
+
+        vm.deal(alice, 1 ether);
+        vm.prank(alice);
+        controller.renew{value: prices.base}(name, duration);
+    }
+}