added controller tests

Signed-off-by: chrismaree <[email protected]>
UMAprotocol · May 20, 2024 · 16960de · 16960de
1 parent 5d8409b
commit 16960de
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/test/unit/BaseController.sol b/test/unit/BaseController.sol
@@ -65,4 +65,17 @@ contract OvalUnlockLatestValue is CommonTest {
         vm.expectRevert("Ownable: caller is not the owner");
         baseController.setMaxTraversal(100);
     }
+
+    function testOwnerCanSetMaxAge() public {
+        uint256 newMaxAge = 7200; // 2 hours in seconds, different from default 1 day
+        vm.prank(owner);
+        baseController.setMaxAge(newMaxAge);
+        assertTrue(baseController.maxAge() == newMaxAge);
+    }
+
+    function testNonOwnerCannotSetMaxAge() public {
+        vm.prank(random);
+        vm.expectRevert("Ownable: caller is not the owner");
+        baseController.setMaxAge(7200);
+    }
 }