Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

test: add EnumerableSet tests #12

Closed
wants to merge 1 commit into from
Closed

test: add EnumerableSet tests #12

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
144 changes: 144 additions & 0 deletions packages/contracts/test/libraries/EnumerableSet.t.sol
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,144 @@
// SPDX-License-Identifier: MIT
pragma solidity ^0.8.19;

import "forge-std/Test.sol";
import {EnumerableSet} from "../../src/dollar/libraries/EnumerableSet.sol";

contract EnumerableSetTest is Test {
using EnumerableSet for EnumerableSet.AddressSet;
using EnumerableSet for EnumerableSet.Bytes32Set;
using EnumerableSet for EnumerableSet.UintSet;

EnumerableSet.Bytes32Set mbs;
EnumerableSet.AddressSet mas;
EnumerableSet.UintSet mus;

function testEnumerableSetBytes32SetOperations() public {
// empty
assertEq(mbs.length(), 0);
assertEq(mbs.toArray().length, 0);
assertFalse(mbs.contains("a"));
// add
assertTrue(mbs.add("a"));
assertTrue(mbs.contains("a"));
assertEq(mbs.length(), 1);
assertTrue(mbs.add("b"));
assertEq(mbs.length(), 2);
// add "a" again
assertFalse(mbs.add("a"));
assertEq(mbs.length(), 2);
bytes32[] memory values = mbs.toArray();
assertEq("a", values[0]);
assertEq("b", values[1]);
assertTrue(mbs.add("c"));
assertTrue(mbs.add("d"));
assertEq(mbs.length(), 4);
// remove
// inner array: ["a","b","c","d"]
assertTrue(mbs.contains("b"));
assertTrue(mbs.remove("b"));
assertFalse(mbs.contains("b"));
assertEq(mbs.length(), 3);
// remove "b" again
assertFalse(mbs.remove("b"));
assertEq(mbs.length(), 3);
// inner array after remove: ["a","d","c"]
assertEq(mbs.at(0), "a");
assertEq(mbs.at(1), "d");
assertEq(mbs.at(2), "c");
// check values()
values = mbs.toArray();
assertEq("a", values[0]);
assertEq("d", values[1]);
assertEq("c", values[2]);
// revert if out of bounds
vm.expectRevert();
mbs.at(1024);
}

function testEnumerableSetAddressSetOperations() public {
// empty
assertEq(mas.length(), 0);
assertEq(mas.toArray().length, 0);
assertFalse(mas.contains(address(1)));
// add
assertTrue(mas.add(address(1)));
assertTrue(mas.contains(address(1)));
assertEq(mas.length(), 1);
assertTrue(mas.add(address(2)));
assertEq(mas.length(), 2);
// add address(1) again
assertFalse(mas.add(address(1)));
assertEq(mas.length(), 2);
address[] memory values = mas.toArray();
assertEq(address(1), values[0]);
assertEq(address(2), values[1]);
assertTrue(mas.add(address(4)));
assertTrue(mas.add(address(8)));
assertEq(mas.length(), 4);
// remove
// inner array: [address(1),address(2),address(4),address(8)]
assertTrue(mas.contains(address(2)));
assertTrue(mas.remove(address(2)));
assertFalse(mas.contains(address(2)));
assertEq(mas.length(), 3);
// remove address(2) again
assertFalse(mas.remove(address(2)));
assertEq(mas.length(), 3);
// inner array after remove: [address(1),address(8),address(4)]
assertEq(mas.at(0), address(1));
assertEq(mas.at(1), address(8));
assertEq(mas.at(2), address(4));
// check values()
values = mas.toArray();
assertEq(address(1), values[0]);
assertEq(address(8), values[1]);
assertEq(address(4), values[2]);
// revert if out of bounds
vm.expectRevert();
mas.at(1024);
}

function testEnumerableSetUintSetOperations() public {
// empty
assertEq(mus.length(), 0);
assertEq(mus.toArray().length, 0);
assertFalse(mus.contains(1));
// add
assertTrue(mus.add(1));
assertTrue(mus.contains(1));
assertEq(mus.length(), 1);
assertTrue(mus.add(2));
assertEq(mus.length(), 2);
// add 1 again
assertFalse(mus.add(1));
assertEq(mus.length(), 2);
uint[] memory values = mus.toArray();
assertEq(1, values[0]);
assertEq(2, values[1]);
assertTrue(mus.add(4));
assertTrue(mus.add(8));
assertEq(mus.length(), 4);
// remove
// inner array: [1,2,4,8]
assertTrue(mus.contains(2));
assertTrue(mus.remove(2));
assertFalse(mus.contains(2));
assertEq(mus.length(), 3);
// remove 2 again
assertFalse(mus.remove(2));
assertEq(mus.length(), 3);
// inner array after remove: [1,8,4]
assertEq(mus.at(0), 1);
assertEq(mus.at(1), 8);
assertEq(mus.at(2), 4);
// check values()
values = mus.toArray();
assertEq(1, values[0]);
assertEq(8, values[1]);
assertEq(4, values[2]);
// revert if out of bounds
vm.expectRevert();
mus.at(1024);
}
}
Loading