111 lines
5.3 KiB
Python
111 lines
5.3 KiB
Python
|
# standard imports
|
||
|
|
from decimal import Decimal
|
||
|
|
|
||
|
|
# external imports
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
# local imports
|
||
|
|
from cic_ussd.account.chain import Chain
|
||
|
|
from cic_ussd.account.transaction import (aux_transaction_data,
|
||
|
|
from_wei,
|
||
|
|
to_wei,
|
||
|
|
truncate,
|
||
|
|
transaction_actors,
|
||
|
|
validate_transaction_account,
|
||
|
|
OutgoingTransaction)
|
||
|
|
from cic_ussd.db.models.account import Account
|
||
|
|
from cic_ussd.error import UnknownUssdRecipient
|
||
|
|
from cic_ussd.translation import translation_for
|
||
|
|
|
||
|
|
|
||
|
|
# test imports
|
||
|
|
|
||
|
|
|
||
|
|
def check_aux_data(action_tag_key, direction_tag_key, preferred_language, transaction_with_aux_data):
|
||
|
|
assert transaction_with_aux_data.get('action_tag') == translation_for(action_tag_key, preferred_language)
|
||
|
|
assert transaction_with_aux_data.get('direction_tag') == translation_for(direction_tag_key, preferred_language)
|
||
|
|
|
||
|
|
|
||
|
|
def test_aux_transaction_data(preferences, set_locale_files, transactions_list):
|
||
|
|
sample_transaction = transactions_list[0]
|
||
|
|
preferred_language = preferences.get('preferred_language')
|
||
|
|
recipient_transaction, sender_transaction = transaction_actors(sample_transaction)
|
||
|
|
recipient_tx_aux_data = aux_transaction_data(preferred_language, recipient_transaction)
|
||
|
|
check_aux_data('helpers.received', 'helpers.from', preferred_language, recipient_tx_aux_data)
|
||
|
|
sender_tx_aux_data = aux_transaction_data(preferred_language, sender_transaction)
|
||
|
|
check_aux_data('helpers.sent', 'helpers.to', preferred_language, sender_tx_aux_data)
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize("wei, expected_result", [
|
||
|
|
(50000000, Decimal('50.00')),
|
||
|
|
(100000, Decimal('0.10'))
|
||
|
|
])
|
||
|
|
def test_from_wei(wei, expected_result):
|
||
|
|
assert from_wei(wei) == expected_result
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize("value, expected_result", [
|
||
|
|
(50, 50000000),
|
||
|
|
(0.10, 100000)
|
||
|
|
])
|
||
|
|
def test_to_wei(value, expected_result):
|
||
|
|
assert to_wei(value) == expected_result
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize("decimals, value, expected_result", [
|
||
|
|
(3, 1234.32875, 1234.328),
|
||
|
|
(2, 98.998, 98.99)
|
||
|
|
])
|
||
|
|
def test_truncate(decimals, value, expected_result):
|
||
|
|
assert truncate(value=value, decimals=decimals).__float__() == expected_result
|
||
|
|
|
||
|
|
|
||
|
|
def test_transaction_actors(activated_account, transaction_result, valid_recipient):
|
||
|
|
recipient_transaction, sender_transaction = transaction_actors(transaction_result)
|
||
|
|
assert recipient_transaction.get('blockchain_address') == valid_recipient.blockchain_address
|
||
|
|
assert sender_transaction.get('blockchain_address') == activated_account.blockchain_address
|
||
|
|
assert recipient_transaction.get('role') == 'recipient'
|
||
|
|
assert sender_transaction.get('role') == 'sender'
|
||
|
|
assert recipient_transaction.get('token_symbol') == transaction_result.get('destination_token_symbol')
|
||
|
|
assert sender_transaction.get('token_symbol') == transaction_result.get('source_token_symbol')
|
||
|
|
assert recipient_transaction.get('token_value') == transaction_result.get('destination_token_value')
|
||
|
|
assert sender_transaction.get('token_value') == transaction_result.get('source_token_value')
|
||
|
|
|
||
|
|
|
||
|
|
def test_validate_transaction_account(activated_account, init_database, transactions_list):
|
||
|
|
sample_transaction = transactions_list[0]
|
||
|
|
recipient_transaction, sender_transaction = transaction_actors(sample_transaction)
|
||
|
|
recipient_account = validate_transaction_account(init_database, recipient_transaction)
|
||
|
|
sender_account = validate_transaction_account(init_database, sender_transaction)
|
||
|
|
assert isinstance(recipient_account, Account)
|
||
|
|
assert isinstance(sender_account, Account)
|
||
|
|
sample_transaction = transactions_list[1]
|
||
|
|
recipient_transaction, sender_transaction = transaction_actors(sample_transaction)
|
||
|
|
with pytest.raises(UnknownUssdRecipient) as error:
|
||
|
|
validate_transaction_account(init_database, recipient_transaction)
|
||
|
|
assert str(
|
||
|
|
error.value) == f'Tx for recipient: {recipient_transaction.get("blockchain_address")} has no matching account in the system.'
|
||
|
|
validate_transaction_account(init_database, sender_transaction)
|
||
|
|
assert f'Tx from sender: {sender_transaction.get("blockchain_address")} has no matching account in system.'
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize("amount", [50, 0.10])
|
||
|
|
def test_outgoing_transaction_processor(activated_account,
|
||
|
|
amount,
|
||
|
|
celery_session_worker,
|
||
|
|
load_config,
|
||
|
|
load_chain_spec,
|
||
|
|
mock_transfer_api,
|
||
|
|
valid_recipient):
|
||
|
|
chain_str = Chain.spec.__str__()
|
||
|
|
token_symbol = load_config.get('TEST_TOKEN_SYMBOL')
|
||
|
|
outgoing_tx_processor = OutgoingTransaction(chain_str,
|
||
|
|
activated_account.blockchain_address,
|
||
|
|
valid_recipient.blockchain_address)
|
||
|
|
|
||
|
|
outgoing_tx_processor.transfer(amount, token_symbol)
|
||
|
|
assert mock_transfer_api.get('from_address') == activated_account.blockchain_address
|
||
|
|
assert mock_transfer_api.get('to_address') == valid_recipient.blockchain_address
|
||
|
|
assert mock_transfer_api.get('value') == to_wei(amount)
|
||
|
|
assert mock_transfer_api.get('token_symbol') == token_symbol
|