Merge pull request #144 from solidstate-network/EnumerableSetTests

Update and test `EnumerableSet`

Author: ItsNickBarry

contracts/data/EnumerableSet.sol

@@ -150,6 +150,57 @@ library EnumerableSet {
         return _remove(set._inner, bytes32(value));
     }
 
+    function toArray(Bytes32Set storage set)
+        internal
+        view
+        returns (bytes32[] memory)
+    {
+        uint256 len = _length(set._inner);
+        bytes32[] memory arr = new bytes32[](len);
+
+        unchecked {
+            for (uint256 index; index < len; ++index) {
+                arr[index] = at(set, index);
+            }
+        }
+
+        return arr;
+    }
+
+    function toArray(AddressSet storage set)
+        internal
+        view
+        returns (address[] memory)
+    {
+        uint256 len = _length(set._inner);
+        address[] memory arr = new address[](len);
+
+        unchecked {
+            for (uint256 index; index < len; ++index) {
+                arr[index] = at(set, index);
+            }
+        }
+
+        return arr;
+    }
+
+    function toArray(UintSet storage set)
+        internal
+        view
+        returns (uint256[] memory)
+    {
+        uint256 len = _length(set._inner);
+        uint256[] memory arr = new uint256[](len);
+
+        unchecked {
+            for (uint256 index; index < len; ++index) {
+                arr[index] = at(set, index);
+            }
+        }
+
+        return arr;
+    }
+
     function _at(Set storage set, uint256 index)
         private
         view

contracts/data/EnumerableSetAddressMock.sol

@@ -0,0 +1,39 @@
+// SPDX-License-Identifier: MIT
+
+pragma solidity ^0.8.8;
+
+import { EnumerableSet } from './EnumerableSet.sol';
+
+contract EnumerableSetAddressMock {
+    using EnumerableSet for EnumerableSet.AddressSet;
+
+    EnumerableSet.AddressSet internal addressSet;
+
+    function at(uint256 index) external view returns (address) {
+        return addressSet.at(index);
+    }
+
+    function contains(address value) external view returns (bool) {
+        return addressSet.contains(value);
+    }
+
+    function indexOf(address value) external view returns (uint256) {
+        return addressSet.indexOf(value);
+    }
+
+    function length() external view returns (uint256) {
+        return addressSet.length();
+    }
+
+    function add(address value) external returns (bool) {
+        return addressSet.add(value);
+    }
+
+    function remove(address value) external returns (bool) {
+        return addressSet.remove(value);
+    }
+
+    function toArray() external view returns (address[] memory) {
+        return addressSet.toArray();
+    }
+}

contracts/data/EnumerableSetBytes32Mock.sol

@@ -0,0 +1,39 @@
+// SPDX-License-Identifier: MIT
+
+pragma solidity ^0.8.8;
+
+import { EnumerableSet } from './EnumerableSet.sol';
+
+contract EnumerableSetBytes32Mock {
+    using EnumerableSet for EnumerableSet.Bytes32Set;
+
+    EnumerableSet.Bytes32Set internal bytes32Set;
+
+    function at(uint256 index) external view returns (bytes32) {
+        return bytes32Set.at(index);
+    }
+
+    function contains(bytes32 value) external view returns (bool) {
+        return bytes32Set.contains(value);
+    }
+
+    function indexOf(bytes32 value) external view returns (uint256) {
+        return bytes32Set.indexOf(value);
+    }
+
+    function length() external view returns (uint256) {
+        return bytes32Set.length();
+    }
+
+    function add(bytes32 value) external returns (bool) {
+        return bytes32Set.add(value);
+    }
+
+    function remove(bytes32 value) external returns (bool) {
+        return bytes32Set.remove(value);
+    }
+
+    function toArray() external view returns (bytes32[] memory) {
+        return bytes32Set.toArray();
+    }
+}

contracts/data/EnumerableSetUintMock.sol

@@ -0,0 +1,39 @@
+// SPDX-License-Identifier: MIT
+
+pragma solidity ^0.8.8;
+
+import { EnumerableSet } from './EnumerableSet.sol';
+
+contract EnumerableSetUintMock {
+    using EnumerableSet for EnumerableSet.UintSet;
+
+    EnumerableSet.UintSet internal uintSet;
+
+    function at(uint256 index) external view returns (uint256) {
+        return uintSet.at(index);
+    }
+
+    function contains(uint256 value) external view returns (bool) {
+        return uintSet.contains(value);
+    }
+
+    function indexOf(uint256 value) external view returns (uint256) {
+        return uintSet.indexOf(value);
+    }
+
+    function length() external view returns (uint256) {
+        return uintSet.length();
+    }
+
+    function add(uint256 value) external returns (bool) {
+        return uintSet.add(value);
+    }
+
+    function remove(uint256 value) external returns (bool) {
+        return uintSet.remove(value);
+    }
+
+    function toArray() external view returns (uint256[] memory) {
+        return uintSet.toArray();
+    }
+}