logos-messaging/logos-messaging-rlnv2-contract
1
0
Fork 0
mirror of https://github.com/logos-messaging/logos-messaging-rlnv2-contract.git synced 2026-01-08 17:03:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: test getCommitments too

Browse Source
This commit is contained in:
rymnc 2024-05-22 18:17:32 +05:30
parent db2701250f
commit 82e7ce58be
No known key found for this signature in database
GPG Key ID: AAA088D5C68ECD34
3 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.gas-snapshot
View File

@ -1 +1 @@
WakuRlnV2Test:test__ValidRegistration() (gas: 108659)
WakuRlnV2Test:test__ValidRegistration() (gas: 108661)

9
src/WakuRlnV2.sol
View File

@ -133,8 +133,7 @@ contract WakuRlnV2 {
if (idCommitmentIndex >= SET_SIZE) revert FullTree();
uint256 rateCommitment = PoseidonT3.hash([idCommitment, userMessageLimit]);
MembershipInfo memory member =
MembershipInfo({ userMessageLimit: uint32(userMessageLimit), index: idCommitmentIndex });
MembershipInfo memory member = MembershipInfo({ userMessageLimit: userMessageLimit, index: idCommitmentIndex });
LazyIMT.insert(imtData, rateCommitment);
memberInfo[idCommitment] = member;
@ -150,13 +149,13 @@ contract WakuRlnV2 {
return imtData.elements[LazyIMT.indexForElement(0, index)];
}
function getCommitments(uint256 startIndex, uint256 endIndex) public view returns (uint256[] memory) {
function getCommitments(uint32 startIndex, uint32 endIndex) public view returns (uint256[] memory) {
if (startIndex >= endIndex) revert InvalidPaginationQuery(startIndex, endIndex);
if (endIndex > idCommitmentIndex) revert InvalidPaginationQuery(startIndex, endIndex);
uint256[] memory commitments = new uint256[](endIndex - startIndex);
for (uint256 i = startIndex; i < endIndex; i++) {
commitments[i - startIndex] = indexToCommitment(uint32(i));
for (uint32 i = startIndex; i < endIndex; i++) {
commitments[i - startIndex] = indexToCommitment(i);
}
return commitments;
}

3
test/WakuRlnV2.t.sol
View File

@ -34,6 +34,9 @@ contract WakuRlnV2Test is Test {
assertEq(index, 0);
uint256 rateCommitment = PoseidonT3.hash([idCommitment, userMessageLimit]);
assertEq(w.indexToCommitment(0), rateCommitment);
uint256[] memory commitments = w.getCommitments(0, 1);
assertEq(commitments.length, 1);
assertEq(commitments[0], rateCommitment);
vm.resumeGasMetering();
}
}