EnumerableSetMockUpgradeable.sol 3.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132
  1. // SPDX-License-Identifier: MIT
  2. pragma solidity ^0.8.0;
  3. import "../utils/structs/EnumerableSetUpgradeable.sol";
  4. import "../proxy/utils/Initializable.sol";
  5. // Bytes32Set
  6. contract EnumerableBytes32SetMockUpgradeable is Initializable {
  7. function __EnumerableBytes32SetMock_init() internal onlyInitializing {
  8. __EnumerableBytes32SetMock_init_unchained();
  9. }
  10. function __EnumerableBytes32SetMock_init_unchained() internal onlyInitializing {
  11. }
  12. using EnumerableSetUpgradeable for EnumerableSetUpgradeable.Bytes32Set;
  13. event OperationResult(bool result);
  14. EnumerableSetUpgradeable.Bytes32Set private _set;
  15. function contains(bytes32 value) public view returns (bool) {
  16. return _set.contains(value);
  17. }
  18. function add(bytes32 value) public {
  19. bool result = _set.add(value);
  20. emit OperationResult(result);
  21. }
  22. function remove(bytes32 value) public {
  23. bool result = _set.remove(value);
  24. emit OperationResult(result);
  25. }
  26. function length() public view returns (uint256) {
  27. return _set.length();
  28. }
  29. function at(uint256 index) public view returns (bytes32) {
  30. return _set.at(index);
  31. }
  32. function values() public view returns (bytes32[] memory) {
  33. return _set.values();
  34. }
  35. uint256[48] private __gap;
  36. }
  37. // AddressSet
  38. contract EnumerableAddressSetMockUpgradeable is Initializable {
  39. function __EnumerableAddressSetMock_init() internal onlyInitializing {
  40. __EnumerableAddressSetMock_init_unchained();
  41. }
  42. function __EnumerableAddressSetMock_init_unchained() internal onlyInitializing {
  43. }
  44. using EnumerableSetUpgradeable for EnumerableSetUpgradeable.AddressSet;
  45. event OperationResult(bool result);
  46. EnumerableSetUpgradeable.AddressSet private _set;
  47. function contains(address value) public view returns (bool) {
  48. return _set.contains(value);
  49. }
  50. function add(address value) public {
  51. bool result = _set.add(value);
  52. emit OperationResult(result);
  53. }
  54. function remove(address value) public {
  55. bool result = _set.remove(value);
  56. emit OperationResult(result);
  57. }
  58. function length() public view returns (uint256) {
  59. return _set.length();
  60. }
  61. function at(uint256 index) public view returns (address) {
  62. return _set.at(index);
  63. }
  64. function values() public view returns (address[] memory) {
  65. return _set.values();
  66. }
  67. uint256[48] private __gap;
  68. }
  69. // UintSet
  70. contract EnumerableUintSetMockUpgradeable is Initializable {
  71. function __EnumerableUintSetMock_init() internal onlyInitializing {
  72. __EnumerableUintSetMock_init_unchained();
  73. }
  74. function __EnumerableUintSetMock_init_unchained() internal onlyInitializing {
  75. }
  76. using EnumerableSetUpgradeable for EnumerableSetUpgradeable.UintSet;
  77. event OperationResult(bool result);
  78. EnumerableSetUpgradeable.UintSet private _set;
  79. function contains(uint256 value) public view returns (bool) {
  80. return _set.contains(value);
  81. }
  82. function add(uint256 value) public {
  83. bool result = _set.add(value);
  84. emit OperationResult(result);
  85. }
  86. function remove(uint256 value) public {
  87. bool result = _set.remove(value);
  88. emit OperationResult(result);
  89. }
  90. function length() public view returns (uint256) {
  91. return _set.length();
  92. }
  93. function at(uint256 index) public view returns (uint256) {
  94. return _set.at(index);
  95. }
  96. function values() public view returns (uint256[] memory) {
  97. return _set.values();
  98. }
  99. uint256[48] private __gap;
  100. }