cic-internal-integration/apps/cic-ussd/tests/cic_ussd/account/test_balance.py

# standard imports

# external imports
import pytest
from cic_types.condiments import MetadataPointer

# local imports
from cic_ussd.account.balance import (calculate_available_balance,
                                      get_balances,
                                      get_cached_adjusted_balance,
                                      get_cached_available_balance)
from cic_ussd.account.chain import Chain
from cic_ussd.account.tokens import get_cached_token_data_list
from cic_ussd.cache import cache_data_key, get_cached_data
from cic_ussd.error import CachedDataNotFoundError

# test imports
from tests.helpers.accounts import blockchain_address


def test_async_get_balances(activated_account,
                            celery_session_worker,
                            load_chain_spec,
                            load_config,
                            mock_async_balance_api_query):
    blockchain_address = activated_account.blockchain_address
    chain_str = Chain.spec.__str__()
    token_symbol = load_config.get('TEST_TOKEN_SYMBOL')
    get_balances(blockchain_address, chain_str, token_symbol, asynchronous=True)
    assert mock_async_balance_api_query.get('address') == blockchain_address
    assert mock_async_balance_api_query.get('token_symbol') == token_symbol


def test_sync_get_balances(activated_account,
                           balances,
                           celery_session_worker,
                           load_chain_spec,
                           load_config,
                           mock_sync_balance_api_query):
    blockchain_address = activated_account.blockchain_address
    chain_str = Chain.spec.__str__()
    token_symbol = load_config.get('TEST_TOKEN_SYMBOL')
    res = get_balances(blockchain_address, chain_str, token_symbol, asynchronous=False)
    assert res == balances


@pytest.mark.parametrize('balance_incoming, balance_network, balance_outgoing, available_balance', [
    (0, 50000000, 0, 50.00),
    (5000000, 89000000, 67000000, 27.00)
])
def test_calculate_available_balance(activated_account,
                                     available_balance,
                                     balance_incoming,
                                     balance_network,
                                     balance_outgoing,
                                     cache_balances,
                                     cache_default_token_data,
                                     load_chain_spec):
    balances = {
        'address': activated_account.blockchain_address,
        'converters': [],
        'balance_network': balance_network,
        'balance_outgoing': balance_outgoing,
        'balance_incoming': balance_incoming
    }
    assert calculate_available_balance(balances, 6) == available_balance


def test_get_cached_available_balance(activated_account,
                                      balances,
                                      cache_balances,
                                      cache_default_token_data,
                                      load_chain_spec,
                                      token_symbol):
    identifier = [bytes.fromhex(activated_account.blockchain_address), token_symbol.encode('utf-8')]
    cached_available_balance = get_cached_available_balance(6, identifier)
    available_balance = calculate_available_balance(balances[0], 6)
    assert cached_available_balance == available_balance
    address = blockchain_address()
    with pytest.raises(CachedDataNotFoundError) as error:
        identifier = [bytes.fromhex(address), token_symbol.encode('utf-8')]
        key = cache_data_key(identifier=identifier, salt=MetadataPointer.BALANCES)
        cached_available_balance = get_cached_available_balance(6, identifier)
        assert cached_available_balance is None
    assert str(error.value) == f'No cached available balance at {key}'


def test_get_cached_adjusted_balance(activated_account, cache_adjusted_balances, token_symbol):
    identifier = bytes.fromhex(activated_account.blockchain_address)
    balances_identifier = [identifier, token_symbol.encode('utf-8')]
    key = cache_data_key(balances_identifier, MetadataPointer.BALANCES_ADJUSTED)
    adjusted_balances = get_cached_data(key)
    assert get_cached_adjusted_balance(balances_identifier) == adjusted_balances


def test_get_account_tokens_balance(activated_account,
                                    cache_token_data_list,
                                    celery_session_worker,
                                    load_chain_spec,
                                    load_config,
                                    mock_async_balance_api_query,
                                    token_symbol):
    blockchain_address = activated_account.blockchain_address
    chain_str = Chain.spec.__str__()
    get_balances(blockchain_address, chain_str, token_symbol, asynchronous=True)
    assert mock_async_balance_api_query.get('address') == blockchain_address
    assert mock_async_balance_api_query.get('token_symbol') == token_symbol