InitializableMock.sol 2.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130
  1. // SPDX-License-Identifier: MIT
  2. pragma solidity ^0.8.20;
  3. import {Initializable} from "../proxy/utils/Initializable.sol";
  4. /**
  5. * @title InitializableMock
  6. * @dev This contract is a mock to test initializable functionality
  7. */
  8. contract InitializableMock is Initializable {
  9. bool public initializerRan;
  10. bool public onlyInitializingRan;
  11. uint256 public x;
  12. function isInitializing() public view returns (bool) {
  13. return _isInitializing();
  14. }
  15. function initialize() public initializer {
  16. initializerRan = true;
  17. }
  18. function initializeOnlyInitializing() public onlyInitializing {
  19. onlyInitializingRan = true;
  20. }
  21. function initializerNested() public initializer {
  22. initialize();
  23. }
  24. function onlyInitializingNested() public initializer {
  25. initializeOnlyInitializing();
  26. }
  27. function initializeWithX(uint256 _x) public payable initializer {
  28. x = _x;
  29. }
  30. function nonInitializable(uint256 _x) public payable {
  31. x = _x;
  32. }
  33. function fail() public pure {
  34. require(false, "InitializableMock forced failure");
  35. }
  36. }
  37. contract ConstructorInitializableMock is Initializable {
  38. bool public initializerRan;
  39. bool public onlyInitializingRan;
  40. constructor() initializer {
  41. initialize();
  42. initializeOnlyInitializing();
  43. }
  44. function initialize() public initializer {
  45. initializerRan = true;
  46. }
  47. function initializeOnlyInitializing() public onlyInitializing {
  48. onlyInitializingRan = true;
  49. }
  50. }
  51. contract ChildConstructorInitializableMock is ConstructorInitializableMock {
  52. bool public childInitializerRan;
  53. constructor() initializer {
  54. childInitialize();
  55. }
  56. function childInitialize() public initializer {
  57. childInitializerRan = true;
  58. }
  59. }
  60. contract ReinitializerMock is Initializable {
  61. uint256 public counter;
  62. function getInitializedVersion() public view returns (uint8) {
  63. return _getInitializedVersion();
  64. }
  65. function initialize() public initializer {
  66. doStuff();
  67. }
  68. function reinitialize(uint8 i) public reinitializer(i) {
  69. doStuff();
  70. }
  71. function nestedReinitialize(uint8 i, uint8 j) public reinitializer(i) {
  72. reinitialize(j);
  73. }
  74. function chainReinitialize(uint8 i, uint8 j) public {
  75. reinitialize(i);
  76. reinitialize(j);
  77. }
  78. function disableInitializers() public {
  79. _disableInitializers();
  80. }
  81. function doStuff() public onlyInitializing {
  82. counter++;
  83. }
  84. }
  85. contract DisableNew is Initializable {
  86. constructor() {
  87. _disableInitializers();
  88. }
  89. }
  90. contract DisableOld is Initializable {
  91. constructor() initializer {}
  92. }
  93. contract DisableBad1 is DisableNew, DisableOld {}
  94. contract DisableBad2 is Initializable {
  95. constructor() initializer {
  96. _disableInitializers();
  97. }
  98. }
  99. contract DisableOk is DisableOld, DisableNew {}