cic-internal-integration/apps/cic-ussd/tests/cic_ussd/test_cache.py

# standard imports
import hashlib
import json

# external imports
from cic_types.condiments import MetadataPointer

# local imports
from cic_ussd.cache import cache_data, cache_data_key, get_cached_data

# test imports


def test_cache_data(init_cache):
    identifier = 'some_key'.encode()
    key = cache_data_key(identifier, MetadataPointer.PERSON)
    assert get_cached_data(key) is None
    cache_data(key, json.dumps('some_value'))
    assert get_cached_data(key) is not None


def test_cache_data_key():
    identifier = 'some_key'.encode()
    key = cache_data_key(identifier, MetadataPointer.PERSON)
    hash_object = hashlib.new("sha256")
    hash_object.update(identifier)
    hash_object.update(':cic.person'.encode(encoding="utf-8"))
    assert hash_object.digest().hex() == key


def test_get_cached_data(cached_ussd_session):
    ussd_session = get_cached_data(cached_ussd_session.external_session_id)
    ussd_session = json.loads(ussd_session)
    assert ussd_session.get('msisdn') == cached_ussd_session.msisdn