Merge pull request #1862 from kleros/fix/sm-change-governor-upgrade

SortitionModule upgrade to add changeGovernor()

Author: jaybuidl

contracts/deploy/upgrade-sortition-module.ts renamed to contracts/deploy/upgrade-sortition-module-v0.9.0.ts

@@ -15,7 +15,7 @@ const deployUpgradeSortitionModule: DeployFunction = async (hre: HardhatRuntimeE
     console.log("upgrading SortitionModuleNeo...");
     await deployUpgradable(deployments, "SortitionModuleNeo", {
       newImplementation: "SortitionModuleNeo",
-      initializer: "initialize",
+      initializer: "initialize3",
       from: deployer,
       // Warning: do not reinitialize everything, only the new variables
       args: [],

contracts/src/arbitration/SortitionModule.sol

@@ -44,6 +44,10 @@ contract SortitionModule is SortitionModuleBase {
         __SortitionModuleBase_initialize(_governor, _core, _minStakingTime, _maxDrawingTime, _rng, _rngLookahead);
     }
 
+    function initialize3() external reinitializer(3) {
+        // NOP
+    }
+
     // ************************************* //
     // *             Governance            * //
     // ************************************* //

contracts/src/arbitration/SortitionModuleBase.sol

@@ -152,6 +152,12 @@ abstract contract SortitionModuleBase is ISortitionModule, Initializable, UUPSPr
     // *             Governance            * //
     // ************************************* //
 
+    /// @dev Changes the governor of the contract.
+    /// @param _governor The new governor.
+    function changeGovernor(address _governor) external onlyByGovernor {
+        governor = _governor;
+    }
+
     /// @dev Changes the `minStakingTime` storage variable.
     /// @param _minStakingTime The new value for the `minStakingTime` storage variable.
     function changeMinStakingTime(uint256 _minStakingTime) external onlyByGovernor {

contracts/src/arbitration/SortitionModuleNeo.sol

@@ -58,6 +58,10 @@ contract SortitionModuleNeo is SortitionModuleBase {
         maxTotalStaked = _maxTotalStaked;
     }
 
+    function initialize3() external reinitializer(3) {
+        // NOP
+    }
+
     // ************************************* //
     // *             Governance            * //
     // ************************************* //