def test_cache_group_key_one_group(self):
@memoize(cache_group_keys=([0], ))
def my_func(*args, **kwargs):
return None
@memoize(cache_group_keys=([0], ))
def my_func2(*args, **kwargs):
return None
my_func('key')
my_func2('key')
keys = test_sentinel.master.keys()
assert len(keys) == 3
assert get_cache_group_key('key') in keys
delete_cache_group('key')
assert not test_sentinel.master.keys()
def test_cache_group_key_two_groups_one_key(self):
@memoize(cache_group_keys=([0], [1]))
def my_func(*args, **kwargs):
return None
my_func('key1', 'key2')
keys = test_sentinel.master.keys()
assert len(keys) == 3
assert get_cache_group_key('key1') in keys
assert get_cache_group_key('key2') in keys
delete_cache_group('key1')
keys = test_sentinel.master.keys()
assert len(keys) == 1
assert get_cache_group_key('key1') not in keys
assert get_cache_group_key('key2') in keys
delete_cache_group('key2')
assert not test_sentinel.master.keys()
def clean_project(project_id, category=None):
"""Clean cache for a specific project"""
project = db.session.query(Project).get(project_id)
delete_cache_group(project_id)
if project:
delete_cache_group(project.category.short_name)
delete_cache_group('get_all_draft')
def test_delete_cache_group_no_group(self):
assert not test_sentinel.master.keys()
delete_cache_group('key')
assert not test_sentinel.master.keys()
