test_version_module.py

import pytest

from web3 import (
    EthereumTesterProvider,
    Web3,
)
from web3.eth import (
    Eth,
)
from web3.providers.eth_tester.main import (
    AsyncEthereumTesterProvider,
)
from web3.version import (
    AsyncVersion,
    BlockingVersion,
    Version,
)

# This file is being left in since the Version module is
# being experimented on for async behavior. But, this file
# along with web3/version.py should be removed eventually.


@pytest.fixture
def blocking_w3():
    return Web3(
        EthereumTesterProvider(),
        modules={
            "blocking_version": (BlockingVersion,),
            "legacy_version": (Version,),
            "eth": (Eth,),
        },
    )


@pytest.fixture
def async_w3():
    return Web3(
        AsyncEthereumTesterProvider(),
        middlewares=[],
        modules={
            "async_version": (AsyncVersion,),
        },
    )


def test_legacy_version_deprecation(blocking_w3):
    with pytest.raises(DeprecationWarning):
        blocking_w3.legacy_version.node
    with pytest.raises(DeprecationWarning):
        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.api

    assert await async_w3.async_version.node == blocking_w3.client_version