def test_get_templates_by_ids_successful(notify_db, notify_db_session): template_1 = create_sample_template(notify_db, notify_db_session, template_name='Sample Template 1', template_type="sms", content="Template content") template_2 = create_sample_template(notify_db, notify_db_session, template_name='Sample Template 2', template_type="sms", content="Template content") create_sample_template(notify_db, notify_db_session, template_name='Sample Template 3', template_type="email", content="Template content") sample_cache_dict = { str.encode(str(template_1.id)): str.encode('2'), str.encode(str(template_2.id)): str.encode('3') } cache = [[k, v] for k, v in sample_cache_dict.items()] templates = dao_get_templates_for_cache(cache) assert len(templates) == 2 assert [ (template_1.id, template_1.template_type, template_1.name, False, 2), (template_2.id, template_2.template_type, template_2.name, False, 3) ] == templates
def get_template_statistics_for_7_days(limit_days, service_id): cache_key = cache_key_for_service_template_counter(service_id) template_stats_by_id = redis_store.get_all_from_hash(cache_key) if not template_stats_by_id: stats = dao_get_template_usage(service_id, limit_days=limit_days) cache_values = dict([(x.template_id, x.count) for x in stats]) if cache_values: redis_store.set_hash_and_expire( cache_key, cache_values, current_app.config['EXPIRE_CACHE_TEN_MINUTES']) else: stats = dao_get_templates_for_cache(template_stats_by_id.items()) return stats
def test_get_letter_templates_by_ids_successful(notify_db, notify_db_session): template_1 = create_sample_template( notify_db, notify_db_session, template_name=PRECOMPILED_TEMPLATE_NAME, template_type="letter", content="Template content", hidden=True ) template_2 = create_sample_template( notify_db, notify_db_session, template_name='Sample Template 2', template_type="letter", content="Template content" ) sample_cache_dict = {str.encode(str(template_1.id)): str.encode('2'), str.encode(str(template_2.id)): str.encode('3')} cache = [[k, v] for k, v in sample_cache_dict.items()] templates = dao_get_templates_for_cache(cache) assert len(templates) == 2 assert [(template_1.id, template_1.template_type, template_1.name, True, 2), (template_2.id, template_2.template_type, template_2.name, False, 3)] == templates
def test_get_templates_by_ids_returns_empty_list(): assert dao_get_templates_for_cache({}) == [] assert dao_get_templates_for_cache(None) == []