pywiki/tests/sdk-pyhmy/test_account.py

import pytest
import requests

from pyhmy import account

from pyhmy.rpc import exceptions


explorer_endpoint = "http://localhost:9700"
endpoint_shard_one = "http://localhost:9502"
local_test_address = "one155jp2y76nazx8uw5sa94fr0m4s5aj8e5xm6fu3"
test_validator_address = local_test_address
genesis_block_number = 0
test_block_number = 1
fake_shard = "http://example.com"


def _test_account_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)}")
        pytest.fail(f"Unexpected error: {e.__class__} {e}")
    return response


def test_get_balance(setup_blockchain):
    balance = _test_account_rpc(account.get_balance, local_test_address)
    assert isinstance(balance, int)
    assert balance > 0


def test_get_balance_by_block(setup_blockchain):
    balance = _test_account_rpc(
        account.get_balance_by_block, local_test_address, genesis_block_number
    )
    assert isinstance(balance, int)
    assert balance > 0


def test_get_account_nonce(setup_blockchain):
    true_nonce = _test_account_rpc(
        account.get_account_nonce,
        local_test_address,
        test_block_number,
        endpoint=endpoint_shard_one,
    )
    assert isinstance(true_nonce, int)


def test_get_transaction_history(setup_blockchain):
    tx_history = _test_account_rpc(
        account.get_transaction_history, local_test_address, endpoint=explorer_endpoint
    )
    assert isinstance(tx_history, list)
    assert len(tx_history) >= 0


def test_get_staking_transaction_history(setup_blockchain):
    staking_tx_history = _test_account_rpc(
        account.get_staking_transaction_history,
        test_validator_address,
        endpoint=explorer_endpoint,
    )
    assert isinstance(staking_tx_history, list)
    assert len(staking_tx_history) > 0


def test_get_balance_on_all_shards(setup_blockchain):
    balances = _test_account_rpc(account.get_balance_on_all_shards, local_test_address)
    assert isinstance(balances, list)
    assert len(balances) == 2


def test_get_total_balance(setup_blockchain):
    total_balance = _test_account_rpc(account.get_total_balance, local_test_address)
    assert isinstance(total_balance, int)
    assert total_balance > 0


def test_is_valid_address():
    assert account.is_valid_address("one1zksj3evekayy90xt4psrz8h6j2v3hla4qwz4ur")
    assert not account.is_valid_address("one1wje75aedczmj4dwjs0812xcg7vx0dy231cajk0")


def test_get_transaction_count(setup_blockchain):
    tx_count = _test_account_rpc(
        account.get_transaction_count, local_test_address, "latest", explorer_endpoint
    )
    assert isinstance(tx_count, int)
    assert tx_count > 0


def test_get_transactions_count(setup_blockchain):
    tx_count = _test_account_rpc(
        account.get_transactions_count, local_test_address, "ALL", explorer_endpoint
    )


def test_get_staking_transactions_count(setup_blockchain):
    tx_count = _test_account_rpc(
        account.get_staking_transactions_count,
        local_test_address,
        "ALL",
        explorer_endpoint,
    )
    assert isinstance(tx_count, int)


def test_errors():
    with pytest.raises(exceptions.RPCError):
        account.get_balance("", fake_shard)
    with pytest.raises(exceptions.RPCError):
        account.get_balance_by_block("", 1, fake_shard)
    with pytest.raises(exceptions.RPCError):
        account.get_account_nonce("", 1, fake_shard)
    with pytest.raises(exceptions.RPCError):
        account.get_transaction_count("", 1, fake_shard)
    with pytest.raises(exceptions.RPCError):
        account.get_transactions_count("", 1, fake_shard)
    with pytest.raises(exceptions.RPCError):
        account.get_transactions_count("", "ALL", fake_shard)
    with pytest.raises(exceptions.RPCError):
        account.get_transaction_history("", endpoint=fake_shard)
    with pytest.raises(exceptions.RPCError):
        account.get_staking_transaction_history("", endpoint=fake_shard)
    with pytest.raises(exceptions.RPCError):
        account.get_balance_on_all_shards("", endpoint=fake_shard)
    with pytest.raises(exceptions.RPCError):
        account.get_total_balance("", endpoint=fake_shard)