test: isUpdatedIdExecuted

bgd-labs · Sep 29, 2024 · 84f0bb8 · 84f0bb8
1 parent ab106df
commit 84f0bb8
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/src/contracts/AaveStewardInjector.sol b/src/contracts/AaveStewardInjector.sol
@@ -79,6 +79,8 @@ contract AaveStewardInjector is Ownable, IAaveStewardInjector {
 
     IRiskOracle.RiskParameterUpdate memory riskParams = IRiskOracle(RISK_ORACLE).getUpdateById(updateIdToExecute);
     IRiskSteward(RISK_STEWARD).updateRates(_repackRateUpdate(riskParams));
+    _isUpdateIdExecuted[updateIdToExecute] = true;
+
     emit ActionSucceeded(updateIdToExecute);
   }
 

diff --git a/tests/AaveStewardsInjector.t.sol b/tests/AaveStewardsInjector.t.sol
@@ -195,6 +195,20 @@ contract AaveStewardsInjector_Test is TestnetProcedures {
     assertTrue(isAutomationPerformed);
   }
 
+  function test_isUpdatedIdExecuted() public {
+     // add rate update to risk oracle
+    _addUpdateToRiskOracle();
+
+    assertFalse(_stewardInjector.isUpdateIdExecuted(1));
+
+    bool isAutomationPerformed = _checkAndPerformAutomation();
+    assertTrue(isAutomationPerformed);
+    assertTrue(_stewardInjector.isUpdateIdExecuted(1));
+
+    isAutomationPerformed = _checkAndPerformAutomation();
+    assertFalse(isAutomationPerformed);
+  }
+
   function _addUpdateToRiskOracle() internal {
     vm.startPrank(_riskOracleOwner);