def test_node_info_cache(get_cluster_info):
from django_elastipymemcache.backend import ElastiPymemcache
servers = ['h1:0', 'h2:0']
get_cluster_info.return_value = {'nodes': servers}
backend = ElastiPymemcache('h:0', {})
backend._lib.Client = Mock()
backend.set('key1', 'val')
backend.get('key1')
backend.set('key2', 'val')
backend.get('key2')
backend._lib.Client.assert_called_once_with(
servers,
ignore_exc=True,
)
eq_(backend._cache.get.call_count, 2)
eq_(backend._cache.set.call_count, 2)
get_cluster_info.assert_called_once()
def test_invalidate_cache(get_cluster_info):
from django_elastipymemcache.backend import ElastiPymemcache
servers = ['h1:0', 'h2:0']
get_cluster_info.return_value = {'nodes': servers}
backend = ElastiPymemcache('h:0', {})
backend._lib.Client = Mock()
assert backend._cache
backend._cache.get = Mock()
backend._cache.get.side_effect = Exception()
try:
backend.get('key1', 'val')
except Exception:
pass
# invalidate cached client
container = getattr(backend, '_local', backend)
container._client = None
try:
backend.get('key1', 'val')
except Exception:
pass
eq_(backend._cache.get.call_count, 2)
eq_(get_cluster_info.call_count, 3)