@@ -76,6 +76,10 @@ contract TokenRouterTest is Test, ITokenRouterEvents {
76
76
ITokenRouter router;
77
77
SigningWormholeSimulator wormholeSimulator;
78
78
CircleSimulator circleSimulator;
79
+
80
+ // Core bridge.
81
+ uint256 immutable WORMHOLE_MESSAGE_FEE_STORAGE_SLOT = 7 ;
82
+ uint256 immutable WORMHOLE_MESSAGE_FEE = 1 ;
79
83
80
84
function deployProxy (address _token , address _wormhole , address _tokenMessenger )
81
85
internal
@@ -128,6 +132,9 @@ contract TokenRouterTest is Test, ITokenRouterEvents {
128
132
129
133
circleSimulator = new CircleSimulator (TESTING_SIGNER, MESSAGE_TRANSMITTER);
130
134
circleSimulator.setupCircleAttester ();
135
+
136
+ // Set core bridge messageFee > 0
137
+ vm.store (address (wormhole), bytes32 (uint256 (WORMHOLE_MESSAGE_FEE_STORAGE_SLOT)), bytes32 (uint256 (WORMHOLE_MESSAGE_FEE)));
131
138
}
132
139
133
140
/**
@@ -1607,7 +1614,9 @@ contract TokenRouterTest is Test, ITokenRouterEvents {
1607
1614
vm.recordLogs ();
1608
1615
1609
1616
// Place the order.
1610
- _router.placeMarketOrder (
1617
+ _router.placeMarketOrder {
1618
+ value: WORMHOLE_MESSAGE_FEE
1619
+ }(
1611
1620
amountIn, minAmountOut, targetChain, redeemer, redeemerMessage, refundAddress
1612
1621
);
1613
1622
@@ -1637,7 +1646,7 @@ contract TokenRouterTest is Test, ITokenRouterEvents {
1637
1646
vm.recordLogs ();
1638
1647
1639
1648
// Place the order.
1640
- _router.placeMarketOrder (amountIn, targetChain, redeemer, redeemerMessage);
1649
+ _router.placeMarketOrder {value: WORMHOLE_MESSAGE_FEE} (amountIn, targetChain, redeemer, redeemerMessage);
1641
1650
1642
1651
// Fetch the logs for Wormhole message.
1643
1652
Vm.Log[] memory logs = vm.getRecordedLogs ();
@@ -1663,7 +1672,9 @@ contract TokenRouterTest is Test, ITokenRouterEvents {
1663
1672
vm.recordLogs ();
1664
1673
1665
1674
// Place the order.
1666
- _router.placeFastMarketOrder (
1675
+ _router.placeFastMarketOrder {
1676
+ value: WORMHOLE_MESSAGE_FEE * 2
1677
+ }(
1667
1678
expectedOrder.amountIn,
1668
1679
expectedOrder.minAmountOut,
1669
1680
expectedOrder.targetChain,
@@ -1708,7 +1719,9 @@ contract TokenRouterTest is Test, ITokenRouterEvents {
1708
1719
vm.recordLogs ();
1709
1720
1710
1721
// Place the order.
1711
- _router.placeFastMarketOrder (
1722
+ _router.placeFastMarketOrder {
1723
+ value: WORMHOLE_MESSAGE_FEE * 2
1724
+ }(
1712
1725
amountIn, targetChain, redeemer, redeemerMessage, maxFee, deadline
1713
1726
);
1714
1727
0 commit comments