feat: add owner to interest rate contract (#131)

* add owner to InterestRate contract

* deploy WETH IRM

Author: thorseldon

contracts/protocol/InterestRate.sol

@@ -7,6 +7,7 @@ import {WadRayMath} from "../libraries/math/WadRayMath.sol";
 import {PercentageMath} from "../libraries/math/PercentageMath.sol";
 
 import {IERC20Upgradeable} from "@openzeppelin/contracts-upgradeable/token/ERC20/IERC20Upgradeable.sol";
+import {Ownable} from "@openzeppelin/contracts/access/Ownable.sol";
 
 /**
  * @title InterestRate contract
@@ -15,42 +16,42 @@ import {IERC20Upgradeable} from "@openzeppelin/contracts-upgradeable/token/ERC20
  * point of utilization and another from that one to 100%
  * @author Bend
  **/
-contract InterestRate is IInterestRate {
+contract InterestRate is IInterestRate, Ownable {
   using WadRayMath for uint256;
   using PercentageMath for uint256;
 
-  ILendPoolAddressesProvider public immutable addressesProvider;
+  ILendPoolAddressesProvider public addressesProvider;
 
   /**
    * @dev this constant represents the utilization rate at which the pool aims to obtain most competitive borrow rates.
    * Expressed in ray
    **/
-  uint256 public immutable OPTIMAL_UTILIZATION_RATE;
+  uint256 public OPTIMAL_UTILIZATION_RATE;
 
   /**
    * @dev This constant represents the excess utilization rate above the optimal. It's always equal to
    * 1-optimal utilization rate. Added as a constant here for gas optimizations.
    * Expressed in ray
    **/
 
-  uint256 public immutable EXCESS_UTILIZATION_RATE;
+  uint256 public EXCESS_UTILIZATION_RATE;
 
   // Base variable borrow rate when Utilization rate = 0. Expressed in ray
-  uint256 internal immutable _baseVariableBorrowRate;
+  uint256 internal _baseVariableBorrowRate;
 
   // Slope of the variable interest curve when utilization rate > 0 and <= OPTIMAL_UTILIZATION_RATE. Expressed in ray
-  uint256 internal immutable _variableRateSlope1;
+  uint256 internal _variableRateSlope1;
 
   // Slope of the variable interest curve when utilization rate > OPTIMAL_UTILIZATION_RATE. Expressed in ray
-  uint256 internal immutable _variableRateSlope2;
+  uint256 internal _variableRateSlope2;
 
   constructor(
     ILendPoolAddressesProvider provider,
     uint256 optimalUtilizationRate_,
     uint256 baseVariableBorrowRate_,
     uint256 variableRateSlope1_,
     uint256 variableRateSlope2_
-  ) {
+  ) Ownable() {
     addressesProvider = provider;
     OPTIMAL_UTILIZATION_RATE = optimalUtilizationRate_;
     EXCESS_UTILIZATION_RATE = WadRayMath.ray() - (optimalUtilizationRate_);
@@ -59,6 +60,19 @@ contract InterestRate is IInterestRate {
     _variableRateSlope2 = variableRateSlope2_;
   }
 
+  function setInterestRateParams(
+    uint256 optimalUtilizationRate_,
+    uint256 baseVariableBorrowRate_,
+    uint256 variableRateSlope1_,
+    uint256 variableRateSlope2_
+  ) public onlyOwner {
+    OPTIMAL_UTILIZATION_RATE = optimalUtilizationRate_;
+    EXCESS_UTILIZATION_RATE = WadRayMath.ray() - (optimalUtilizationRate_);
+    _baseVariableBorrowRate = baseVariableBorrowRate_;
+    _variableRateSlope1 = variableRateSlope1_;
+    _variableRateSlope2 = variableRateSlope2_;
+  }
+
   function variableRateSlope1() external view returns (uint256) {
     return _variableRateSlope1;
   }

deployments/deployed-contracts-main.json

@@ -268,5 +268,9 @@
   "TokenOracle": {
     "address": "0x09AbE7f7297B27e4343E1f3906267b0BE6dBe4eb",
     "deployer": "0x868964fa49a6fd6e116FE82c8f4165904406f479"
+  },
+  "rateStrategyWETH250205": {
+    "address": "0xeDc614E4e9A173Fb72e21b1814AC068A458C64e7",
+    "deployer": "0x868964fa49a6fd6e116FE82c8f4165904406f479"
   }
 }

deployments/deployed-contracts-sepolia.json

@@ -175,5 +175,9 @@
   "TokenOracle": {
     "address": "0xA840dC2B97F565BF5179CccC8e4f6E72d5eD1eab",
     "deployer": "0xafF5C36642385b6c7Aaf7585eC785aB2316b5db6"
+  },
+  "rateStrategyWETH250205": {
+    "address": "0xF1F89AE4a28318e36026ADcFe1B7188e2122db6B",
+    "deployer": "0xafF5C36642385b6c7Aaf7585eC785aB2316b5db6"
   }
 }

test/interest-rate.spec.ts

@@ -35,6 +35,35 @@ makeSuite("Interest rate tests", (testEnv: TestEnv) => {
     );
   });
 
+  it("Test the ownership check", async () => {
+    const { users } = testEnv;
+
+    const owner = await rateInstance.owner();
+
+    await rateInstance.transferOwnership(users[1].address);
+
+    await expect(rateInstance.setInterestRateParams(100, 100, 100, 100)).to.be.revertedWith(
+      "Ownable: caller is not the owner"
+    );
+
+    await rateInstance.connect(users[1].signer).transferOwnership(owner);
+  });
+
+  it("Test the set params", async () => {
+    await rateInstance.setInterestRateParams(100, 200, 300, 400);
+
+    await expect(await rateInstance.baseVariableBorrowRate()).to.be.equal(200, "baseRate not eq");
+    await expect(await rateInstance.variableRateSlope1()).to.be.equal(300, "slope1 not eq");
+    await expect(await rateInstance.variableRateSlope2()).to.be.equal(400, "slope2 not eq");
+
+    await rateInstance.setInterestRateParams(
+      rateStrategyStableOne.optimalUtilizationRate,
+      rateStrategyStableOne.baseVariableBorrowRate,
+      rateStrategyStableOne.variableRateSlope1,
+      rateStrategyStableOne.variableRateSlope2
+    );
+  });
+
   it("Checks rates at 0% utilization rate, empty reserve", async () => {
     const { 0: currentLiquidityRate, 1: currentVariableBorrowRate } = await rateInstance[
       "calculateInterestRates(address,address,uint256,uint256,uint256,uint256)"