fix: revert when first every timestamp is expired

mento-protocol · Jul 9, 2024 · c2195a3 · c2195a3
1 parent a123f85
commit c2195a3
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 6 deletions.
diff --git a/contracts/oracles/ChainlinkRelayerV1.sol b/contracts/oracles/ChainlinkRelayerV1.sol
@@ -103,10 +103,10 @@ contract ChainlinkRelayerV1 is IChainlinkRelayer {
       if (timestamp <= lastTimestamp) {
         revert TimestampNotNew();
       }
+    }
 
-      if (isTimestampExpired(timestamp)) {
-        revert ExpiredTimestamp();
-      }
+    if (isTimestampExpired(timestamp)) {
+      revert ExpiredTimestamp();
     }
 
     if (price < 0) {

diff --git a/test/oracles/ChainlinkRelayer.t.sol b/test/oracles/ChainlinkRelayer.t.sol
@@ -20,6 +20,7 @@ contract ChainlinkRelayerTest is BaseTest {
   int256 aPrice = 420000000;
   uint256 expectedReport = 4200000000000000000000000;
   uint256 aReport = 4100000000000000000000000;
+  uint256 expirySeconds = 600;
 
   function setUp() public {
     sortedOracles = new SortedOracles(true);
@@ -31,6 +32,7 @@ contract ChainlinkRelayerTest is BaseTest {
       )
     );
     sortedOracles.addOracle(rateFeedId, address(relayer));
+    sortedOracles.setTokenReportExpiry(rateFeedId, expirySeconds);
   }
 }
 
@@ -73,7 +75,7 @@ contract ChainlinkRelayerTest_relay is ChainlinkRelayerTest {
   }
 
   function test_revertsOnNegativePrice() public {
-    chainlinkAggregator.setRoundData(-1 * aPrice, uint256(1337));
+    chainlinkAggregator.setRoundData(-1 * aPrice, block.timestamp);
     vm.expectRevert(NEGATIVE_PRICE_ERROR);
     relayer.relay();
   }
@@ -99,9 +101,15 @@ contract ChainlinkRelayerTest_relay is ChainlinkRelayerTest {
   function test_revertsOnExpiredTimestamp() public {
     vm.prank(address(relayer));
     sortedOracles.report(rateFeedId, aReport, address(0), address(0));
-    uint256 expiry = sortedOracles.getTokenReportExpirySeconds(rateFeedId);
     chainlinkAggregator.setRoundData(aPrice, block.timestamp + 1);
-    vm.warp(block.timestamp + expiry + 1);
+    vm.warp(block.timestamp + expirySeconds + 1);
+    vm.expectRevert(EXPIRED_TIMESTAMP_ERROR);
+    relayer.relay();
+  }
+
+  function test_revertsWhenFirstTimestampIsExpired() public {
+    chainlinkAggregator.setRoundData(aPrice, block.timestamp);
+    vm.warp(block.timestamp + expirySeconds);
     vm.expectRevert(EXPIRED_TIMESTAMP_ERROR);
     relayer.relay();
   }