feat: Add isControlledFacilitator function

aave · Oct 14, 2024 · 5f92f09 · 5f92f09
1 parent a8f654a
commit 5f92f09
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 0 deletions.
diff --git a/src/contracts/misc/GhoBucketSteward.sol b/src/contracts/misc/GhoBucketSteward.sol
@@ -93,6 +93,11 @@ contract GhoBucketSteward is Ownable, RiskCouncilControlled, IGhoBucketSteward {
     return _controlledFacilitators.values();
   }
 
+  /// @inheritdoc IGhoBucketSteward
+  function isControlledFacilitator(address facilitator) external view returns (bool) {
+    return _controlledFacilitatorsByAddress[facilitator];
+  }
+
   /// @inheritdoc IGhoBucketSteward
   function getFacilitatorBucketCapacityTimelock(
     address facilitator

diff --git a/src/contracts/misc/interfaces/IGhoBucketSteward.sol b/src/contracts/misc/interfaces/IGhoBucketSteward.sol
@@ -32,6 +32,13 @@ interface IGhoBucketSteward {
    */
   function getControlledFacilitators() external view returns (address[] memory);
 
+  /**
+   * @notice Checks if a facilitator is controlled by this steward
+   * @param facilitator The facilitator address to check
+   * @return True if the facilitator is controlled by this steward
+   */
+  function isControlledFacilitator(address facilitator) external view returns (bool);
+
   /**
    * @notice Returns timestamp of the facilitators last bucket capacity update
    * @param facilitator The facilitator address

diff --git a/src/test/TestGhoBucketSteward.t.sol b/src/test/TestGhoBucketSteward.t.sol
@@ -212,4 +212,15 @@ contract TestGhoBucketSteward is TestGhoBase {
     newGsmList[0] = address(GHO_GSM_4626);
     GHO_BUCKET_STEWARD.setControlledFacilitator(newGsmList, true);
   }
+
+  function testIsControlledFacilitator() public {
+    address facilitator = makeAddr('FACILITATOR');
+    address[] memory controlledFacilitators = new address[](1);
+    controlledFacilitators[0] = facilitator;
+    vm.prank(SHORT_EXECUTOR);
+    GHO_BUCKET_STEWARD.setControlledFacilitator(controlledFacilitators, true);
+    assertTrue(GHO_BUCKET_STEWARD.isControlledFacilitator(facilitator));
+    address nonFacilitator = makeAddr('NON_FACILITATOR');
+    assertFalse(GHO_BUCKET_STEWARD.isControlledFacilitator(nonFacilitator));
+  }
 }