Merge branch 'master' of github.com:vyperlang/titanoboa into etherscan

boa/__init__.py

@@ -37,25 +37,20 @@
 
 @contextlib.contextmanager
 def swap_env(new_env):
-    old_env = env
-    try:
-        set_env(new_env)
+    with set_env(new_env):
         yield
-    finally:
-        set_env(old_env)
 
 
-def set_env(new_env):
+def _set_env(new):
     global env
-    env = new_env
+    env = new
+    Env._singleton = new
 
-    Env._singleton = new_env
 
-
-def _env_mgr(new_env):
+def set_env(new_env):
     global env
     get_env = lambda: env  # noqa: E731
-    return Open(get_env, set_env, new_env)
+    return Open(get_env, _set_env, new_env)
 
 
 def fork(
@@ -69,20 +64,20 @@ def fork(
 
     new_env = Env()
     new_env.fork(url=url, block_identifier=block_identifier, deprecated=False, **kwargs)
-    return _env_mgr(new_env)
+    return set_env(new_env)
 
 
 def set_browser_env(address=None):
     """Set the environment to use the browser's network in Jupyter/Colab"""
     # import locally because jupyter is generally not installed
     from boa.integrations.jupyter import BrowserEnv
 
-    return _env_mgr(BrowserEnv(address))
+    return set_env(BrowserEnv(address))
 
 
 def set_network_env(url):
     """Set the environment to use a custom network URL"""
-    return _env_mgr(NetworkEnv.from_url(url))
+    return set_env(NetworkEnv.from_url(url))
 
 
 def set_etherscan(*args, **kwargs):

boa/contracts/vvm/vvm_contract.py

@@ -54,7 +54,7 @@ def constructor(self):
                 return ABIFunction(t, contract_name=self.filename)
         return None
 
-    def deploy(self, *args, env=None):
+    def deploy(self, *args, env=None, **kwargs):
         encoded_args = b""
         if self.constructor is not None:
             encoded_args = self.constructor.prepare_calldata(*args)
@@ -64,7 +64,7 @@ def deploy(self, *args, env=None):
         if env is None:
             env = Env.get_singleton()
 
-        address, _ = env.deploy_code(bytecode=self.bytecode + encoded_args)
+        address, _ = env.deploy_code(bytecode=self.bytecode + encoded_args, **kwargs)
 
         return self.at(address)
 

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "titanoboa"
-version = "0.2.3"
+version = "0.2.4"
 description = "A Vyper interpreter"
 #authors = []
 license = { file = "LICENSE" }

tests/integration/network/sepolia/test_sepolia_env.py

@@ -43,18 +43,15 @@ def simple_contract():
 def verifier(request):
     if request.param == Blockscout:
         api_key = os.getenv("BLOCKSCOUT_API_KEY")
-        verifier = Blockscout("https://eth-sepolia.blockscout.com", api_key)
+        return Blockscout("https://eth-sepolia.blockscout.com", api_key)
     elif request.param == Etherscan:
         api_key = os.environ["ETHERSCAN_API_KEY"]
-        verifier = Etherscan("https://api-sepolia.etherscan.io/api", api_key)
-    else:
-        raise ValueError(f"Unknown verifier: {request.param}")
-    with boa.set_verifier(verifier):
-        yield verifier
+        return Etherscan("https://api-sepolia.etherscan.io/api", api_key)
+    raise ValueError(f"Unknown verifier: {request.param}")
 
 
 def test_verify(simple_contract, verifier):
-    result = boa.verify(simple_contract)
+    result = boa.verify(simple_contract, verifier)
     result.wait_for_verification()
     assert result.is_verified()
 

tests/unitary/contracts/vvm/test_vvm.py

@@ -37,3 +37,16 @@ def test_loads_vvm():
 
     assert contract.foo() == 42
     assert contract.bar() == 43
+
+
+def test_forward_args_on_deploy():
+    with open(mock_3_10_path) as f:
+        code = f.read()
+
+    contract_vvm_deployer = boa.loads_partial(code)
+
+    random_addy = boa.env.generate_address()
+
+    contract = contract_vvm_deployer.deploy(43, override_address=random_addy)
+
+    assert random_addy == contract.address

tests/unitary/test_boa.py

@@ -4,7 +4,7 @@
 def test_env_mgr_noctx():
     s = boa.env
     t = boa.Env()
-    boa._env_mgr(t)
+    boa.set_env(t)
     assert boa.env is not s
     assert boa.env is t
 
@@ -13,7 +13,7 @@ def test_env_mgr_with_ctx():
     s = boa.env
     t = boa.Env()
 
-    with boa._env_mgr(t):
+    with boa.set_env(t):
         assert boa.env is not s
         assert boa.env is t