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 )