Skip to content

Commit 42a08e0

Browse files
committed
Add receiving test
1 parent df515bd commit 42a08e0

File tree

1 file changed

+28
-1
lines changed

1 file changed

+28
-1
lines changed

evm/test/NttManager.t.sol

+28-1
Original file line numberDiff line numberDiff line change
@@ -92,8 +92,8 @@ contract TestNttManager is Test, INttManagerEvents, IRateLimiterEvents {
9292

9393
t.mintDummy(address(user_A), 5 * 10 ** decimals);
9494

95+
// Test outgoing transfers complete successfully with rate limit disabled
9596
vm.startPrank(user_A);
96-
9797
t.approve(address(nttManagerZeroRateLimiter), 3 * 10 ** decimals);
9898

9999
uint64 s1 = nttManagerZeroRateLimiter.transfer(
@@ -105,10 +105,37 @@ contract TestNttManager is Test, INttManagerEvents, IRateLimiterEvents {
105105
uint64 s3 = nttManagerZeroRateLimiter.transfer(
106106
1 * 10 ** decimals, chainId, toWormholeFormat(user_B)
107107
);
108+
vm.stopPrank();
108109

109110
assertEq(s1, 0);
110111
assertEq(s2, 1);
111112
assertEq(s3, 2);
113+
114+
// Test incoming transfer completes successfully with rate limit disabled
115+
(DummyTransceiver e1,) = TransceiverHelpersLib.setup_transceivers(nttManagerZeroRateLimiter);
116+
nttManagerZeroRateLimiter.setThreshold(2);
117+
118+
// register nttManager peer
119+
bytes32 peer = toWormholeFormat(address(nttManager));
120+
nttManagerZeroRateLimiter.setPeer(TransceiverHelpersLib.SENDING_CHAIN_ID, peer, 9);
121+
122+
TransceiverStructs.NttManagerMessage memory nttManagerMessage;
123+
bytes memory transceiverMessage;
124+
(nttManagerMessage, transceiverMessage) = TransceiverHelpersLib
125+
.buildTransceiverMessageWithNttManagerPayload(
126+
0,
127+
bytes32(0),
128+
peer,
129+
toWormholeFormat(address(nttManagerZeroRateLimiter)),
130+
abi.encode("payload")
131+
);
132+
133+
e1.receiveMessage(transceiverMessage);
134+
135+
bytes32 hash = TransceiverStructs.nttManagerMessageDigest(
136+
TransceiverHelpersLib.SENDING_CHAIN_ID, nttManagerMessage
137+
);
138+
assertEq(nttManagerZeroRateLimiter.messageAttestations(hash), 1);
112139
}
113140

114141
// === ownership

0 commit comments

Comments
 (0)