From 7959bb82a507de432ab96c244a92374f05d30df0 Mon Sep 17 00:00:00 2001
From: antazoey <jules@apeworx.io>
Date: Tue, 23 Jan 2024 10:35:59 -0600
Subject: [PATCH] fix: allow easierly setting local gas lmt (#1878)

---
 src/ape_ethereum/ecosystem.py        | 17 +++++++++++++----
 tests/functional/test_network_api.py |  1 +
 2 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/src/ape_ethereum/ecosystem.py b/src/ape_ethereum/ecosystem.py
index f0c4c22fe8..4f6dc31a35 100644
--- a/src/ape_ethereum/ecosystem.py
+++ b/src/ape_ethereum/ecosystem.py
@@ -137,10 +137,12 @@ class ForkedNetworkConfig(NetworkConfig):
 def create_local_network_config(
     default_provider: Optional[str] = None, use_fork: bool = False, **kwargs
 ):
+    if "gas_limit" not in kwargs:
+        kwargs["gas_limit"] = "max"
+
     return create_network_config(
         base_fee_multiplier=1.0,
         default_provider=default_provider,
-        gas_limit="max",
         required_confirmations=0,
         transaction_acceptance_timeout=DEFAULT_LOCAL_TRANSACTION_ACCEPTANCE_TIMEOUT,
         cls=ForkedNetworkConfig if use_fork else NetworkConfig,
@@ -167,6 +169,7 @@ class BaseEthereumConfig(PluginConfig):
     """
 
     DEFAULT_TRANSACTION_TYPE: ClassVar[int] = TransactionType.DYNAMIC.value
+    DEFAULT_LOCAL_GAS_LIMIT: ClassVar[GasLimit] = "max"
     NETWORKS: ClassVar[Dict[str, Tuple[int, int]]] = NETWORKS
 
     default_network: str = LOCAL_NETWORK_NAME
@@ -189,7 +192,9 @@ def load_network_configs(cls, values):
             if net_name.endswith("_fork"):
                 key = net_name.replace("_fork", "")
                 default_fork_model = create_local_network_config(
-                    use_fork=True, default_transaction_type=cls.DEFAULT_TRANSACTION_TYPE
+                    use_fork=True,
+                    default_transaction_type=cls.DEFAULT_TRANSACTION_TYPE,
+                    gas_limit=cls.DEFAULT_LOCAL_GAS_LIMIT,
                 ).model_dump(mode="json", by_alias=True)
                 data = merge_configs(default_fork_model, obj)
                 cfg_forks[key] = ForkedNetworkConfig.model_validate(data)
@@ -209,7 +214,9 @@ def load_network_configs(cls, values):
     @cached_property
     def local(self) -> NetworkConfig:
         return create_local_network_config(
-            default_provider="test", default_transaction_type=self.DEFAULT_TRANSACTION_TYPE
+            default_provider="test",
+            default_transaction_type=self.DEFAULT_TRANSACTION_TYPE,
+            gas_limit=self.DEFAULT_LOCAL_GAS_LIMIT,
         )
 
     def __getattr__(self, key: str) -> Any:
@@ -254,7 +261,9 @@ def _get_forked_config(self, name: str) -> Optional[ForkedNetworkConfig]:
         if live_cfg := self.get(live_key):
             if isinstance(live_cfg, NetworkConfig):
                 fork_cfg = create_local_network_config(
-                    use_fork=True, default_transaction_type=self.DEFAULT_TRANSACTION_TYPE
+                    use_fork=True,
+                    default_transaction_type=self.DEFAULT_TRANSACTION_TYPE,
+                    gas_limit=self.DEFAULT_LOCAL_GAS_LIMIT,
                 )
                 self._forked_configs[live_key] = fork_cfg
                 return fork_cfg
diff --git a/tests/functional/test_network_api.py b/tests/functional/test_network_api.py
index de8a9150ae..ccc93cdabe 100644
--- a/tests/functional/test_network_api.py
+++ b/tests/functional/test_network_api.py
@@ -93,6 +93,7 @@ def test_forked_network_with_config(temp_config, ethereum):
         assert cfg.base_fee_multiplier == 1.0
         assert cfg.transaction_acceptance_timeout == 20
         assert cfg.max_receipt_retries == 20
+        assert cfg.gas_limit == "max"
 
 
 def test_data_folder_custom_network(custom_network, ethereum, custom_network_name_0):