feat: add nickname to Env (#266)

I added the nickname attribute to the Env and NetworkEnv objects.

* added nicknames to envs
* fix url masking

---------

Co-authored-by: Charles Cooper <[email protected]>

boa/environment.py

@@ -43,6 +43,8 @@ def __init__(self, fork_try_prefetch_state=False, fast_mode_enabled=False):
 
         self._gas_tracker = 0
 
+        self.nickname = "pyevm"
+
         self.evm = PyEVM(self, fast_mode_enabled, fork_try_prefetch_state)
 
     def set_random_seed(self, seed=None):

boa/network.py

@@ -149,6 +149,7 @@ class NetworkEnv(Env):
     def __init__(
         self,
         rpc: str | RPC,
+        nickname: str = None,
         accounts: dict[str, Account] = None,
         fork_try_prefetch_state=True,
         **kwargs,
@@ -164,6 +165,8 @@ def __init__(
 
         self._rpc: RPC = rpc
 
+        self.nickname = nickname or rpc.name
+
         self._reset_fork()
 
         self._accounts = accounts or {}

boa/rpc.py

@@ -111,7 +111,10 @@ def name(self):
         # url stripped out (content which you might not want to end up
         # in logs)
         parse_result = urlparse(self._rpc_url)
-        return f"{parse_result.scheme}://{parse_result.netloc} (URL partially masked for privacy)"
+        partial_ret = f"{parse_result.scheme}://{parse_result.netloc}"
+        if partial_ret != self._rpc_url:
+            return f"{partial_ret} (URL partially masked for privacy)"
+        return self._rpc_url
 
     def fetch(self, method, params):
         # the obvious thing to do here is dispatch into fetch_multi.

tests/integration/network/anvil/test_network_env.py

@@ -36,6 +36,10 @@ def test_env_type():
     assert isinstance(boa.env, NetworkEnv)
 
 
+def test_network_env_nickname(free_port):
+    assert boa.env.nickname == f"http://localhost:{free_port}"
+
+
 def test_total_supply(simple_contract):
     assert simple_contract.totalSupply() == STARTING_SUPPLY
 

tests/unitary/test_env_defaults.py

@@ -0,0 +1,12 @@
+import pytest
+
+from boa import Env
+
+
+@pytest.fixture
+def env():
+    return Env()
+
+
+def test_env_nickname(env):
+    assert env.nickname == "pyevm"