Merge pull request #1 from Kwenta/internal-bootstrap-helpers

🧹 Update bootstrap helpers to be internal
Kwenta · Oct 25, 2023 · 0a5f854 · 0a5f854
2 parents c499e48 + 2921a70
commit 0a5f854
Showing 1 changed file with 18 additions and 3 deletions.
diff --git a/test/utils/Bootstrap.sol b/test/utils/Bootstrap.sol
@@ -13,16 +13,23 @@ import {Test} from "lib/forge-std/src/Test.sol";
 contract Bootstrap is Test {
     using console2 for *;
 
-    Counter public counter;
+    Counter internal counter;
 
-    function initializeOptimismGoerli() public {
+    function initializeLocal() internal {
+        BootstrapLocal bootstrap = new BootstrapLocal();
+        (address counterAddress) = bootstrap.init();
+
+        counter = Counter(counterAddress);
+    }
+
+    function initializeOptimismGoerli() internal {
         BootstrapOptimismGoerli bootstrap = new BootstrapOptimismGoerli();
         (address counterAddress) = bootstrap.init();
 
         counter = Counter(counterAddress);
     }
 
-    function initializeOptimism() public {
+    function initializeOptimism() internal {
         BootstrapOptimismGoerli bootstrap = new BootstrapOptimismGoerli();
         (address counterAddress) = bootstrap.init();
 
@@ -32,6 +39,14 @@ contract Bootstrap is Test {
     /// @dev add other networks here as needed (ex: Base, BaseGoerli)
 }
 
+contract BootstrapLocal is Setup {
+    function init() public returns (address) {
+        address counterAddress = Setup.deploySystem();
+
+        return counterAddress;
+    }
+}
+
 contract BootstrapOptimism is Setup, OptimismParameters {
     function init() public returns (address) {
         address counterAddress = Setup.deploySystem();