Add async chain_id setter

Author: kclowes

tests/core/eth-module/test_eth_properties.py

@@ -1,5 +1,23 @@
 import pytest
 
+from web3 import Web3
+from web3.eth import (
+    AsyncEth,
+)
+from web3.providers.eth_tester.main import (
+    AsyncEthereumTesterProvider,
+)
+
+
+@pytest.fixture
+def async_w3():
+    return Web3(
+        AsyncEthereumTesterProvider(),
+        middlewares=[],
+        modules={
+            'eth': (AsyncEth,),
+        })
+
 
 def test_eth_protocol_version(w3):
     with pytest.warns(DeprecationWarning):
@@ -28,3 +46,14 @@ def test_set_chain_id(w3):
 
     w3.eth.chain_id = None
     assert w3.eth.chain_id == 61
+
+
+@pytest.mark.asyncio
+async def test_async_set_chain_id(async_w3):
+    assert await async_w3.eth.chain_id == 61
+
+    async_w3.eth.chain_id = 72
+    assert await async_w3.eth.chain_id == 72
+
+    async_w3.eth.chain_id = None
+    assert await async_w3.eth.chain_id == 61

web3/_utils/module_testing/web3_module.py

@@ -179,7 +179,7 @@ def test_solidityKeccak(
         self, w3: "Web3", types: Sequence[TypeStr], values: Sequence[Any], expected: HexBytes
     ) -> None:
         if isinstance(expected, type) and issubclass(expected, Exception):
-            with pytest.raises(expected):  # type: ignore
+            with pytest.raises(expected):
                 w3.solidityKeccak(types, values)
             return
 

web3/eth.py

@@ -358,7 +358,14 @@ async def block_number(self) -> BlockNumber:
 
     @property
     async def chain_id(self) -> int:
-        return await self._chain_id()  # type: ignore
+        if self._default_chain_id is None:
+            return await self._chain_id()  # type: ignore
+        else:
+            return self._default_chain_id
+
+    @chain_id.setter
+    def chain_id(self, value: int) -> None:
+        self._default_chain_id = value
 
     @property
     async def coinbase(self) -> ChecksumAddress: