Merge pull request #577 from hats-finance/exclude-addrc-from-timelock

Exclude addRewardController from timelock
hats-finance · Jul 4, 2024 · b05b7db · b05b7db
2 parents 385bf0a + b96576f
commit b05b7db
Show file tree

Hide file tree

Showing 3 changed files with 41 additions and 0 deletions.
diff --git a/contracts/HATTimelockController.sol b/contracts/HATTimelockController.sol
@@ -54,6 +54,10 @@ contract HATTimelockController is TimelockController {
     external onlyRole(PROPOSER_ROLE) {
         _rewardController.setAllocPoint(address(_vault), _allocPoint);
     }
+
+    function addRewardController(IHATVault _vault, IRewardController _rewardController) external onlyRole(PROPOSER_ROLE) {
+        _vault.addRewardController(_rewardController);
+    }
 
     function swapAndSend(
         IHATVaultsRegistry _registry,

diff --git a/docs/dodoc/HATTimelockController.md b/docs/dodoc/HATTimelockController.md
@@ -112,6 +112,23 @@ function TIMELOCK_ADMIN_ROLE() external view returns (bytes32)
 |---|---|---|
 | _0 | bytes32 | undefined |
 
+### addRewardController
+
+```solidity
+function addRewardController(contract IHATVault _vault, contract IRewardController _rewardController) external nonpayable
+```
+
+
+
+
+
+#### Parameters
+
+| Name | Type | Description |
+|---|---|---|
+| _vault | contract IHATVault | undefined |
+| _rewardController | contract IRewardController | undefined |
+
 ### approveClaim
 
 ```solidity

diff --git a/test/hattimelockcontroller.js b/test/hattimelockcontroller.js
@@ -302,6 +302,26 @@ contract("HatTimelockController", (accounts) => {
     await vault.deposit(web3.utils.toWei("1"), staker, { from: staker });
   });
 
+  it("Add reward controller", async () => {
+    await setup(accounts);
+    try {
+      await hatTimelockController.addRewardController(vault.address, accounts[1], {
+        from: accounts[1],
+      });
+      assert(false, "only governance");
+    } catch (ex) {
+      assertVMException(ex);
+    }
+
+    try {
+      await vault.addRewardController(accounts[1]);
+      assert(false, "only governance");
+    } catch (ex) {
+      assertVMException(ex);
+    }
+    await hatTimelockController.addRewardController(vault.address, accounts[1], { from: accounts[0] });
+  });
+
   it("swapAndSend", async () => {
     await setup(accounts);
     var staker = accounts[4];