63 lines
2.7 KiB
Python
63 lines
2.7 KiB
Python
# standard imports
|
|
import datetime
|
|
import json
|
|
|
|
# external imports
|
|
import pytest
|
|
from chainlib.hash import strip_0x
|
|
from cic_types.models.person import get_contact_data_from_vcard
|
|
|
|
# local imports
|
|
from cic_ussd.account.metadata import get_cached_preferred_language
|
|
from cic_ussd.metadata import PersonMetadata
|
|
from cic_ussd.processor.util import latest_input, parse_person_metadata, resume_last_ussd_session
|
|
from cic_ussd.translation import translation_for
|
|
|
|
|
|
# test imports
|
|
|
|
|
|
@pytest.mark.parametrize('user_input, expected_value', [
|
|
('1*9*6*7', '7'),
|
|
('1', '1'),
|
|
('', '')
|
|
])
|
|
def test_latest_input(user_input, expected_value):
|
|
assert latest_input(user_input) == expected_value
|
|
|
|
|
|
def test_parse_person_metadata(activated_account, cache_person_metadata, cache_preferences):
|
|
identifier = bytes.fromhex(strip_0x(activated_account.blockchain_address))
|
|
person_metadata = PersonMetadata(identifier)
|
|
cached_person_metadata = person_metadata.get_cached_metadata()
|
|
person_metadata = json.loads(cached_person_metadata)
|
|
preferred_language = get_cached_preferred_language(activated_account.blockchain_address)
|
|
display_key = 'ussd.kenya.display_person_metadata'
|
|
parsed_person_metadata = parse_person_metadata(cached_person_metadata,
|
|
display_key,
|
|
preferred_language)
|
|
contact_data = get_contact_data_from_vcard(person_metadata.get('vcard'))
|
|
full_name = f'{contact_data.get("given")} {contact_data.get("family")}'
|
|
date_of_birth = person_metadata.get('date_of_birth')
|
|
year_of_birth = date_of_birth.get('year')
|
|
present_year = datetime.datetime.now().year
|
|
age = present_year - year_of_birth
|
|
gender = person_metadata.get('gender')
|
|
products = ', '.join(person_metadata.get('products'))
|
|
location = person_metadata.get('location').get('area_name')
|
|
assert parsed_person_metadata == translation_for(key=display_key,
|
|
preferred_language=preferred_language,
|
|
full_name=full_name,
|
|
age=age,
|
|
gender=gender,
|
|
location=location,
|
|
products=products)
|
|
|
|
|
|
@pytest.mark.parametrize('last_state, expected_menu_name', [
|
|
('account_creation_prompt', 'start'),
|
|
('enter_transaction_recipient', 'enter_transaction_recipient')
|
|
])
|
|
def test_resume_last_ussd_session(expected_menu_name, last_state, load_ussd_menu):
|
|
assert resume_last_ussd_session(last_state).get('name') == expected_menu_name
|