test: actually mock the callback to revert

defi-wonderland · Oct 9, 2024 · bc2db59 · bc2db59
1 parent 900a06c
commit bc2db59
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/solidity/test/unit/modules/finality/CallbackModule.t.sol b/solidity/test/unit/modules/finality/CallbackModule.t.sol
@@ -127,8 +127,11 @@ contract CallbackModule_Unit_FinalizeRequest is BaseTest {
     mockRequest.finalityModuleData = abi.encode(ICallbackModule.RequestParameters({target: _target, data: _data}));
     mockResponse.requestId = _getId(mockRequest);
 
-    // Mock and expect the callback
-    _mockAndExpect(_target, abi.encodeWithSelector(IProphetCallback.prophetCallback.selector, _data), abi.encode(''));
+    // Mock revert and expect the callback
+    vm.mockCallRevert(
+      _target, abi.encodeWithSelector(IProphetCallback.prophetCallback.selector, _data), abi.encode('err')
+    );
+    vm.expectCall(_target, abi.encodeWithSelector(IProphetCallback.prophetCallback.selector, _data));
 
     vm.prank(address(oracle));
     callbackModule.finalizeRequest(mockRequest, mockResponse, _proposer);