72 lines
3.0 KiB
Python
72 lines
3.0 KiB
Python
# standard imports
|
|
import datetime
|
|
|
|
# external imports
|
|
import celery
|
|
|
|
# local imports
|
|
from cic_ussd.account.transaction import from_wei
|
|
from cic_ussd.phone_number import Support
|
|
from cic_ussd.translation import translation_for
|
|
|
|
|
|
# tests imports
|
|
|
|
|
|
def test_transaction(celery_session_worker,
|
|
load_support_phone,
|
|
mock_notifier_api,
|
|
notification_data,
|
|
set_locale_files):
|
|
notification_data['transaction_type'] = 'transfer'
|
|
amount = from_wei(notification_data.get('token_value'))
|
|
balance = notification_data.get('available_balance')
|
|
phone_number = notification_data.get('phone_number')
|
|
preferred_language = notification_data.get('preferred_language')
|
|
token_symbol = notification_data.get('token_symbol')
|
|
transaction_account_metadata = notification_data.get('metadata_id')
|
|
timestamp = datetime.datetime.now().strftime('%d-%m-%y, %H:%M %p')
|
|
s_transaction = celery.signature(
|
|
'cic_ussd.tasks.notifications.transaction', [notification_data]
|
|
)
|
|
s_transaction.apply_async().get()
|
|
assert mock_notifier_api.get('recipient') == phone_number
|
|
message = translation_for(key='sms.sent_tokens',
|
|
phone_number=phone_number,
|
|
preferred_language=preferred_language,
|
|
amount=amount,
|
|
token_symbol=token_symbol,
|
|
tx_recipient_information=transaction_account_metadata,
|
|
timestamp=timestamp,
|
|
balance=balance)
|
|
assert mock_notifier_api.get('message') == message
|
|
|
|
notification_data['role'] = 'recipient'
|
|
notification_data['direction_tag'] = 'From'
|
|
s_transaction = celery.signature(
|
|
'cic_ussd.tasks.notifications.transaction', [notification_data]
|
|
)
|
|
s_transaction.apply_async().get()
|
|
message = translation_for(key='sms.received_tokens',
|
|
phone_number=phone_number,
|
|
preferred_language=preferred_language,
|
|
amount=amount,
|
|
token_symbol=token_symbol,
|
|
tx_sender_information=transaction_account_metadata,
|
|
timestamp=timestamp,
|
|
balance=balance)
|
|
assert mock_notifier_api.get('message') == message
|
|
|
|
notification_data['transaction_type'] = 'tokengift'
|
|
s_transaction = celery.signature(
|
|
'cic_ussd.tasks.notifications.transaction', [notification_data]
|
|
)
|
|
s_transaction.apply_async().get()
|
|
support_phone = Support.phone_number
|
|
message = translation_for(key='sms.account_successfully_created',
|
|
preferred_language=preferred_language,
|
|
balance=balance,
|
|
support_phone=support_phone,
|
|
token_symbol=token_symbol)
|
|
assert mock_notifier_api.get('message') == message
|