test: queryMessageHashes

alt-research · May 16, 2024 · b9c53af · b9c53af
1 parent e2db3d2
commit b9c53af
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/contracts/test/MachServiceManager.t.sol b/contracts/test/MachServiceManager.t.sol
@@ -515,4 +515,17 @@ contract MachServiceManagerTest is BLSAVSDeployer {
         vm.expectRevert("Ownable: caller is not the owner");
         serviceManager.removeAlert("foo");
     }
+
+    function test_QueryMessageHashes() public {
+        test_confirmAlert();
+        bytes32[] memory results = serviceManager.queryMessageHashes(0, 2);
+        assertTrue(results.length == 1);
+        assertTrue(results[0] == "foo");
+    }
+
+    function test_QueryMessageHashes_RevertIfInvalidStartIndex() public {
+        test_confirmAlert();
+        vm.expectRevert(InvalidStartIndex.selector);
+        bytes32[] memory results = serviceManager.queryMessageHashes(1, 2);
+    }
 }