chore: enhance mocks

Layr-Labs · Aug 14, 2024 · 5dcb3a8 · 5dcb3a8
1 parent 7157066
commit 5dcb3a8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/packages/contracts-bedrock/test/invariants/symbolic/MockL2ToL2Messenger.sol b/packages/contracts-bedrock/test/invariants/symbolic/MockL2ToL2Messenger.sol
@@ -88,13 +88,12 @@ contract MockL2ToL2Messenger is ITestL2ToL2CrossDomainMessenger {
         // TODO: Add more logic? Like replacing the `TSTORE` updates with `SSTORE` - or add the checks
 
         (bool succ, bytes memory ret) = _target.call{ value: msg.value }(_message);
-
         if (!succ) revert(string(ret));
     }
 
     // TODO
     function crossDomainMessageSource() external view returns (uint256 _source) {
-        _source = block.chainid;
+        _source = block.chainid + 1;
     }
 
     function crossDomainMessageSender() external view returns (address _sender) {

diff --git a/packages/contracts-bedrock/test/invariants/symbolic/OptimismSuperchainERC20.t.sol b/packages/contracts-bedrock/test/invariants/symbolic/OptimismSuperchainERC20.t.sol
@@ -109,7 +109,7 @@ contract OptimismSuperchainERC20_SymTest is HalmosTest {
         vm.assume(_to != address(0));
         MESSENGER.forTest_setCurrentXDomSender(address(optimismSuperchainERC20));
 
-        vm.prank(_sender);
+        vm.prank(address(MESSENGER));
         try optimismSuperchainERC20.relayERC20(_from, _to, _amount) {
             console.log(7);
             assert(_sender == address(MESSENGER));