def test_memcache_wrapper(self):
try:
from django.core.cache.backends import memcached
from django.core.cache.backends import locmem
except InvalidCacheBackendError:
sys.stderr.write(
"\n* Memcache library is not installed. Skipping test.\n")
return
prev_cache_cls = memcached.CacheClass
memcached.CacheClass = locmem.CacheClass
prev_backend_module = sys.modules.pop("celery.backends.cache")
try:
from celery.backends.cache import cache, DjangoMemcacheWrapper
self.assertIsInstance(cache, DjangoMemcacheWrapper)
key = "cu.test_memcache_wrapper"
val = "The quick brown fox."
default = "The lazy dog."
self.assertEqual(cache.get(key, default=default), default)
cache.set(key, val)
self.assertEqual(pickle.loads(cache.get(key, default=default)),
val)
finally:
memcached.CacheClass = prev_cache_cls
sys.modules["celery.backends.cache"] = prev_backend_module
def test_memcache_wrapper(self):
from django.core.cache.backends import memcached
from django.core.cache.backends import locmem
prev_cache_cls = memcached.CacheClass
memcached.CacheClass = locmem.CacheClass
prev_backend_module = sys.modules.pop("celery.backends.cache")
try:
from celery.backends.cache import cache, DjangoMemcacheWrapper
self.assertTrue(isinstance(cache, DjangoMemcacheWrapper))
key = "cu.test_memcache_wrapper"
val = "The quick brown fox."
default = "The lazy dog."
self.assertEquals(cache.get(key, default=default), default)
cache.set(key, val)
self.assertEquals(pickle.loads(cache.get(key, default=default)),
val)
finally:
memcached.CacheClass = prev_cache_cls
sys.modules["celery.backends.cache"] = prev_backend_module