fix(test): use _mockAdaptersForwardMessage

Moonsong-Labs · Apr 30, 2024 · bb3c20e · bb3c20e
1 parent 8d0bfb1
commit bb3c20e
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 5 deletions.
diff --git a/tests/BaseCCForwarder.t.sol b/tests/BaseCCForwarder.t.sol
@@ -239,11 +239,6 @@ contract BaseCCForwarderTest is BaseTest, CrossChainForwarder {
       abi.encodeWithSelector(IBaseAdapter.setupPayments.selector),
       abi.encode()
     );
-    vm.mockCall(
-      address(currentChainBridgeAdapter),
-      abi.encodeWithSelector(IBaseAdapter.forwardMessage.selector),
-      abi.encode()
-    );
     _enableBridgeAdapters(bridgeAdaptersInfo);
   }
 

diff --git a/tests/Forwarder.t.sol b/tests/Forwarder.t.sol
@@ -406,6 +406,7 @@ contract ForwarderTest is BaseCCForwarderTest {
     validateTransactionNonceIncrement
     validateTransactionRegistry(extendedTx)
   {
+    _mockAdaptersForwardMessage(extendedTx.envelope.destinationChainId);
     UsedAdapter[] memory usedAdapters = _currentlyUsedAdaptersByChain[
       extendedTx.envelope.destinationChainId
     ];