pywiki/tests/sdk-pyhmy/test_contract.py

import pytest

from pyhmy import contract

from pyhmy.rpc import exceptions

explorer_endpoint = "http://localhost:9599"
contract_tx_hash = "0xa605852dd2fa39ed42e101c17aaca9d344d352ba9b24b14b9af94ec9cb58b31f"
# deployedBytecode from json file
contract_code = "0x6080604052348015600f57600080fd5b506004361060285760003560e01c80634936cd3614602d575b600080fd5b604080516001815290519081900360200190f3fea2646970667358221220fa3fa0e8d0267831a59f4dd5edf39a513d07e98461cb06660ad28d4beda744cd64736f6c634300080f0033"
contract_address = None
fake_shard = "http://example.com"


def _test_contract_rpc(fn, *args, **kwargs):
    if not callable(fn):
        pytest.fail(f"Invalid function: {fn}")

    try:
        response = fn(*args, **kwargs)
    except Exception as e:
        if isinstance(
            e, exceptions.RPCError
        ) and "does not exist/is not available" in str(e):
            pytest.skip(f"{str(e)}")
        elif isinstance(e, exceptions.RPCError) and "estimateGas returned" in str(e):
            pytest.skip(f"{str(e)}")
        pytest.fail(f"Unexpected error: {e.__class__} {e}")
    return response


def test_get_contract_address_from_hash(setup_blockchain):
    global contract_address
    contract_address = _test_contract_rpc(
        contract.get_contract_address_from_hash, contract_tx_hash
    )
    assert isinstance(contract_address, str)


def test_call(setup_blockchain):
    if not contract_address:
        pytest.skip("Contract address not loaded yet")
    called = _test_contract_rpc(contract.call, contract_address, "latest")
    assert isinstance(called, str) and called.startswith("0x")


def test_estimate_gas(setup_blockchain):
    if not contract_address:
        pytest.skip("Contract address not loaded yet")
    gas = _test_contract_rpc(contract.estimate_gas, contract_address)
    assert isinstance(gas, int)


def test_get_code(setup_blockchain):
    if not contract_address:
        pytest.skip("Contract address not loaded yet")
    code = _test_contract_rpc(contract.get_code, contract_address, "latest")
    assert code == contract_code


def test_get_storage_at(setup_blockchain):
    if not contract_address:
        pytest.skip("Contract address not loaded yet")
    storage = _test_contract_rpc(
        contract.get_storage_at, contract_address, "0x0", "latest"
    )
    assert isinstance(storage, str) and storage.startswith("0x")


def test_errors():
    with pytest.raises(exceptions.RPCError):
        contract.get_contract_address_from_hash("", fake_shard)
    with pytest.raises(exceptions.RPCError):
        contract.call("", "", endpoint=fake_shard)
    with pytest.raises(exceptions.RPCError):
        contract.estimate_gas("", endpoint=fake_shard)
    with pytest.raises(exceptions.RPCError):
        contract.get_code("", "latest", endpoint=fake_shard)
    with pytest.raises(exceptions.RPCError):
        contract.get_storage_at("", 1, "latest", endpoint=fake_shard)