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 test_cache_group_key_callable(self):
def cache_group_key_fn(*args, **kwargs):
return args[0]
@memoize(cache_group_keys=(cache_group_key_fn, ))
def my_func(*args, **kwargs):
return None
my_func('a')
assert get_cache_group_key('a') in test_sentinel.master.keys()
def management_dashboard_stats_cached():
stats_cached = all([
bool(list(sentinel.slave.smembers(get_cache_group_key((ms)))))
for ms in management_dashboard_stats
])
if not stats_cached:
# reset stats for any missing stat
map(delete_cache_group, management_dashboard_stats)
return stats_cached
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()