Demo async/blocking module definition with web3.Version

Author: dylanjw

tests/core/version-module/test_version_module.py

@@ -0,0 +1,52 @@
+import pytest
+
+from web3 import (
+    EthereumTesterProvider,
+    Web3,
+)
+from web3.providers.eth_tester.main import (
+    AsyncEthereumTesterProvider,
+)
+from web3.version import (
+    AsyncVersion,
+    BlockingVersion,
+    Version,
+)
+
+
+@pytest.fixture
+def blocking_w3():
+    return Web3(
+        EthereumTesterProvider(),
+        modules={
+            'blocking_version': BlockingVersion,
+            'legacy_version': Version
+        })
+
+
+@pytest.fixture
+def async_w3():
+    return Web3(
+        AsyncEthereumTesterProvider(),
+        middlewares=[],
+        modules={
+            'async_version': AsyncVersion,
+        })
+
+
+def test_blocking_version(blocking_w3):
+    assert blocking_w3.blocking_version.api == blocking_w3.legacy_version.api
+    assert blocking_w3.blocking_version.node == blocking_w3.legacy_version.node
+    assert blocking_w3.blocking_version.ethereum == blocking_w3.legacy_version.ethereum
+
+
+@pytest.mark.asyncio
+async def test_async_blocking_version(async_w3, blocking_w3):
+    assert async_w3.async_version.api == blocking_w3.legacy_version.api
+
+    assert await async_w3.async_version.node == blocking_w3.legacy_version.node
+    with pytest.raises(
+        ValueError,
+        message="RPC Endpoint has not been implemented: eth_protocolVersion"
+    ):
+        assert await async_w3.async_version.ethereum == blocking_w3.legacy_version.ethereum

web3/version.py

@@ -1,8 +1,46 @@
+from web3.method import (
+    Method,
+)
 from web3.module import (
     Module,
+    ModuleV2,
 )
 
 
+class BaseVersion(ModuleV2):
+    retrieve_caller_fn = None
+
+    _get_node_version = Method('web3_clientVersion')
+    _get_protocol_version = Method('eth_protocolVersion')
+
+    @property
+    def api(self):
+        from web3 import __version__
+        return __version__
+
+
+class AsyncVersion(BaseVersion):
+    is_async = True
+
+    @property
+    async def node(self):
+        return await self._get_node_version()
+
+    @property
+    async def ethereum(self):
+        return await self._get_protocol_version()
+
+
+class BlockingVersion(BaseVersion):
+    @property
+    def node(self):
+        return self._get_node_version()
+
+    @property
+    def ethereum(self):
+        return self._get_protocol_version()
+
+
 class Version(Module):
     @property
     def api(self):