fix: custom config number of test accounts (ApeWorX#918)

Author: NotPeopling2day

src/ape_geth/providers.py

@@ -17,7 +17,7 @@
 from geth.chain import initialize_chain  # type: ignore
 from geth.process import BaseGethProcess  # type: ignore
 from geth.wrapper import construct_test_chain_kwargs  # type: ignore
-from pydantic import Extra
+from pydantic import Extra, PositiveInt
 from requests.exceptions import ConnectionError
 from web3 import HTTPProvider, Web3
 from web3.exceptions import ExtraDataLengthError
@@ -47,7 +47,7 @@ def __init__(
         hostname: str,
         port: int,
         mnemonic: str,
-        number_of_accounts: int,
+        number_of_accounts: PositiveInt,
         chain_id: int = 1337,
         initial_balance: Union[str, int] = to_wei(10000, "ether"),
     ):

src/ape_test/__init__.py

@@ -1,3 +1,5 @@
+from pydantic import PositiveInt
+
 from ape import plugins
 from ape.api import PluginConfig
 from ape.api.networks import LOCAL_NETWORK_NAME
@@ -9,7 +11,7 @@
 
 class Config(PluginConfig):
     mnemonic: str = DEFAULT_TEST_MNEMONIC
-    number_of_accounts: int = DEFAULT_NUMBER_OF_TEST_ACCOUNTS
+    number_of_accounts: PositiveInt = DEFAULT_NUMBER_OF_TEST_ACCOUNTS
 
 
 @plugins.register(plugins.Config)

src/ape_test/accounts.py

@@ -17,7 +17,7 @@ def config(self):
     @cached_property
     def _dev_accounts(self) -> List[GeneratedDevAccount]:
         mnemonic = self.config["mnemonic"]
-        return generate_dev_accounts(mnemonic)
+        return generate_dev_accounts(mnemonic, number_of_accounts=self.config["number_of_accounts"])
 
     @property
     def aliases(self) -> Iterator[str]:

tests/functional/test_accounts.py

@@ -257,3 +257,20 @@ def test_unlock_from_prompt_and_sign_transaction(runner, keyfile_account, receiv
     with runner.isolation(input="y\n"):
         receipt = keyfile_account.transfer(receiver, "1 gwei")
         assert receipt.receiver == receiver
+
+
+def test_custom_num_of_test_accts_config(test_accounts, temp_config):
+    from ape.utils.testing import DEFAULT_NUMBER_OF_TEST_ACCOUNTS
+
+    CUSTOM_NUMBER_OF_TEST_ACCOUNTS = 20
+
+    test_config = {
+        "test": {
+            "number_of_accounts": CUSTOM_NUMBER_OF_TEST_ACCOUNTS,
+        }
+    }
+
+    assert len(test_accounts) == DEFAULT_NUMBER_OF_TEST_ACCOUNTS
+
+    with temp_config(test_config):
+        assert len(test_accounts) == CUSTOM_NUMBER_OF_TEST_ACCOUNTS