Minor refactors:
- Renames s_assemble to s_brief - Link s_local to s_brief
This commit is contained in:
@@ -0,0 +1,28 @@
|
||||
import pytest
|
||||
|
||||
|
||||
def test_single_head_revision(alembic_runner):
|
||||
heads = alembic_runner.heads
|
||||
head_count = len(heads)
|
||||
assert head_count == 1
|
||||
|
||||
|
||||
def test_upgrade(alembic_runner):
|
||||
try:
|
||||
alembic_runner.migrate_up_to("head")
|
||||
except RuntimeError:
|
||||
pytest.fail('Failed to upgrade to the head revision.')
|
||||
|
||||
|
||||
def test_up_down_consistency(alembic_runner):
|
||||
try:
|
||||
for revision in alembic_runner.history.revisions:
|
||||
alembic_runner.migrate_up_to(revision)
|
||||
except RuntimeError:
|
||||
pytest.fail('Failed to upgrade through each revision individually.')
|
||||
|
||||
try:
|
||||
for revision in reversed(alembic_runner.history.revisions):
|
||||
alembic_runner.migrate_down_to(revision)
|
||||
except RuntimeError:
|
||||
pytest.fail('Failed to downgrade through each revision individually.')
|
||||
14
apps/cic-ussd/tests/cic_ussd/db/models/test_task_tracker.py
Normal file
14
apps/cic-ussd/tests/cic_ussd/db/models/test_task_tracker.py
Normal file
@@ -0,0 +1,14 @@
|
||||
# local imports
|
||||
from cic_ussd.db.models.task_tracker import TaskTracker
|
||||
|
||||
|
||||
def test_task_tracker(init_database):
|
||||
task_uuid = '31e85315-feee-4b6d-995e-223569082cc4'
|
||||
task_in_tracker = TaskTracker(task_uuid=task_uuid)
|
||||
|
||||
session = init_database
|
||||
session.add(task_in_tracker)
|
||||
session.commit()
|
||||
|
||||
queried_task = session.query(TaskTracker).get(1)
|
||||
assert queried_task.task_uuid == task_uuid
|
||||
40
apps/cic-ussd/tests/cic_ussd/db/models/test_user.py
Normal file
40
apps/cic-ussd/tests/cic_ussd/db/models/test_user.py
Normal file
@@ -0,0 +1,40 @@
|
||||
"""Tests the persistence of the user record and associated functions to the user object"""
|
||||
|
||||
# standard imports
|
||||
import pytest
|
||||
|
||||
# platform imports
|
||||
from cic_ussd.db.models.user import User
|
||||
|
||||
|
||||
def test_user(init_database, set_fernet_key):
|
||||
user = User(blockchain_address='0x417f5962fc52dc33ff0689659b25848680dec6dcedc6785b03d1df60fc6d5c51',
|
||||
phone_number='+254700000000')
|
||||
user.create_password('0000')
|
||||
|
||||
session = User.session
|
||||
session.add(user)
|
||||
session.commit()
|
||||
|
||||
queried_user = session.query(User).get(1)
|
||||
assert queried_user.blockchain_address == '0x417f5962fc52dc33ff0689659b25848680dec6dcedc6785b03d1df60fc6d5c51'
|
||||
assert queried_user.phone_number == '+254700000000'
|
||||
assert queried_user.failed_pin_attempts == 0
|
||||
assert queried_user.verify_password('0000') is True
|
||||
|
||||
|
||||
def test_user_state_transition(create_pending_user):
|
||||
user = create_pending_user
|
||||
session = User.session
|
||||
|
||||
assert user.get_account_status() == 'PENDING'
|
||||
user.activate_account()
|
||||
assert user.get_account_status() == 'ACTIVE'
|
||||
user.failed_pin_attempts = 3
|
||||
assert user.get_account_status() == 'LOCKED'
|
||||
user.reset_account_pin()
|
||||
assert user.get_account_status() == 'RESET'
|
||||
user.activate_account()
|
||||
assert user.get_account_status() == 'ACTIVE'
|
||||
session.add(user)
|
||||
session.commit()
|
||||
54
apps/cic-ussd/tests/cic_ussd/db/models/test_ussd_session.py
Normal file
54
apps/cic-ussd/tests/cic_ussd/db/models/test_ussd_session.py
Normal file
@@ -0,0 +1,54 @@
|
||||
# third party imports
|
||||
import pytest
|
||||
|
||||
# local imports
|
||||
from cic_ussd.db.models.ussd_session import UssdSession
|
||||
from cic_ussd.error import VersionTooLowError
|
||||
|
||||
|
||||
def test_ussd_session(init_database, create_in_redis_ussd_session, create_activated_user):
|
||||
session = init_database
|
||||
|
||||
ussd_session = UssdSession(
|
||||
external_session_id='AT65423',
|
||||
service_code='*123#',
|
||||
msisdn=create_activated_user.phone_number,
|
||||
user_input='1',
|
||||
state='start',
|
||||
session_data={},
|
||||
version=1,
|
||||
)
|
||||
|
||||
session.add(ussd_session)
|
||||
session.commit()
|
||||
|
||||
ussd_session.set_data(key='foo', session=init_database, value='bar')
|
||||
|
||||
assert ussd_session.get_data('foo') == 'bar'
|
||||
ussd_session.update(
|
||||
session=init_database,
|
||||
user_input='3',
|
||||
state='next',
|
||||
version=2
|
||||
)
|
||||
assert ussd_session.version == 2
|
||||
session.add(ussd_session)
|
||||
session.commit()
|
||||
|
||||
assert UssdSession.have_session_for_phone(create_activated_user.phone_number) is True
|
||||
|
||||
|
||||
def test_version_too_low_error(init_database, create_in_redis_ussd_session, create_activated_user):
|
||||
with pytest.raises(VersionTooLowError) as e:
|
||||
session = UssdSession(
|
||||
external_session_id='AT38745',
|
||||
service_code='*123#',
|
||||
msisdn=create_activated_user.phone_number,
|
||||
user_input='1',
|
||||
state='start',
|
||||
session_data={},
|
||||
version=3,
|
||||
)
|
||||
assert session.check_version(1)
|
||||
assert session.check_version(3)
|
||||
assert str(e.value) == 'New session version number is not greater than last saved version!'
|
||||
30
apps/cic-ussd/tests/cic_ussd/db/test_db.py
Normal file
30
apps/cic-ussd/tests/cic_ussd/db/test_db.py
Normal file
@@ -0,0 +1,30 @@
|
||||
# standard imports
|
||||
import os
|
||||
|
||||
# third-party imports
|
||||
|
||||
# local imports
|
||||
from cic_ussd.db import dsn_from_config
|
||||
|
||||
|
||||
def test_dsn_from_config(load_config):
|
||||
"""
|
||||
"""
|
||||
# test dsn for sqlite engine
|
||||
dsn = dsn_from_config(load_config)
|
||||
scheme = f'{load_config.get("DATABASE_ENGINE")}+{load_config.get("DATABASE_DRIVER")}'
|
||||
assert dsn == f'{scheme}:///{load_config.get("DATABASE_NAME")}'
|
||||
|
||||
# test dsn for other db formats
|
||||
overrides = {
|
||||
'DATABASE_PASSWORD': 'password',
|
||||
'DATABASE_DRIVER': 'psycopg2',
|
||||
'DATABASE_ENGINE': 'postgresql'
|
||||
}
|
||||
load_config.dict_override(dct=overrides, dct_description='Override values to test different db formats.')
|
||||
|
||||
scheme = f'{load_config.get("DATABASE_ENGINE")}+{load_config.get("DATABASE_DRIVER")}'
|
||||
|
||||
dsn = dsn_from_config(load_config)
|
||||
assert dsn == f"{scheme}://{load_config.get('DATABASE_USER')}:{load_config.get('DATABASE_PASSWORD')}@{load_config.get('DATABASE_HOST')}:{load_config.get('DATABASE_PORT')}/{load_config.get('DATABASE_NAME')}"
|
||||
|
||||
55
apps/cic-ussd/tests/cic_ussd/files/test_local_files.py
Normal file
55
apps/cic-ussd/tests/cic_ussd/files/test_local_files.py
Normal file
@@ -0,0 +1,55 @@
|
||||
# standard imports
|
||||
import json
|
||||
import os
|
||||
|
||||
# third-party imports
|
||||
from tinydb import Query
|
||||
|
||||
# local imports
|
||||
from cic_ussd.files.local_files import create_local_file_data_stores, json_file_parser
|
||||
from tests.fixtures.config import root_directory
|
||||
from tests.helpers.tmp_files import create_tmp_file
|
||||
|
||||
|
||||
def test_create_in_memory_data_stores():
|
||||
"""
|
||||
GIVEN the cic-ussd application component
|
||||
WHEN the create_in_memory_data_stores function is passed a file and table name
|
||||
THEN it creates a tiny dn data store that can be written to and queried
|
||||
"""
|
||||
descriptor, tmp_file = create_tmp_file()
|
||||
test_file = create_local_file_data_stores(file_location=tmp_file, table_name='test_table')
|
||||
# write to data store
|
||||
test_file.insert({
|
||||
'foo': 'bar'
|
||||
})
|
||||
query = Query()
|
||||
inserted_record = test_file.get(query.foo == 'bar')
|
||||
assert inserted_record == {'foo': 'bar'}
|
||||
os.close(descriptor)
|
||||
os.remove(tmp_file)
|
||||
|
||||
|
||||
def test_json_file_parser(load_config):
|
||||
"""
|
||||
GIVEN the cic-ussd application component
|
||||
WHEN the json_file_parser function is passed a directory path containing JSON files
|
||||
THEN it dynamically loads all the files and compiles their content into one python array
|
||||
"""
|
||||
files_dir = load_config.get('STATEMACHINE_TRANSITIONS')
|
||||
files_dir = os.path.join(root_directory, files_dir)
|
||||
|
||||
# total files len
|
||||
file_content_length = 0
|
||||
|
||||
for filepath in os.listdir(files_dir):
|
||||
# get path of data files
|
||||
filepath = os.path.join(files_dir, filepath)
|
||||
|
||||
# read file content
|
||||
data_file = open(filepath)
|
||||
data = json.load(data_file)
|
||||
file_content_length += len(data)
|
||||
data_file.close()
|
||||
assert len(json_file_parser(filepath=files_dir)) == file_content_length
|
||||
|
||||
52
apps/cic-ussd/tests/cic_ussd/menu/test_ussd_menu.py
Normal file
52
apps/cic-ussd/tests/cic_ussd/menu/test_ussd_menu.py
Normal file
@@ -0,0 +1,52 @@
|
||||
# standard imports
|
||||
import os
|
||||
|
||||
# third party imports
|
||||
import pytest
|
||||
|
||||
# local imports
|
||||
from cic_ussd.files.local_files import create_local_file_data_stores
|
||||
from cic_ussd.menu.ussd_menu import UssdMenu
|
||||
from tests.helpers.tmp_files import create_tmp_file
|
||||
|
||||
|
||||
@pytest.mark.parametrize('menu_name, expected_parent_menu_name', [
|
||||
('initial_language_selection', None),
|
||||
('account_management', 'start'),
|
||||
('enter_current_pin', 'account_management')
|
||||
])
|
||||
def test_ussd_menu(load_ussd_menu, menu_name, expected_parent_menu_name):
|
||||
ussd_menu = UssdMenu.find_by_name(name=menu_name)
|
||||
assert ussd_menu.get('parent') == expected_parent_menu_name
|
||||
|
||||
|
||||
def test_create_ussd_menu():
|
||||
descriptor, tmp_file = create_tmp_file()
|
||||
ussd_menu_db = create_local_file_data_stores(file_location=tmp_file, table_name="ussd_menu")
|
||||
UssdMenu.ussd_menu_db = ussd_menu_db
|
||||
UssdMenu(name='foo', description='foo-bar', parent=None)
|
||||
assert UssdMenu.find_by_name(name='foo')['description'] == 'foo-bar'
|
||||
UssdMenu.set_description(name='foo', description='bar')
|
||||
assert UssdMenu.find_by_name(name='foo')['description'] == 'bar'
|
||||
menu2 = UssdMenu(name='fizz', description='buzz', parent='foo')
|
||||
assert UssdMenu.parent_menu(menu2.name)['description'] == 'bar'
|
||||
os.close(descriptor)
|
||||
os.remove(tmp_file)
|
||||
|
||||
|
||||
def test_failed_create_ussd_menu():
|
||||
descriptor, tmp_file = create_tmp_file()
|
||||
ussd_menu_db = create_local_file_data_stores(file_location=tmp_file, table_name="ussd_menu")
|
||||
UssdMenu.ussd_menu_db = ussd_menu_db
|
||||
UssdMenu(name='foo', description='foo-bar', parent=None)
|
||||
assert UssdMenu.find_by_name(name='foo')['description'] == 'foo-bar'
|
||||
UssdMenu.set_description(name='foo', description='bar')
|
||||
assert UssdMenu.find_by_name(name='foo')['description'] == 'bar'
|
||||
|
||||
with pytest.raises(ValueError) as error:
|
||||
UssdMenu(name='foo', description='foo-bar', parent=None)
|
||||
|
||||
assert str(error.value) == "Menu already exists!"
|
||||
os.close(descriptor)
|
||||
os.remove(tmp_file)
|
||||
|
||||
80
apps/cic-ussd/tests/cic_ussd/metadata/test_metadata.py
Normal file
80
apps/cic-ussd/tests/cic_ussd/metadata/test_metadata.py
Normal file
@@ -0,0 +1,80 @@
|
||||
# standard imports
|
||||
import json
|
||||
|
||||
# third-party imports
|
||||
import pytest
|
||||
import requests
|
||||
import requests_mock
|
||||
|
||||
# local imports
|
||||
from cic_ussd.error import UnsupportedMethodError
|
||||
from cic_ussd.metadata import blockchain_address_to_metadata_pointer, make_request
|
||||
|
||||
|
||||
def test_make_request(define_metadata_pointer_url, mock_meta_get_response, mock_meta_post_response, person_metadata):
|
||||
with requests_mock.Mocker(real_http=False) as request_mocker:
|
||||
request_mocker.register_uri(
|
||||
'GET',
|
||||
define_metadata_pointer_url,
|
||||
status_code=200,
|
||||
reason='OK',
|
||||
content=json.dumps(mock_meta_get_response).encode('utf-8')
|
||||
)
|
||||
response = make_request(method='GET', url=define_metadata_pointer_url)
|
||||
assert response.content == requests.get(define_metadata_pointer_url).content
|
||||
|
||||
with requests_mock.Mocker(real_http=False) as request_mocker:
|
||||
request_mocker.register_uri(
|
||||
'POST',
|
||||
define_metadata_pointer_url,
|
||||
status_code=201,
|
||||
reason='CREATED',
|
||||
content=json.dumps(mock_meta_post_response).encode('utf-8')
|
||||
)
|
||||
response = make_request(
|
||||
method='POST',
|
||||
url=define_metadata_pointer_url,
|
||||
data=json.dumps(person_metadata).encode('utf-8'),
|
||||
headers={
|
||||
'X-CIC-AUTOMERGE': 'server',
|
||||
'Content-Type': 'application/json'
|
||||
}
|
||||
)
|
||||
assert response.content == requests.post(define_metadata_pointer_url).content
|
||||
|
||||
with requests_mock.Mocker(real_http=False) as request_mocker:
|
||||
request_mocker.register_uri(
|
||||
'PUT',
|
||||
define_metadata_pointer_url,
|
||||
status_code=200,
|
||||
reason='OK'
|
||||
)
|
||||
response = make_request(
|
||||
method='PUT',
|
||||
url=define_metadata_pointer_url,
|
||||
data=json.dumps(person_metadata).encode('utf-8'),
|
||||
headers={
|
||||
'X-CIC-AUTOMERGE': 'server',
|
||||
'Content-Type': 'application/json'
|
||||
}
|
||||
)
|
||||
assert response.content == requests.put(define_metadata_pointer_url).content
|
||||
|
||||
with pytest.raises(UnsupportedMethodError) as error:
|
||||
with requests_mock.Mocker(real_http=False) as request_mocker:
|
||||
request_mocker.register_uri(
|
||||
'DELETE',
|
||||
define_metadata_pointer_url,
|
||||
status_code=200,
|
||||
reason='OK'
|
||||
)
|
||||
make_request(
|
||||
method='DELETE',
|
||||
url=define_metadata_pointer_url
|
||||
)
|
||||
assert str(error.value) == 'Unsupported method: DELETE'
|
||||
|
||||
|
||||
def test_blockchain_address_to_metadata_pointer(create_activated_user):
|
||||
blockchain_address = create_activated_user.blockchain_address
|
||||
assert type(blockchain_address_to_metadata_pointer(blockchain_address)) == bytes
|
||||
34
apps/cic-ussd/tests/cic_ussd/metadata/test_signer.py
Normal file
34
apps/cic-ussd/tests/cic_ussd/metadata/test_signer.py
Normal file
@@ -0,0 +1,34 @@
|
||||
# standard imports
|
||||
import shutil
|
||||
|
||||
# third-party imports
|
||||
|
||||
# local imports
|
||||
from cic_ussd.metadata.signer import Signer
|
||||
|
||||
|
||||
def test_client(load_config, setup_metadata_signer, person_metadata):
|
||||
signer = Signer()
|
||||
# get gpg used
|
||||
digest = 'a4337bc45a8fc544c03f52dc550cd6e1e87021bc896588bd79e901e2'
|
||||
person_metadata['digest'] = digest
|
||||
gpg = signer.gpg
|
||||
|
||||
# check that key data was loaded
|
||||
assert signer.key_data is not None
|
||||
|
||||
# check that correct operational key is returned
|
||||
gpg.import_keys(key_data=signer.key_data)
|
||||
gpg_keys = gpg.list_keys()
|
||||
assert signer.get_operational_key() == gpg_keys[0]
|
||||
|
||||
# check that correct signature is returned
|
||||
key_id = signer.get_operational_key().get('keyid')
|
||||
signature = gpg.sign(message=digest, passphrase=load_config.get('KEYS_PASSPHRASE'), keyid=key_id)
|
||||
assert str(signature) == signer.sign_digest(data=person_metadata)
|
||||
|
||||
# remove tmp gpg file
|
||||
shutil.rmtree(Signer.gpg_path)
|
||||
|
||||
|
||||
|
||||
123
apps/cic-ussd/tests/cic_ussd/metadata/test_user_metadata.py
Normal file
123
apps/cic-ussd/tests/cic_ussd/metadata/test_user_metadata.py
Normal file
@@ -0,0 +1,123 @@
|
||||
# standard imports
|
||||
import json
|
||||
|
||||
# third-party imports
|
||||
import pytest
|
||||
import requests_mock
|
||||
from cic_types.models.person import generate_metadata_pointer
|
||||
|
||||
# local imports
|
||||
from cic_ussd.metadata import blockchain_address_to_metadata_pointer
|
||||
from cic_ussd.metadata.signer import Signer
|
||||
from cic_ussd.metadata.user import UserMetadata
|
||||
from cic_ussd.redis import get_cached_data
|
||||
|
||||
|
||||
def test_user_metadata(create_activated_user, define_metadata_pointer_url, load_config):
|
||||
UserMetadata.base_url = load_config.get('CIC_META_URL')
|
||||
identifier = blockchain_address_to_metadata_pointer(blockchain_address=create_activated_user.blockchain_address)
|
||||
user_metadata_client = UserMetadata(identifier=identifier)
|
||||
|
||||
assert user_metadata_client.url == define_metadata_pointer_url
|
||||
|
||||
|
||||
def test_create_user_metadata(caplog,
|
||||
create_activated_user,
|
||||
define_metadata_pointer_url,
|
||||
load_config,
|
||||
mock_meta_post_response,
|
||||
person_metadata):
|
||||
identifier = blockchain_address_to_metadata_pointer(blockchain_address=create_activated_user.blockchain_address)
|
||||
user_metadata_client = UserMetadata(identifier=identifier)
|
||||
|
||||
with requests_mock.Mocker(real_http=False) as request_mocker:
|
||||
request_mocker.register_uri(
|
||||
'POST',
|
||||
define_metadata_pointer_url,
|
||||
status_code=201,
|
||||
reason='CREATED',
|
||||
content=json.dumps(mock_meta_post_response).encode('utf-8')
|
||||
)
|
||||
user_metadata_client.create(data=person_metadata)
|
||||
assert 'Get signed material response status: 201' in caplog.text
|
||||
|
||||
with pytest.raises(RuntimeError) as error:
|
||||
with requests_mock.Mocker(real_http=False) as request_mocker:
|
||||
request_mocker.register_uri(
|
||||
'POST',
|
||||
define_metadata_pointer_url,
|
||||
status_code=400,
|
||||
reason='BAD REQUEST'
|
||||
)
|
||||
user_metadata_client.create(data=person_metadata)
|
||||
assert str(error.value) == f'400 Client Error: BAD REQUEST for url: {define_metadata_pointer_url}'
|
||||
|
||||
|
||||
def test_edit_user_metadata(caplog,
|
||||
create_activated_user,
|
||||
define_metadata_pointer_url,
|
||||
load_config,
|
||||
person_metadata,
|
||||
setup_metadata_signer):
|
||||
Signer.gpg_passphrase = load_config.get('KEYS_PASSPHRASE')
|
||||
identifier = blockchain_address_to_metadata_pointer(blockchain_address=create_activated_user.blockchain_address)
|
||||
user_metadata_client = UserMetadata(identifier=identifier)
|
||||
with requests_mock.Mocker(real_http=False) as request_mocker:
|
||||
request_mocker.register_uri(
|
||||
'PUT',
|
||||
define_metadata_pointer_url,
|
||||
status_code=200,
|
||||
reason='OK'
|
||||
)
|
||||
user_metadata_client.edit(data=person_metadata, engine='pgp')
|
||||
assert 'Signed content submission status: 200' in caplog.text
|
||||
|
||||
with pytest.raises(RuntimeError) as error:
|
||||
with requests_mock.Mocker(real_http=False) as request_mocker:
|
||||
request_mocker.register_uri(
|
||||
'PUT',
|
||||
define_metadata_pointer_url,
|
||||
status_code=400,
|
||||
reason='BAD REQUEST'
|
||||
)
|
||||
user_metadata_client.edit(data=person_metadata, engine='pgp')
|
||||
assert str(error.value) == f'400 Client Error: BAD REQUEST for url: {define_metadata_pointer_url}'
|
||||
|
||||
|
||||
def test_get_user_metadata(caplog,
|
||||
create_activated_user,
|
||||
define_metadata_pointer_url,
|
||||
init_redis_cache,
|
||||
load_config,
|
||||
person_metadata,
|
||||
setup_metadata_signer):
|
||||
identifier = blockchain_address_to_metadata_pointer(blockchain_address=create_activated_user.blockchain_address)
|
||||
user_metadata_client = UserMetadata(identifier=identifier)
|
||||
with requests_mock.Mocker(real_http=False) as request_mocker:
|
||||
request_mocker.register_uri(
|
||||
'GET',
|
||||
define_metadata_pointer_url,
|
||||
status_code=200,
|
||||
content=json.dumps(person_metadata).encode('utf-8'),
|
||||
reason='OK'
|
||||
)
|
||||
user_metadata_client.query()
|
||||
assert 'Get latest data status: 200' in caplog.text
|
||||
key = generate_metadata_pointer(
|
||||
identifier=identifier,
|
||||
cic_type='cic.person'
|
||||
)
|
||||
cached_user_metadata = get_cached_data(key=key)
|
||||
assert cached_user_metadata
|
||||
|
||||
with pytest.raises(RuntimeError) as error:
|
||||
with requests_mock.Mocker(real_http=False) as request_mocker:
|
||||
request_mocker.register_uri(
|
||||
'GET',
|
||||
define_metadata_pointer_url,
|
||||
status_code=404,
|
||||
reason='NOT FOUND'
|
||||
)
|
||||
user_metadata_client.query()
|
||||
assert 'The data is not available and might need to be added.' in caplog.text
|
||||
assert str(error.value) == f'400 Client Error: NOT FOUND for url: {define_metadata_pointer_url}'
|
||||
@@ -0,0 +1,11 @@
|
||||
# standard imports
|
||||
import json
|
||||
|
||||
# local imports
|
||||
from cic_ussd.menu.ussd_menu import UssdMenu
|
||||
from cic_ussd.session.ussd_session import UssdSession
|
||||
|
||||
|
||||
def test_ussd_session(load_ussd_menu, get_in_redis_ussd_session):
|
||||
ussd_session = get_in_redis_ussd_session
|
||||
assert UssdMenu.find_by_name(name='initial_language_selection').get('name') == ussd_session.state
|
||||
@@ -0,0 +1,21 @@
|
||||
# local imports
|
||||
from cic_ussd.state_machine.logic.menu import (menu_one_selected,
|
||||
menu_two_selected,
|
||||
menu_three_selected,
|
||||
menu_four_selected)
|
||||
|
||||
|
||||
def test_menu_selection(create_pending_user, create_in_db_ussd_session):
|
||||
serialized_in_db_ussd_session = create_in_db_ussd_session.to_json()
|
||||
assert menu_one_selected(('1', serialized_in_db_ussd_session, create_pending_user)) is True
|
||||
assert menu_one_selected(('x', serialized_in_db_ussd_session, create_pending_user)) is False
|
||||
|
||||
assert menu_two_selected(('2', serialized_in_db_ussd_session, create_pending_user)) is True
|
||||
assert menu_two_selected(('1', serialized_in_db_ussd_session, create_pending_user)) is False
|
||||
|
||||
assert menu_three_selected(('3', serialized_in_db_ussd_session, create_pending_user)) is True
|
||||
assert menu_three_selected(('4', serialized_in_db_ussd_session, create_pending_user)) is False
|
||||
|
||||
assert menu_four_selected(('4', serialized_in_db_ussd_session, create_pending_user)) is True
|
||||
assert menu_four_selected(('d', serialized_in_db_ussd_session, create_pending_user)) is False
|
||||
|
||||
94
apps/cic-ussd/tests/cic_ussd/state_machine/logic/test_pin.py
Normal file
94
apps/cic-ussd/tests/cic_ussd/state_machine/logic/test_pin.py
Normal file
@@ -0,0 +1,94 @@
|
||||
# standards imports
|
||||
import json
|
||||
|
||||
# third party imports
|
||||
import pytest
|
||||
|
||||
# local imports
|
||||
from cic_ussd.encoder import check_password_hash, create_password_hash
|
||||
from cic_ussd.state_machine.logic.pin import (complete_pin_change,
|
||||
is_valid_pin,
|
||||
is_valid_new_pin,
|
||||
is_authorized_pin,
|
||||
is_blocked_pin,
|
||||
pins_match,
|
||||
save_initial_pin_to_session_data)
|
||||
|
||||
|
||||
def test_complete_pin_change(init_database, create_pending_user, create_in_db_ussd_session):
|
||||
serialized_in_db_ussd_session = create_in_db_ussd_session.to_json()
|
||||
state_machine_data = ('1212', serialized_in_db_ussd_session, create_pending_user)
|
||||
assert create_pending_user.password_hash is None
|
||||
create_in_db_ussd_session.set_data(key='initial_pin', session=init_database, value=create_password_hash('1212'))
|
||||
complete_pin_change(state_machine_data)
|
||||
assert create_pending_user.password_hash is not None
|
||||
assert create_pending_user.verify_password(password='1212') is True
|
||||
|
||||
|
||||
@pytest.mark.parametrize('user_input, expected', [
|
||||
('4562', True),
|
||||
('jksu', False),
|
||||
('ij45', False),
|
||||
])
|
||||
def test_is_valid_pin(create_pending_user, create_in_db_ussd_session, user_input, expected):
|
||||
serialized_in_db_ussd_session = create_in_db_ussd_session.to_json()
|
||||
state_machine_data = (user_input, serialized_in_db_ussd_session, create_pending_user)
|
||||
assert is_valid_pin(state_machine_data) is expected
|
||||
|
||||
|
||||
@pytest.mark.parametrize('user_input, expected', [
|
||||
('1212', True),
|
||||
('0000', False)
|
||||
])
|
||||
def test_pins_match(init_database, create_pending_user, create_in_db_ussd_session, user_input, expected):
|
||||
serialized_in_db_ussd_session = create_in_db_ussd_session.to_json()
|
||||
state_machine_data = (user_input, serialized_in_db_ussd_session, create_pending_user)
|
||||
create_in_db_ussd_session.set_data(key='initial_pin', session=init_database, value=create_password_hash(user_input))
|
||||
assert pins_match(state_machine_data) is True
|
||||
|
||||
|
||||
def test_save_initial_pin_to_session_data(create_pending_user,
|
||||
create_in_redis_ussd_session,
|
||||
create_in_db_ussd_session,
|
||||
celery_session_worker):
|
||||
serialized_in_db_ussd_session = create_in_db_ussd_session.to_json()
|
||||
state_machine_data = ('1212', serialized_in_db_ussd_session, create_pending_user)
|
||||
save_initial_pin_to_session_data(state_machine_data)
|
||||
external_session_id = create_in_db_ussd_session.external_session_id
|
||||
in_memory_ussd_session = create_in_redis_ussd_session.get(external_session_id)
|
||||
in_memory_ussd_session = json.loads(in_memory_ussd_session)
|
||||
assert check_password_hash(
|
||||
password='1212', hashed_password=in_memory_ussd_session.get('session_data')['initial_pin'])
|
||||
|
||||
|
||||
@pytest.mark.parametrize('user_input, expected_result', [
|
||||
('1212', False),
|
||||
('0000', True)
|
||||
])
|
||||
def test_is_authorized_pin(create_activated_user, create_in_db_ussd_session, expected_result, user_input):
|
||||
serialized_in_db_ussd_session = create_in_db_ussd_session.to_json()
|
||||
state_machine_data = (user_input, serialized_in_db_ussd_session, create_activated_user)
|
||||
assert is_authorized_pin(state_machine_data=state_machine_data) is expected_result
|
||||
|
||||
|
||||
def test_is_not_blocked_pin(create_activated_user, create_in_db_ussd_session):
|
||||
serialized_in_db_ussd_session = create_in_db_ussd_session.to_json()
|
||||
state_machine_data = ('', serialized_in_db_ussd_session, create_activated_user)
|
||||
assert is_blocked_pin(state_machine_data=state_machine_data) is False
|
||||
|
||||
|
||||
def test_is_blocked_pin(create_pin_blocked_user, create_in_db_ussd_session):
|
||||
serialized_in_db_ussd_session = create_in_db_ussd_session.to_json()
|
||||
alt_state_machine_data = ('user_input', serialized_in_db_ussd_session, create_pin_blocked_user)
|
||||
assert is_blocked_pin(state_machine_data=alt_state_machine_data) is True
|
||||
|
||||
|
||||
@pytest.mark.parametrize('user_input, expected_result', [
|
||||
('1212', True),
|
||||
('0000', False)
|
||||
])
|
||||
def test_is_valid_new_pin(create_activated_user, create_in_db_ussd_session, expected_result, user_input):
|
||||
serialized_in_db_ussd_session = create_in_db_ussd_session.to_json()
|
||||
state_machine_data = (user_input, serialized_in_db_ussd_session, create_activated_user)
|
||||
assert is_valid_new_pin(state_machine_data=state_machine_data) is expected_result
|
||||
|
||||
36
apps/cic-ussd/tests/cic_ussd/state_machine/logic/test_sms.py
Normal file
36
apps/cic-ussd/tests/cic_ussd/state_machine/logic/test_sms.py
Normal file
@@ -0,0 +1,36 @@
|
||||
# standard imports
|
||||
|
||||
# third-party imports
|
||||
import pytest
|
||||
|
||||
# local imports
|
||||
from cic_ussd.state_machine.logic.sms import (send_terms_to_user_if_required,
|
||||
process_mini_statement_request,
|
||||
upsell_unregistered_recipient)
|
||||
|
||||
|
||||
def test_send_terms_to_user_if_required(caplog,
|
||||
create_in_db_ussd_session,
|
||||
create_activated_user):
|
||||
serialized_in_db_ussd_session = create_in_db_ussd_session.to_json()
|
||||
state_machine_data = ('', serialized_in_db_ussd_session, create_activated_user)
|
||||
send_terms_to_user_if_required(state_machine_data=state_machine_data)
|
||||
assert 'Requires integration to cic-notify.' in caplog.text
|
||||
|
||||
|
||||
def test_process_mini_statement_request(caplog,
|
||||
create_in_db_ussd_session,
|
||||
create_activated_user):
|
||||
serialized_in_db_ussd_session = create_in_db_ussd_session.to_json()
|
||||
state_machine_data = ('', serialized_in_db_ussd_session, create_activated_user)
|
||||
process_mini_statement_request(state_machine_data=state_machine_data)
|
||||
assert 'Requires integration to cic-notify.' in caplog.text
|
||||
|
||||
|
||||
def test_upsell_unregistered_recipient(caplog,
|
||||
create_in_db_ussd_session,
|
||||
create_activated_user):
|
||||
serialized_in_db_ussd_session = create_in_db_ussd_session.to_json()
|
||||
state_machine_data = ('', serialized_in_db_ussd_session, create_activated_user)
|
||||
upsell_unregistered_recipient(state_machine_data=state_machine_data)
|
||||
assert 'Requires integration to cic-notify.' in caplog.text
|
||||
@@ -0,0 +1,111 @@
|
||||
# standard imports
|
||||
import json
|
||||
|
||||
# third-party imports
|
||||
import pytest
|
||||
|
||||
# local imports
|
||||
from cic_ussd.state_machine.logic.transaction import (has_sufficient_balance,
|
||||
is_valid_recipient,
|
||||
is_valid_transaction_amount,
|
||||
process_transaction_request,
|
||||
save_recipient_phone_to_session_data,
|
||||
save_transaction_amount_to_session_data)
|
||||
from cic_ussd.redis import InMemoryStore
|
||||
|
||||
|
||||
@pytest.mark.parametrize("amount, expected_result", [
|
||||
('50', True),
|
||||
('', False)
|
||||
])
|
||||
def test_is_valid_transaction_amount(create_activated_user, create_in_db_ussd_session, amount, expected_result):
|
||||
state_machine_data = (amount, create_in_db_ussd_session, create_activated_user)
|
||||
validity = is_valid_transaction_amount(state_machine_data=state_machine_data)
|
||||
assert validity == expected_result
|
||||
|
||||
|
||||
def test_save_recipient_phone_to_session_data(create_activated_user,
|
||||
create_in_db_ussd_session,
|
||||
celery_session_worker,
|
||||
create_in_redis_ussd_session,
|
||||
init_database):
|
||||
phone_number = '+254712345678'
|
||||
in_memory_ussd_session = InMemoryStore.cache.get('AT974186')
|
||||
in_memory_ussd_session = json.loads(in_memory_ussd_session)
|
||||
|
||||
assert in_memory_ussd_session.get('session_data') == {}
|
||||
serialized_in_db_ussd_session = create_in_db_ussd_session.to_json()
|
||||
state_machine_data = (phone_number, serialized_in_db_ussd_session, create_activated_user)
|
||||
save_recipient_phone_to_session_data(state_machine_data=state_machine_data)
|
||||
|
||||
in_memory_ussd_session = InMemoryStore.cache.get('AT974186')
|
||||
in_memory_ussd_session = json.loads(in_memory_ussd_session)
|
||||
|
||||
assert in_memory_ussd_session.get('session_data')['recipient_phone_number'] == phone_number
|
||||
|
||||
|
||||
def test_save_transaction_amount_to_session_data(create_activated_user,
|
||||
create_in_db_ussd_session,
|
||||
celery_session_worker,
|
||||
create_in_redis_ussd_session,
|
||||
init_database):
|
||||
transaction_amount = '100'
|
||||
in_memory_ussd_session = InMemoryStore.cache.get('AT974186')
|
||||
in_memory_ussd_session = json.loads(in_memory_ussd_session)
|
||||
|
||||
assert in_memory_ussd_session.get('session_data') == {}
|
||||
serialized_in_db_ussd_session = create_in_db_ussd_session.to_json()
|
||||
state_machine_data = (transaction_amount, serialized_in_db_ussd_session, create_activated_user)
|
||||
save_transaction_amount_to_session_data(state_machine_data=state_machine_data)
|
||||
|
||||
in_memory_ussd_session = InMemoryStore.cache.get('AT974186')
|
||||
in_memory_ussd_session = json.loads(in_memory_ussd_session)
|
||||
|
||||
assert in_memory_ussd_session.get('session_data')['transaction_amount'] == transaction_amount
|
||||
|
||||
|
||||
@pytest.mark.parametrize("test_value, expected_result", [
|
||||
('45', True),
|
||||
('75', False)
|
||||
])
|
||||
def test_has_sufficient_balance(mock_balance,
|
||||
create_in_db_ussd_session,
|
||||
create_valid_tx_sender,
|
||||
expected_result,
|
||||
test_value):
|
||||
mock_balance(60)
|
||||
serialized_in_db_ussd_session = create_in_db_ussd_session.to_json()
|
||||
state_machine_data = (test_value, serialized_in_db_ussd_session, create_valid_tx_sender)
|
||||
result = has_sufficient_balance(state_machine_data=state_machine_data)
|
||||
assert result == expected_result
|
||||
|
||||
|
||||
@pytest.mark.parametrize("test_value, expected_result", [
|
||||
('+25498765432', True),
|
||||
('+25498765433', False)
|
||||
])
|
||||
def test_is_valid_recipient(create_in_db_ussd_session,
|
||||
create_valid_tx_recipient,
|
||||
create_valid_tx_sender,
|
||||
expected_result,
|
||||
test_value):
|
||||
serialized_in_db_ussd_session = create_in_db_ussd_session.to_json()
|
||||
state_machine_data = (test_value, serialized_in_db_ussd_session, create_valid_tx_sender)
|
||||
result = is_valid_recipient(state_machine_data=state_machine_data)
|
||||
assert result == expected_result
|
||||
|
||||
|
||||
def test_process_transaction_request(create_valid_tx_recipient,
|
||||
create_valid_tx_sender,
|
||||
load_config,
|
||||
mock_outgoing_transactions,
|
||||
setup_chain_spec,
|
||||
ussd_session_data):
|
||||
ussd_session_data['session_data'] = {
|
||||
'recipient_phone_number': create_valid_tx_recipient.phone_number,
|
||||
'transaction_amount': '50'
|
||||
}
|
||||
state_machine_data = ('', ussd_session_data, create_valid_tx_sender)
|
||||
process_transaction_request(state_machine_data=state_machine_data)
|
||||
assert mock_outgoing_transactions[0].get('amount') == 50.0
|
||||
assert mock_outgoing_transactions[0].get('token_symbol') == 'SRF'
|
||||
@@ -0,0 +1,155 @@
|
||||
# standard imports
|
||||
import json
|
||||
|
||||
# third-party-imports
|
||||
import pytest
|
||||
|
||||
# local imports
|
||||
from cic_ussd.chain import Chain
|
||||
from cic_ussd.redis import InMemoryStore
|
||||
from cic_ussd.state_machine.logic.user import (
|
||||
change_preferred_language_to_en,
|
||||
change_preferred_language_to_sw,
|
||||
edit_user_metadata_attribute,
|
||||
format_user_metadata,
|
||||
get_user_metadata,
|
||||
save_complete_user_metadata,
|
||||
process_gender_user_input,
|
||||
save_profile_attribute_to_session_data,
|
||||
update_account_status_to_active)
|
||||
|
||||
|
||||
def test_change_preferred_language(create_pending_user, create_in_db_ussd_session):
|
||||
state_machine_data = ('', create_in_db_ussd_session, create_pending_user)
|
||||
assert create_pending_user.preferred_language is None
|
||||
change_preferred_language_to_en(state_machine_data)
|
||||
assert create_pending_user.preferred_language == 'en'
|
||||
change_preferred_language_to_sw(state_machine_data)
|
||||
assert create_pending_user.preferred_language == 'sw'
|
||||
|
||||
|
||||
def test_update_account_status_to_active(create_pending_user, create_in_db_ussd_session):
|
||||
state_machine_data = ('', create_in_db_ussd_session, create_pending_user)
|
||||
update_account_status_to_active(state_machine_data)
|
||||
assert create_pending_user.get_account_status() == 'ACTIVE'
|
||||
|
||||
|
||||
@pytest.mark.parametrize("current_state, expected_key, expected_result, user_input", [
|
||||
("enter_given_name", "given_name", "John", "John"),
|
||||
("enter_family_name", "family_name", "Doe", "Doe"),
|
||||
("enter_gender", "gender", "Male", "1"),
|
||||
("enter_location", "location", "Kangemi", "Kangemi"),
|
||||
("enter_products", "products", "Mandazi", "Mandazi"),
|
||||
])
|
||||
def test_save_save_profile_attribute_to_session_data(current_state,
|
||||
expected_key,
|
||||
expected_result,
|
||||
user_input,
|
||||
celery_session_worker,
|
||||
create_activated_user,
|
||||
create_in_db_ussd_session,
|
||||
create_in_redis_ussd_session):
|
||||
create_in_db_ussd_session.state = current_state
|
||||
serialized_in_db_ussd_session = create_in_db_ussd_session.to_json()
|
||||
state_machine_data = (user_input, serialized_in_db_ussd_session, create_activated_user)
|
||||
in_memory_ussd_session = InMemoryStore.cache.get('AT974186')
|
||||
in_memory_ussd_session = json.loads(in_memory_ussd_session)
|
||||
assert in_memory_ussd_session.get('session_data') == {}
|
||||
serialized_in_db_ussd_session['state'] = current_state
|
||||
save_profile_attribute_to_session_data(state_machine_data=state_machine_data)
|
||||
|
||||
in_memory_ussd_session = InMemoryStore.cache.get('AT974186')
|
||||
in_memory_ussd_session = json.loads(in_memory_ussd_session)
|
||||
|
||||
assert in_memory_ussd_session.get('session_data')[expected_key] == expected_result
|
||||
|
||||
|
||||
@pytest.mark.parametrize("preferred_language, user_input, expected_gender_value", [
|
||||
("en", "1", "Male"),
|
||||
("en", "2", "Female"),
|
||||
("sw", "1", "Mwanaume"),
|
||||
("sw", "2", "Mwanamke"),
|
||||
])
|
||||
def test_process_gender_user_input(create_activated_user, expected_gender_value, preferred_language, user_input):
|
||||
create_activated_user.preferred_language = preferred_language
|
||||
gender = process_gender_user_input(user=create_activated_user, user_input=user_input)
|
||||
assert gender == expected_gender_value
|
||||
|
||||
|
||||
def test_format_user_metadata(create_activated_user,
|
||||
complete_user_metadata,
|
||||
setup_chain_spec):
|
||||
from cic_types.models.person import Person
|
||||
formatted_user_metadata = format_user_metadata(metadata=complete_user_metadata, user=create_activated_user)
|
||||
person = Person()
|
||||
user_metadata = person.deserialize(metadata=formatted_user_metadata)
|
||||
assert formatted_user_metadata == user_metadata.serialize()
|
||||
|
||||
|
||||
def test_save_complete_user_metadata(celery_session_worker,
|
||||
complete_user_metadata,
|
||||
create_activated_user,
|
||||
create_in_redis_ussd_session,
|
||||
mocker,
|
||||
setup_chain_spec,
|
||||
ussd_session_data):
|
||||
ussd_session = create_in_redis_ussd_session.get(ussd_session_data.get('external_session_id'))
|
||||
ussd_session = json.loads(ussd_session)
|
||||
ussd_session['session_data'] = complete_user_metadata
|
||||
user_metadata = format_user_metadata(metadata=ussd_session.get('session_data'), user=create_activated_user)
|
||||
state_machine_data = ('', ussd_session, create_activated_user)
|
||||
mocked_create_metadata_task = mocker.patch('cic_ussd.tasks.metadata.create_user_metadata.apply_async')
|
||||
save_complete_user_metadata(state_machine_data=state_machine_data)
|
||||
mocked_create_metadata_task.assert_called_with(
|
||||
(user_metadata, create_activated_user.blockchain_address),
|
||||
{},
|
||||
queue='cic-ussd'
|
||||
)
|
||||
|
||||
|
||||
def test_edit_user_metadata_attribute(celery_session_worker,
|
||||
cached_user_metadata,
|
||||
create_activated_user,
|
||||
create_in_redis_ussd_session,
|
||||
init_redis_cache,
|
||||
mocker,
|
||||
person_metadata,
|
||||
setup_chain_spec,
|
||||
ussd_session_data):
|
||||
ussd_session = create_in_redis_ussd_session.get(ussd_session_data.get('external_session_id'))
|
||||
ussd_session = json.loads(ussd_session)
|
||||
|
||||
assert person_metadata['location']['area_name'] == 'kayaba'
|
||||
|
||||
# appropriately format session
|
||||
ussd_session['session_data'] = {
|
||||
'location': 'nairobi'
|
||||
}
|
||||
state_machine_data = ('', ussd_session, create_activated_user)
|
||||
|
||||
mocked_edit_metadata = mocker.patch('cic_ussd.tasks.metadata.edit_user_metadata.apply_async')
|
||||
edit_user_metadata_attribute(state_machine_data=state_machine_data)
|
||||
person_metadata['location']['area_name'] = 'nairobi'
|
||||
mocked_edit_metadata.assert_called_with(
|
||||
(create_activated_user.blockchain_address, person_metadata, Chain.spec.engine()),
|
||||
{},
|
||||
queue='cic-ussd'
|
||||
)
|
||||
|
||||
|
||||
def test_get_user_metadata_attribute(celery_session_worker,
|
||||
create_activated_user,
|
||||
create_in_redis_ussd_session,
|
||||
mocker,
|
||||
ussd_session_data):
|
||||
ussd_session = create_in_redis_ussd_session.get(ussd_session_data.get('external_session_id'))
|
||||
ussd_session = json.loads(ussd_session)
|
||||
state_machine_data = ('', ussd_session, create_activated_user)
|
||||
|
||||
mocked_get_metadata = mocker.patch('cic_ussd.tasks.metadata.query_user_metadata.apply_async')
|
||||
get_user_metadata(state_machine_data=state_machine_data)
|
||||
mocked_get_metadata.assert_called_with(
|
||||
(create_activated_user.blockchain_address,),
|
||||
{},
|
||||
queue='cic-ussd'
|
||||
)
|
||||
@@ -0,0 +1,55 @@
|
||||
# standard imports
|
||||
import json
|
||||
|
||||
# third-party imports
|
||||
import pytest
|
||||
from cic_types.models.person import generate_metadata_pointer
|
||||
|
||||
# local imports
|
||||
from cic_ussd.metadata import blockchain_address_to_metadata_pointer
|
||||
from cic_ussd.redis import cache_data
|
||||
from cic_ussd.state_machine.logic.validator import (is_valid_name,
|
||||
is_valid_gender_selection,
|
||||
has_cached_user_metadata)
|
||||
|
||||
|
||||
@pytest.mark.parametrize("user_input, expected_result", [
|
||||
("Arya", True),
|
||||
("1234", False)
|
||||
])
|
||||
def test_is_valid_name(create_in_db_ussd_session, create_pending_user, user_input, expected_result):
|
||||
serialized_in_db_ussd_session = create_in_db_ussd_session.to_json()
|
||||
state_machine_data = (user_input, serialized_in_db_ussd_session, create_pending_user)
|
||||
result = is_valid_name(state_machine_data=state_machine_data)
|
||||
assert result is expected_result
|
||||
|
||||
|
||||
def test_has_cached_user_metadata(create_in_db_ussd_session,
|
||||
create_activated_user,
|
||||
init_redis_cache,
|
||||
person_metadata):
|
||||
serialized_in_db_ussd_session = create_in_db_ussd_session.to_json()
|
||||
state_machine_data = ('', serialized_in_db_ussd_session, create_activated_user)
|
||||
result = has_cached_user_metadata(state_machine_data=state_machine_data)
|
||||
assert result is False
|
||||
# cache metadata
|
||||
user = create_activated_user
|
||||
key = generate_metadata_pointer(
|
||||
identifier=blockchain_address_to_metadata_pointer(blockchain_address=user.blockchain_address),
|
||||
cic_type='cic.person'
|
||||
)
|
||||
cache_data(key=key, data=json.dumps(person_metadata))
|
||||
result = has_cached_user_metadata(state_machine_data=state_machine_data)
|
||||
assert result
|
||||
|
||||
|
||||
@pytest.mark.parametrize("user_input, expected_result", [
|
||||
("1", True),
|
||||
("2", True),
|
||||
("3", False)
|
||||
])
|
||||
def test_is_valid_gender_selection(create_in_db_ussd_session, create_pending_user, user_input, expected_result):
|
||||
serialized_in_db_ussd_session = create_in_db_ussd_session.to_json()
|
||||
state_machine_data = (user_input, serialized_in_db_ussd_session, create_pending_user)
|
||||
result = is_valid_gender_selection(state_machine_data=state_machine_data)
|
||||
assert result is expected_result
|
||||
@@ -0,0 +1,12 @@
|
||||
# local imports
|
||||
from cic_ussd.state_machine import UssdStateMachine
|
||||
|
||||
|
||||
def test_state_machine(create_in_db_ussd_session,
|
||||
get_in_redis_ussd_session,
|
||||
load_data_into_state_machine,
|
||||
create_pending_user):
|
||||
serialized_in_db_ussd_session = create_in_db_ussd_session.to_json()
|
||||
state_machine = UssdStateMachine(ussd_session=get_in_redis_ussd_session.to_json())
|
||||
state_machine.scan_data(('1', serialized_in_db_ussd_session, create_pending_user))
|
||||
assert state_machine.state == 'initial_pin_entry'
|
||||
@@ -0,0 +1,203 @@
|
||||
# standard imports
|
||||
import json
|
||||
import logging
|
||||
from datetime import datetime
|
||||
|
||||
# third party imports
|
||||
import celery
|
||||
import pytest
|
||||
|
||||
# local imports
|
||||
from cic_ussd.db.models.user import User
|
||||
from cic_ussd.error import ActionDataNotFoundError
|
||||
from cic_ussd.conversions import from_wei
|
||||
|
||||
logg = logging.getLogger()
|
||||
|
||||
|
||||
def test_successful_process_account_creation_callback_task(account_creation_action_data,
|
||||
celery_session_worker,
|
||||
init_database,
|
||||
init_redis_cache,
|
||||
mocker,
|
||||
set_account_creation_action_data):
|
||||
phone_number = account_creation_action_data.get('phone_number')
|
||||
task_id = account_creation_action_data.get('task_id')
|
||||
|
||||
mocked_task_request = mocker.patch('celery.app.task.Task.request')
|
||||
|
||||
# WARNING: [THE SETTING OF THE ROOT ID IS A HACK AND SHOULD BE REVIEWED OR IMPROVED]
|
||||
mocked_task_request.root_id = task_id
|
||||
|
||||
user = init_database.query(User).filter_by(phone_number=phone_number).first()
|
||||
assert user is None
|
||||
|
||||
redis_cache = init_redis_cache
|
||||
action_data = redis_cache.get(task_id)
|
||||
action_data = json.loads(action_data)
|
||||
|
||||
assert action_data.get('status') == 'PENDING'
|
||||
|
||||
status_code = 0
|
||||
result = '0x6315c185fd23bDbbba058E2a504197915aCC5065'
|
||||
url = ''
|
||||
|
||||
s_process_callback_request = celery.signature(
|
||||
'cic_ussd.tasks.callback_handler.process_account_creation_callback',
|
||||
[result, url, status_code]
|
||||
)
|
||||
s_process_callback_request.apply_async().get()
|
||||
|
||||
user = init_database.query(User).filter_by(phone_number=phone_number).first()
|
||||
assert user.blockchain_address == result
|
||||
|
||||
action_data = redis_cache.get(task_id)
|
||||
action_data = json.loads(action_data)
|
||||
|
||||
assert action_data.get('status') == 'CREATED'
|
||||
|
||||
|
||||
def test_unsuccessful_process_account_creation_callback_task(init_database,
|
||||
init_redis_cache,
|
||||
celery_session_worker):
|
||||
with pytest.raises(ActionDataNotFoundError) as error:
|
||||
status_code = 0
|
||||
result = '0x6315c185fd23bDbbba058E2a504197915aCC5065'
|
||||
url = ''
|
||||
|
||||
s_process_callback_request = celery.signature(
|
||||
'cic_ussd.tasks.callback_handler.process_account_creation_callback',
|
||||
[result, url, status_code]
|
||||
)
|
||||
result = s_process_callback_request.apply_async()
|
||||
task_id = result.get()
|
||||
|
||||
assert str(error.value) == f'Account creation task: {task_id}, returned unexpected response: {status_code}'
|
||||
|
||||
|
||||
def test_successful_token_gift_incoming_transaction(celery_session_worker,
|
||||
create_activated_user,
|
||||
mock_notifier_api,
|
||||
set_locale_files,
|
||||
successful_incoming_token_gift_callback):
|
||||
result = successful_incoming_token_gift_callback.get('RESULT')
|
||||
param = successful_incoming_token_gift_callback.get('PARAM')
|
||||
status_code = successful_incoming_token_gift_callback.get('STATUS_CODE')
|
||||
|
||||
s_process_token_gift = celery.signature(
|
||||
'cic_ussd.tasks.callback_handler.process_incoming_transfer_callback',
|
||||
[result, param, status_code]
|
||||
)
|
||||
s_process_token_gift.apply_async().get()
|
||||
|
||||
balance = from_wei(result.get('destination_value'))
|
||||
token_symbol = result.get('token_symbol')
|
||||
|
||||
messages = mock_notifier_api
|
||||
|
||||
assert messages[0].get('recipient') == create_activated_user.phone_number
|
||||
assert messages[0].get(
|
||||
'message') == f'Hello {""} you have been registered on Sarafu Network! Your balance is {balance} {token_symbol}. To use dial *483*46#. For help 0757628885.'
|
||||
|
||||
|
||||
def test_successful_transfer_incoming_transaction(celery_session_worker,
|
||||
create_valid_tx_sender,
|
||||
create_valid_tx_recipient,
|
||||
mock_notifier_api,
|
||||
set_locale_files,
|
||||
successful_incoming_transfer_callback):
|
||||
result = successful_incoming_transfer_callback.get('RESULT')
|
||||
param = successful_incoming_transfer_callback.get('PARAM')
|
||||
status_code = successful_incoming_transfer_callback.get('STATUS_CODE')
|
||||
|
||||
s_process_token_gift = celery.signature(
|
||||
'cic_ussd.tasks.callback_handler.process_incoming_transfer_callback',
|
||||
[result, param, status_code]
|
||||
)
|
||||
s_process_token_gift.apply_async().get()
|
||||
|
||||
value = result.get('destination_value')
|
||||
balance = ''
|
||||
token_symbol = result.get('token_symbol')
|
||||
|
||||
sender_first_name = ''
|
||||
sender_last_name = ''
|
||||
phone_number = create_valid_tx_sender.phone_number
|
||||
tx_sender_information = f'{phone_number}, {sender_first_name}, {sender_last_name}'
|
||||
amount = from_wei(value=value)
|
||||
timestamp = datetime.now().strftime('%d-%m-%y, %H:%M %p')
|
||||
|
||||
messages = mock_notifier_api
|
||||
|
||||
assert messages[0].get('recipient') == create_valid_tx_recipient.phone_number
|
||||
assert messages[0].get(
|
||||
'message') == f'Successfully received {amount} {token_symbol} from {tx_sender_information} {timestamp}. New balance is {balance} {token_symbol}.'
|
||||
|
||||
|
||||
def test_unsuccessful_incoming_transaction_recipient_not_found(celery_session_worker,
|
||||
create_valid_tx_sender,
|
||||
successful_incoming_transfer_callback):
|
||||
result = successful_incoming_transfer_callback.get('RESULT')
|
||||
param = successful_incoming_transfer_callback.get('PARAM')
|
||||
status_code = successful_incoming_transfer_callback.get('STATUS_CODE')
|
||||
|
||||
with pytest.raises(ValueError) as error:
|
||||
s_process_token_gift = celery.signature(
|
||||
'cic_ussd.tasks.callback_handler.process_incoming_transfer_callback',
|
||||
[result, param, status_code]
|
||||
)
|
||||
s_process_token_gift.apply_async().get()
|
||||
|
||||
recipient_blockchain_address = result.get('recipient')
|
||||
assert str(error.value) == f'Tx for recipient: {recipient_blockchain_address} was received but has no matching user in the system.'
|
||||
|
||||
|
||||
def test_successful_incoming_transaction_sender_not_found(caplog,
|
||||
celery_session_worker,
|
||||
create_valid_tx_recipient,
|
||||
mock_notifier_api,
|
||||
successful_incoming_transfer_callback):
|
||||
result = successful_incoming_transfer_callback.get('RESULT')
|
||||
param = successful_incoming_transfer_callback.get('PARAM')
|
||||
status_code = successful_incoming_transfer_callback.get('STATUS_CODE')
|
||||
s_process_token_gift = celery.signature(
|
||||
'cic_ussd.tasks.callback_handler.process_incoming_transfer_callback',
|
||||
[result, param, status_code]
|
||||
)
|
||||
s_process_token_gift.apply_async().get()
|
||||
|
||||
sender_blockchain_address = result.get('sender')
|
||||
assert 'Balance requires implementation of cic-eth integration with balance.' in caplog.text
|
||||
# assert f'Tx with sender: {sender_blockchain_address} was received but has no matching user in the system.\n' in caplog.text
|
||||
|
||||
|
||||
def test_unsuccessful_incoming_transaction_invalid_status_code(celery_session_worker,
|
||||
incoming_transfer_callback_invalid_tx_status_code):
|
||||
result = incoming_transfer_callback_invalid_tx_status_code.get('RESULT')
|
||||
param = incoming_transfer_callback_invalid_tx_status_code.get('PARAM')
|
||||
status_code = incoming_transfer_callback_invalid_tx_status_code.get('STATUS_CODE')
|
||||
|
||||
with pytest.raises(ValueError) as error:
|
||||
s_process_token_gift = celery.signature(
|
||||
'cic_ussd.tasks.callback_handler.process_incoming_transfer_callback',
|
||||
[result, param, status_code]
|
||||
)
|
||||
s_process_token_gift.apply_async().get()
|
||||
|
||||
assert str(error.value) == f'Unexpected status code: {status_code}'
|
||||
|
||||
|
||||
def test_unsuccessful_incoming_transaction_invalid_param(celery_session_worker,
|
||||
incoming_transfer_callback_invalid_tx_param):
|
||||
result = incoming_transfer_callback_invalid_tx_param.get('RESULT')
|
||||
param = incoming_transfer_callback_invalid_tx_param.get('PARAM')
|
||||
status_code = incoming_transfer_callback_invalid_tx_param.get('STATUS_CODE')
|
||||
|
||||
with pytest.raises(ValueError) as error:
|
||||
s_process_token_gift = celery.signature(
|
||||
'cic_ussd.tasks.callback_handler.process_incoming_transfer_callback',
|
||||
[result, param, status_code]
|
||||
)
|
||||
s_process_token_gift.apply_async().get()
|
||||
|
||||
assert str(error.value) == f'Unexpected transaction: param {status_code}'
|
||||
@@ -0,0 +1,46 @@
|
||||
# third party imports
|
||||
import celery
|
||||
import pytest
|
||||
|
||||
# local imports
|
||||
from cic_ussd.db.models.ussd_session import UssdSession
|
||||
from cic_ussd.error import SessionNotFoundError
|
||||
|
||||
|
||||
def test_persist_session_to_db_task(
|
||||
init_database,
|
||||
create_activated_user,
|
||||
ussd_session_data,
|
||||
celery_session_worker,
|
||||
create_in_redis_ussd_session):
|
||||
external_session_id = ussd_session_data.get('external_session_id')
|
||||
s_persist_session_to_db = celery.signature(
|
||||
'cic_ussd.tasks.ussd_session.persist_session_to_db',
|
||||
[external_session_id]
|
||||
)
|
||||
result = s_persist_session_to_db.apply_async()
|
||||
result.get()
|
||||
db_session = init_database.query(UssdSession).filter_by(external_session_id=external_session_id).first()
|
||||
assert db_session.external_session_id == 'AT974186'
|
||||
assert db_session.service_code == '*483*46#'
|
||||
assert db_session.msisdn == '+25498765432'
|
||||
assert db_session.user_input == '1'
|
||||
assert db_session.state == 'initial_language_selection'
|
||||
assert db_session.session_data is None
|
||||
assert db_session.version == 2
|
||||
|
||||
assert UssdSession.have_session_for_phone(create_activated_user.phone_number)
|
||||
|
||||
|
||||
def test_session_not_found_error(
|
||||
celery_session_worker,
|
||||
create_in_redis_ussd_session):
|
||||
with pytest.raises(SessionNotFoundError) as error:
|
||||
external_session_id = 'SomeRandomValue'
|
||||
s_persist_session_to_db = celery.signature(
|
||||
'cic_ussd.tasks.ussd_session.persist_session_to_db',
|
||||
[external_session_id]
|
||||
)
|
||||
result = s_persist_session_to_db.apply_async()
|
||||
result.get()
|
||||
assert str(error.value) == "Session does not exist!"
|
||||
20
apps/cic-ussd/tests/cic_ussd/test_accounts.py
Normal file
20
apps/cic-ussd/tests/cic_ussd/test_accounts.py
Normal file
@@ -0,0 +1,20 @@
|
||||
# standard imports
|
||||
|
||||
# third-party imports
|
||||
|
||||
# local imports
|
||||
from cic_ussd.balance import BalanceManager
|
||||
from cic_ussd.chain import Chain
|
||||
|
||||
|
||||
def test_balance_manager(create_valid_tx_recipient, load_config, mocker, setup_chain_spec):
|
||||
chain_str = Chain.spec.__str__()
|
||||
balance_manager = BalanceManager(
|
||||
address=create_valid_tx_recipient.blockchain_address,
|
||||
chain_str=chain_str,
|
||||
token_symbol='SRF'
|
||||
)
|
||||
balance_manager.get_balances = mocker.MagicMock()
|
||||
balance_manager.get_balances()
|
||||
|
||||
balance_manager.get_balances.assert_called_once()
|
||||
19
apps/cic-ussd/tests/cic_ussd/test_encoder.py
Normal file
19
apps/cic-ussd/tests/cic_ussd/test_encoder.py
Normal file
@@ -0,0 +1,19 @@
|
||||
# local imports
|
||||
from cic_ussd.encoder import check_password_hash, create_password_hash, PasswordEncoder
|
||||
|
||||
|
||||
def test_password_encoder(load_config, set_fernet_key):
|
||||
assert PasswordEncoder.key == load_config.get('APP_PASSWORD_PEPPER')
|
||||
|
||||
|
||||
def test_create_password_hash(load_config, set_fernet_key):
|
||||
fernet_key = PasswordEncoder.key
|
||||
assert fernet_key == load_config.get('APP_PASSWORD_PEPPER')
|
||||
password_hash = create_password_hash(password='Password')
|
||||
assert password_hash != 'Password'
|
||||
assert type(password_hash) == str
|
||||
|
||||
|
||||
def test_check_password_hash():
|
||||
password_hash = create_password_hash(password='Password')
|
||||
assert check_password_hash(password='Password', hashed_password=password_hash) is True
|
||||
36
apps/cic-ussd/tests/cic_ussd/test_notifications.py
Normal file
36
apps/cic-ussd/tests/cic_ussd/test_notifications.py
Normal file
@@ -0,0 +1,36 @@
|
||||
# standard imports
|
||||
|
||||
# third-party imports
|
||||
import pytest
|
||||
|
||||
# local imports
|
||||
from cic_ussd.notifications import Notifier
|
||||
|
||||
|
||||
@pytest.mark.parametrize("key, preferred_language, recipient, expected_message", [
|
||||
("ussd.kenya.exit", "en", "+254712345678", "END Thank you for using the service."),
|
||||
("ussd.kenya.exit", "sw", "+254712345678", "END Asante kwa kutumia huduma.")
|
||||
])
|
||||
def test_send_sms_notification(celery_session_worker,
|
||||
expected_message,
|
||||
key,
|
||||
preferred_language,
|
||||
recipient,
|
||||
set_locale_files,
|
||||
mock_notifier_api):
|
||||
|
||||
notifier = Notifier()
|
||||
notifier.queue = None
|
||||
|
||||
notifier.send_sms_notification(key=key, phone_number=recipient, preferred_language=preferred_language)
|
||||
messages = mock_notifier_api
|
||||
|
||||
assert messages[0].get('message') == expected_message
|
||||
assert messages[0].get('recipient') == recipient
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
243
apps/cic-ussd/tests/cic_ussd/test_operations.py
Normal file
243
apps/cic-ussd/tests/cic_ussd/test_operations.py
Normal file
@@ -0,0 +1,243 @@
|
||||
# standard imports
|
||||
import json
|
||||
import uuid
|
||||
|
||||
# third party imports
|
||||
import pytest
|
||||
|
||||
# local imports
|
||||
from cic_ussd.chain import Chain
|
||||
from cic_ussd.db.models.task_tracker import TaskTracker
|
||||
from cic_ussd.menu.ussd_menu import UssdMenu
|
||||
from cic_ussd.operations import (add_tasks_to_tracker,
|
||||
create_ussd_session,
|
||||
create_or_update_session,
|
||||
define_response_with_content,
|
||||
define_multilingual_responses,
|
||||
get_account_status,
|
||||
get_latest_input,
|
||||
initiate_account_creation_request,
|
||||
process_current_menu,
|
||||
process_menu_interaction_requests,
|
||||
cache_account_creation_task_id,
|
||||
reset_pin,
|
||||
update_ussd_session,
|
||||
save_to_in_memory_ussd_session_data)
|
||||
from cic_ussd.phone_number import get_user_by_phone_number,process_phone_number
|
||||
from cic_ussd.transactions import truncate
|
||||
from cic_ussd.redis import InMemoryStore
|
||||
from cic_ussd.session.ussd_session import UssdSession as InMemoryUssdSession
|
||||
|
||||
|
||||
def test_add_tasks_to_tracker(init_database):
|
||||
task_uuid = '31e85315-feee-4b6d-995e-223569082cc4'
|
||||
session = init_database
|
||||
assert len(session.query(TaskTracker).all()) == 0
|
||||
|
||||
add_tasks_to_tracker(task_uuid=task_uuid)
|
||||
task_in_tracker = session.query(TaskTracker).filter_by(task_uuid=task_uuid).first()
|
||||
assert task_in_tracker.id == 1
|
||||
assert task_in_tracker.task_uuid == task_uuid
|
||||
|
||||
|
||||
def test_create_ussd_session(create_in_redis_ussd_session, ussd_session_data):
|
||||
external_session_id = ussd_session_data.get('external_session_id')
|
||||
ussd_session = create_ussd_session(
|
||||
external_session_id=external_session_id,
|
||||
service_code=ussd_session_data.get('service_code'),
|
||||
phone=ussd_session_data.get('msisdn'),
|
||||
user_input=ussd_session_data.get('user_input'),
|
||||
current_menu=ussd_session_data.get('state')
|
||||
)
|
||||
in_memory_ussd_session = create_in_redis_ussd_session.get(external_session_id)
|
||||
assert json.loads(in_memory_ussd_session).get('external_session_id') == ussd_session.external_session_id
|
||||
|
||||
|
||||
def test_create_or_update_session(init_database, create_in_redis_ussd_session, ussd_session_data):
|
||||
external_session_id = ussd_session_data.get('external_session_id')
|
||||
ussd_session = create_or_update_session(external_session_id=external_session_id,
|
||||
service_code=ussd_session_data.get('service_code'),
|
||||
phone=ussd_session_data.get('msisdn'),
|
||||
user_input=ussd_session_data.get('user_input'),
|
||||
current_menu=ussd_session_data.get('state'))
|
||||
in_memory_ussd_session = create_in_redis_ussd_session.get(external_session_id)
|
||||
assert json.loads(in_memory_ussd_session).get('external_session_id') == ussd_session.external_session_id
|
||||
|
||||
|
||||
@pytest.mark.parametrize('headers, response, expected_result',[
|
||||
([('Content-Type', 'text/plain')], 'some-text', (b'some-text', [('Content-Type', 'text/plain'), ('Content-Length', '9')])),
|
||||
([('Content-Type', 'text/plain'), ('Content-Length', '0')], 'some-text', (b'some-text', [('Content-Type', 'text/plain'), ('Content-Length', '9')]))
|
||||
])
|
||||
def test_define_response_with_content(headers, response, expected_result):
|
||||
response_bytes, headers = define_response_with_content(headers=headers, response=response)
|
||||
assert response_bytes, headers == expected_result
|
||||
|
||||
|
||||
def test_define_multilingual_responses(load_ussd_menu, set_locale_files):
|
||||
response = define_multilingual_responses(
|
||||
key='ussd.kenya.account_creation_prompt', locales=['en', 'sw'], prefix='END')
|
||||
assert response == "END Your account is being created. You will receive an SMS when your account is ready.\nAkaunti yako ya Sarafu inatayarishwa. Utapokea ujumbe wa SMS akaunti yako ikiwa tayari.\n"
|
||||
|
||||
|
||||
def test_get_account_status(create_pending_user):
|
||||
user = create_pending_user
|
||||
assert get_account_status(user.phone_number) == 'PENDING'
|
||||
|
||||
|
||||
@pytest.mark.parametrize('user_input, expected_value', [
|
||||
('1*9*6*7', '7'),
|
||||
('1', '1'),
|
||||
('', '')
|
||||
])
|
||||
def test_get_latest_input(user_input, expected_value):
|
||||
assert get_latest_input(user_input=user_input) == expected_value
|
||||
|
||||
|
||||
def test_initiate_account_creation_request(account_creation_action_data,
|
||||
create_in_redis_ussd_session,
|
||||
init_database,
|
||||
load_config,
|
||||
load_ussd_menu,
|
||||
mocker,
|
||||
setup_chain_spec,
|
||||
set_locale_files,
|
||||
ussd_session_data):
|
||||
external_session_id = ussd_session_data.get('external_session_id')
|
||||
phone_number = account_creation_action_data.get('phone_number')
|
||||
task_id = account_creation_action_data.get('task_id')
|
||||
|
||||
class Callable:
|
||||
id = task_id
|
||||
|
||||
mocker.patch('cic_eth.api.api_task.Api.create_account', return_value=Callable)
|
||||
mocked_cache_function = mocker.patch('cic_ussd.operations.cache_account_creation_task_id')
|
||||
mocked_cache_function(phone_number, task_id)
|
||||
|
||||
chain_str = Chain.spec.__str__()
|
||||
response = initiate_account_creation_request(chain_str=chain_str,
|
||||
external_session_id=external_session_id,
|
||||
phone_number=ussd_session_data.get('msisdn'),
|
||||
service_code=ussd_session_data.get('service_code'),
|
||||
user_input=ussd_session_data.get('user_input'))
|
||||
in_memory_ussd_session = InMemoryUssdSession.redis_cache.get(external_session_id)
|
||||
|
||||
# check that ussd session was created
|
||||
assert json.loads(in_memory_ussd_session).get('external_session_id') == external_session_id
|
||||
assert response == "END Your account is being created. You will receive an SMS when your account is ready.\nAkaunti yako ya Sarafu inatayarishwa. Utapokea ujumbe wa SMS akaunti yako ikiwa tayari.\n"
|
||||
|
||||
|
||||
def test_reset_pin(create_pin_blocked_user):
|
||||
user = create_pin_blocked_user
|
||||
assert user.get_account_status() == 'LOCKED'
|
||||
reset_pin(user.phone_number)
|
||||
assert user.get_account_status() == 'RESET'
|
||||
|
||||
|
||||
def test_update_ussd_session(create_in_redis_ussd_session, load_ussd_menu, ussd_session_data):
|
||||
external_session_id = ussd_session_data.get('external_session_id')
|
||||
ussd_session = create_ussd_session(external_session_id=external_session_id,
|
||||
service_code=ussd_session_data.get('service_code'),
|
||||
phone=ussd_session_data.get('msisdn'),
|
||||
user_input=ussd_session_data.get('user_input'),
|
||||
current_menu=ussd_session_data.get('state')
|
||||
)
|
||||
assert ussd_session.user_input == ussd_session_data.get('user_input')
|
||||
assert ussd_session.state == ussd_session_data.get('state')
|
||||
ussd_session = update_ussd_session(ussd_session=ussd_session, user_input='1*2', current_menu='initial_pin_entry')
|
||||
assert ussd_session.user_input == '1*2'
|
||||
assert ussd_session.state == 'initial_pin_entry'
|
||||
|
||||
|
||||
def test_process_current_menu(create_activated_user, create_in_db_ussd_session):
|
||||
ussd_session = create_in_db_ussd_session
|
||||
current_menu = process_current_menu(ussd_session=ussd_session, user=create_activated_user, user_input="")
|
||||
assert current_menu == UssdMenu.find_by_name(name='exit_invalid_input')
|
||||
current_menu = process_current_menu(ussd_session=None, user=create_activated_user, user_input="1*0000")
|
||||
assert current_menu == UssdMenu.find_by_name(name='start')
|
||||
|
||||
|
||||
def test_cache_account_creation_task_id(init_redis_cache):
|
||||
phone_number = '+25412345678'
|
||||
task_id = str(uuid.uuid4())
|
||||
cache_account_creation_task_id(phone_number=phone_number, task_id=task_id)
|
||||
|
||||
redis_cache = init_redis_cache
|
||||
action_data = redis_cache.get(task_id)
|
||||
action_data = json.loads(action_data)
|
||||
|
||||
assert action_data.get('phone_number') == phone_number
|
||||
assert action_data.get('sms_notification_sent') is False
|
||||
assert action_data.get('status') == 'PENDING'
|
||||
assert action_data.get('task_id') == task_id
|
||||
|
||||
|
||||
def test_save_to_in_memory_ussd_session_data(celery_session_worker,
|
||||
create_in_db_ussd_session,
|
||||
create_in_redis_ussd_session,
|
||||
init_database):
|
||||
|
||||
in_memory_ussd_session = InMemoryStore.cache.get('AT974186')
|
||||
in_memory_ussd_session = json.loads(in_memory_ussd_session)
|
||||
|
||||
assert in_memory_ussd_session.get('session_data') == {}
|
||||
|
||||
session_data = {
|
||||
'some_test_key': 'some_test_value'
|
||||
}
|
||||
save_to_in_memory_ussd_session_data(
|
||||
queue='cic-ussd',
|
||||
session_data=session_data,
|
||||
ussd_session=create_in_db_ussd_session.to_json()
|
||||
)
|
||||
|
||||
in_memory_ussd_session = InMemoryStore.cache.get('AT974186')
|
||||
in_memory_ussd_session = json.loads(in_memory_ussd_session)
|
||||
|
||||
assert in_memory_ussd_session.get('session_data') == session_data
|
||||
|
||||
|
||||
@pytest.mark.parametrize("external_session_id, phone_number, expected_response", [
|
||||
("AT123456789", "+254700000000", "END Your account is being created. You will receive an SMS when your account is ready.\nAkaunti yako ya Sarafu inatayarishwa. Utapokea ujumbe wa SMS akaunti yako ikiwa tayari.\n"),
|
||||
("AT974186", "+25498765432", "CON Please enter a PIN to manage your account.\n0. Back")
|
||||
])
|
||||
def test_process_menu_interaction_requests(external_session_id,
|
||||
phone_number,
|
||||
expected_response,
|
||||
load_ussd_menu,
|
||||
load_data_into_state_machine,
|
||||
load_config,
|
||||
setup_chain_spec,
|
||||
celery_session_worker,
|
||||
create_activated_user,
|
||||
create_in_db_ussd_session):
|
||||
chain_str = Chain.spec.__str__()
|
||||
response = process_menu_interaction_requests(
|
||||
chain_str=chain_str,
|
||||
external_session_id=external_session_id,
|
||||
phone_number=phone_number,
|
||||
queue='cic-ussd',
|
||||
service_code=load_config.get('APP_SERVICE_CODE'),
|
||||
user_input='1'
|
||||
)
|
||||
|
||||
assert response == expected_response
|
||||
|
||||
|
||||
@pytest.mark.parametrize("phone_number, region, expected_result", [
|
||||
("0712345678", "KE", "+254712345678"),
|
||||
("+254787654321", "KE", "+254787654321")
|
||||
])
|
||||
def test_process_phone_number(expected_result, phone_number, region):
|
||||
processed_phone_number = process_phone_number(phone_number=phone_number, region=region)
|
||||
assert processed_phone_number == expected_result
|
||||
|
||||
|
||||
def test_get_user_by_phone_number(create_activated_user):
|
||||
known_phone_number = create_activated_user.phone_number
|
||||
user = get_user_by_phone_number(phone_number=known_phone_number)
|
||||
assert user is not None
|
||||
assert create_activated_user.blockchain_address == user.blockchain_address
|
||||
|
||||
unknown_phone_number = '+254700000000'
|
||||
user = get_user_by_phone_number(phone_number=unknown_phone_number)
|
||||
assert user is None
|
||||
130
apps/cic-ussd/tests/cic_ussd/test_processor.py
Normal file
130
apps/cic-ussd/tests/cic_ussd/test_processor.py
Normal file
@@ -0,0 +1,130 @@
|
||||
# local imports
|
||||
from cic_ussd.menu.ussd_menu import UssdMenu
|
||||
from cic_ussd.processor import (custom_display_text,
|
||||
next_state,
|
||||
process_request,
|
||||
process_pin_authorization,
|
||||
process_transaction_pin_authorization,
|
||||
process_exit_insufficient_balance,
|
||||
process_exit_successful_transaction)
|
||||
|
||||
|
||||
def test_process_pin_authorization(create_activated_user,
|
||||
load_ussd_menu,
|
||||
set_locale_files):
|
||||
ussd_menu = UssdMenu.find_by_name(name='standard_pin_authorization')
|
||||
response = process_pin_authorization(
|
||||
display_key=ussd_menu.get('display_key'),
|
||||
user=create_activated_user
|
||||
)
|
||||
assert response == 'CON Please enter your PIN.\n0. Back'
|
||||
|
||||
user_with_one_failed_pin_attempt = create_activated_user
|
||||
user_with_one_failed_pin_attempt.failed_pin_attempts = 1
|
||||
alt_response = process_pin_authorization(
|
||||
display_key=ussd_menu.get('display_key'),
|
||||
user=user_with_one_failed_pin_attempt,
|
||||
)
|
||||
assert alt_response == 'CON Please enter your PIN. You have 2 attempts remaining.\n0. Back'
|
||||
|
||||
|
||||
def test_process_transaction_pin_authorization(create_activated_user,
|
||||
create_in_db_ussd_session,
|
||||
load_ussd_menu,
|
||||
set_locale_files):
|
||||
session_data = {
|
||||
'recipient_phone_number': '+254700000000',
|
||||
}
|
||||
ussd_session = create_in_db_ussd_session.to_json()
|
||||
ussd_session['session_data'] = session_data
|
||||
ussd_session['user_input'] = '1*0700000000*120'
|
||||
ussd_menu = UssdMenu.find_by_name(name='transaction_pin_authorization')
|
||||
response = process_transaction_pin_authorization(
|
||||
display_key=ussd_menu.get('display_key'),
|
||||
user=create_activated_user,
|
||||
ussd_session=ussd_session
|
||||
)
|
||||
assert response == 'CON +254700000000 will receive 120.00 SRF from +25498765432.\nPlease enter your PIN to confirm.\n0. Back'
|
||||
|
||||
|
||||
def test_process_request_for_pending_user(load_ussd_menu, create_pending_user):
|
||||
expected_menu = process_request(user_input="", user=create_pending_user)
|
||||
assert expected_menu == UssdMenu.find_by_name(name='initial_language_selection')
|
||||
|
||||
|
||||
def test_processor_request_for_activated_user(load_ussd_menu, create_activated_user):
|
||||
expected_menu = process_request(user_input="", user=create_activated_user)
|
||||
assert expected_menu == UssdMenu.find_by_name(name="start")
|
||||
|
||||
|
||||
def test_next_state(load_data_into_state_machine, load_ussd_menu, create_in_db_ussd_session, create_pending_user):
|
||||
assert create_in_db_ussd_session.state == "initial_language_selection"
|
||||
successive_state = next_state(
|
||||
ussd_session=create_in_db_ussd_session.to_json(),
|
||||
user=create_pending_user,
|
||||
user_input="1"
|
||||
)
|
||||
assert successive_state == "initial_pin_entry"
|
||||
|
||||
|
||||
def test_custom_display_text(create_activated_user,
|
||||
get_in_redis_ussd_session,
|
||||
load_ussd_menu,
|
||||
set_locale_files):
|
||||
ussd_session = get_in_redis_ussd_session
|
||||
user = create_activated_user
|
||||
ussd_menu = UssdMenu.find_by_name(name='exit_invalid_request')
|
||||
english_translation = custom_display_text(
|
||||
display_key=ussd_menu.get('display_key'),
|
||||
menu_name=ussd_menu.get('name'),
|
||||
user=user,
|
||||
ussd_session=ussd_session
|
||||
)
|
||||
user.preferred_language = 'sw'
|
||||
swahili_translation = custom_display_text(
|
||||
display_key=ussd_menu.get('display_key'),
|
||||
menu_name=ussd_menu.get('name'),
|
||||
user=user,
|
||||
ussd_session=ussd_session
|
||||
)
|
||||
assert swahili_translation == 'END Chaguo si sahihi.'
|
||||
assert english_translation == 'END Invalid request.'
|
||||
|
||||
|
||||
def test_process_exit_insufficient_balance(
|
||||
create_valid_tx_recipient,
|
||||
load_ussd_menu,
|
||||
mock_balance,
|
||||
set_locale_files,
|
||||
ussd_session_data):
|
||||
mock_balance(50)
|
||||
ussd_session_data['user_input'] = f'1*{create_valid_tx_recipient.phone_number}*75'
|
||||
ussd_session_data['session_data'] = {'recipient_phone_number': create_valid_tx_recipient.phone_number}
|
||||
ussd_session_data['display_key'] = 'exit_insufficient_balance'
|
||||
ussd_menu = UssdMenu.find_by_name(name='exit_insufficient_balance')
|
||||
response = process_exit_insufficient_balance(
|
||||
display_key=ussd_menu.get('display_key'),
|
||||
user=create_valid_tx_recipient,
|
||||
ussd_session=ussd_session_data
|
||||
)
|
||||
assert response == 'CON Payment of 75.00 SRF to +25498765432 has failed due to insufficent balance.\nYour Sarafu-Network balances is: 50.00\n00. Back\n99. Exit'
|
||||
|
||||
|
||||
def test_process_exit_successful_transaction(
|
||||
create_valid_tx_recipient,
|
||||
create_valid_tx_sender,
|
||||
load_ussd_menu,
|
||||
set_locale_files,
|
||||
ussd_session_data):
|
||||
ussd_session_data['session_data'] = {
|
||||
'recipient_phone_number': create_valid_tx_recipient.phone_number,
|
||||
'transaction_amount': 75
|
||||
}
|
||||
ussd_session_data['display_key'] = 'exit_successful_transaction'
|
||||
ussd_menu = UssdMenu.find_by_name(name='exit_successful_transaction')
|
||||
response = process_exit_successful_transaction(
|
||||
display_key=ussd_menu.get('display_key'),
|
||||
user=create_valid_tx_sender,
|
||||
ussd_session=ussd_session_data
|
||||
)
|
||||
assert response == 'CON Your request has been sent. +25498765432 will receive 75.00 SRF from +25498765433.\n00. Back\n99. Exit'
|
||||
65
apps/cic-ussd/tests/cic_ussd/test_requests.py
Normal file
65
apps/cic-ussd/tests/cic_ussd/test_requests.py
Normal file
@@ -0,0 +1,65 @@
|
||||
# standard imports
|
||||
import json
|
||||
|
||||
# local imports
|
||||
from cic_ussd.db.models.user import User
|
||||
from cic_ussd.requests import (get_query_parameters,
|
||||
get_request_endpoint,
|
||||
get_request_method,
|
||||
process_pin_reset_requests,
|
||||
process_locked_accounts_requests)
|
||||
|
||||
|
||||
def test_get_query_parameters(get_request_with_params_env):
|
||||
param = get_query_parameters(env=get_request_with_params_env, query_name='phone')
|
||||
assert param == '0700000000'
|
||||
|
||||
|
||||
def test_get_request_endpoint(valid_locked_accounts_env):
|
||||
param = get_request_endpoint(env=valid_locked_accounts_env)
|
||||
assert param == '/accounts/locked/10/10'
|
||||
|
||||
|
||||
def test_get_request_method(valid_locked_accounts_env):
|
||||
param = get_request_method(env=valid_locked_accounts_env)
|
||||
assert param == 'GET'
|
||||
|
||||
|
||||
def test_process_pin_reset_requests(uwsgi_env, create_pin_blocked_user):
|
||||
env = uwsgi_env
|
||||
env['REQUEST_METHOD'] = 'GET'
|
||||
message, status = process_pin_reset_requests(env=env, phone_number='070000000')
|
||||
assert message == 'No user matching 070000000 was found.'
|
||||
assert status == '404 Not Found'
|
||||
|
||||
env['REQUEST_METHOD'] = 'GET'
|
||||
message, status = process_pin_reset_requests(env=env, phone_number=create_pin_blocked_user.phone_number)
|
||||
assert message == '{"status": "LOCKED"}'
|
||||
assert status == '200 OK'
|
||||
|
||||
env['REQUEST_METHOD'] = 'GET'
|
||||
message, status = process_pin_reset_requests(env=env, phone_number=create_pin_blocked_user.phone_number)
|
||||
assert message == '{"status": "LOCKED"}'
|
||||
assert status == '200 OK'
|
||||
|
||||
env['REQUEST_METHOD'] = 'PUT'
|
||||
message, status = process_pin_reset_requests(env=env, phone_number=create_pin_blocked_user.phone_number)
|
||||
assert message == f'Pin reset for user {create_pin_blocked_user.phone_number} is successful!'
|
||||
assert status == '200 OK'
|
||||
assert create_pin_blocked_user.get_account_status() == 'RESET'
|
||||
|
||||
|
||||
def test_process_locked_accounts_requests(create_locked_accounts, valid_locked_accounts_env):
|
||||
|
||||
response, message = process_locked_accounts_requests(env=valid_locked_accounts_env)
|
||||
|
||||
assert message == '200 OK'
|
||||
locked_account_addresses = json.loads(response)
|
||||
assert len(locked_account_addresses) == 10
|
||||
|
||||
# check that blockchain addresses are ordered by most recently accessed
|
||||
user_1 = User.session.query(User).filter_by(blockchain_address=locked_account_addresses[2]).first()
|
||||
user_2 = User.session.query(User).filter_by(blockchain_address=locked_account_addresses[7]).first()
|
||||
|
||||
assert user_1.updated > user_2.updated
|
||||
|
||||
35
apps/cic-ussd/tests/cic_ussd/test_transactions.py
Normal file
35
apps/cic-ussd/tests/cic_ussd/test_transactions.py
Normal file
@@ -0,0 +1,35 @@
|
||||
# standard imports
|
||||
|
||||
# third-party imports
|
||||
import pytest
|
||||
|
||||
# local imports
|
||||
from cic_ussd.chain import Chain
|
||||
from cic_ussd.transactions import OutgoingTransactionProcessor, truncate
|
||||
|
||||
|
||||
def test_outgoing_transaction_processor(load_config,
|
||||
create_valid_tx_recipient,
|
||||
create_valid_tx_sender,
|
||||
mock_outgoing_transactions):
|
||||
chain_str = Chain.spec.__str__()
|
||||
outgoing_tx_processor = OutgoingTransactionProcessor(
|
||||
chain_str=chain_str,
|
||||
from_address=create_valid_tx_sender.blockchain_address,
|
||||
to_address=create_valid_tx_recipient.blockchain_address
|
||||
)
|
||||
|
||||
outgoing_tx_processor.process_outgoing_transfer_transaction(
|
||||
amount=120,
|
||||
token_symbol='SRF'
|
||||
)
|
||||
assert mock_outgoing_transactions[0].get('amount') == 120.0
|
||||
assert mock_outgoing_transactions[0].get('token_symbol') == 'SRF'
|
||||
|
||||
|
||||
@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
|
||||
15
apps/cic-ussd/tests/cic_ussd/test_translation.py
Normal file
15
apps/cic-ussd/tests/cic_ussd/test_translation.py
Normal file
@@ -0,0 +1,15 @@
|
||||
# local imports
|
||||
from cic_ussd.translation import translation_for
|
||||
|
||||
|
||||
def test_translation_for(set_locale_files):
|
||||
english_translation = translation_for(
|
||||
key='ussd.kenya.exit_invalid_request',
|
||||
preferred_language='en'
|
||||
)
|
||||
swahili_translation = translation_for(
|
||||
key='ussd.kenya.exit_invalid_request',
|
||||
preferred_language='sw'
|
||||
)
|
||||
assert swahili_translation == 'END Chaguo si sahihi.'
|
||||
assert english_translation == 'END Invalid request.'
|
||||
53
apps/cic-ussd/tests/cic_ussd/test_validator.py
Normal file
53
apps/cic-ussd/tests/cic_ussd/test_validator.py
Normal file
@@ -0,0 +1,53 @@
|
||||
# third party imports
|
||||
import pytest
|
||||
|
||||
# local imports
|
||||
from cic_ussd.validator import (check_ip,
|
||||
check_request_content_length,
|
||||
check_service_code,
|
||||
check_known_user,
|
||||
check_request_method,
|
||||
validate_phone_number,
|
||||
validate_response_type)
|
||||
|
||||
|
||||
def test_check_ip(load_config, uwsgi_env):
|
||||
assert check_ip(config=load_config, env=uwsgi_env) is True
|
||||
|
||||
|
||||
def test_check_request_content_length(load_config, uwsgi_env):
|
||||
assert check_request_content_length(config=load_config, env=uwsgi_env) is True
|
||||
|
||||
|
||||
def test_check_service_code(load_config):
|
||||
assert check_service_code(code='*483*46#', config=load_config) is True
|
||||
|
||||
|
||||
def test_check_known_user(create_pending_user):
|
||||
user = create_pending_user
|
||||
assert check_known_user(phone=user.phone_number) is True
|
||||
|
||||
|
||||
def test_check_request_method(uwsgi_env):
|
||||
assert check_request_method(env=uwsgi_env) is True
|
||||
|
||||
|
||||
@pytest.mark.parametrize('phone, expected_value', [
|
||||
('653', False),
|
||||
('+654', False),
|
||||
('+254112233445', True),
|
||||
('+254712345678', True)
|
||||
])
|
||||
def test_validate_phone_number(phone, expected_value):
|
||||
assert validate_phone_number(phone=phone) is expected_value
|
||||
|
||||
|
||||
@pytest.mark.parametrize('response, expected_value', [
|
||||
('CON some random text', True),
|
||||
('END some more random tests', True),
|
||||
('Testing', False),
|
||||
('BIO testing', False)
|
||||
])
|
||||
def test_validate_response_type(response, expected_value):
|
||||
assert validate_response_type(response) is expected_value
|
||||
|
||||
Reference in New Issue
Block a user