def mock_queue_client(monkeypatch):
# Stop mockredis from trying to run Lua imports that aren't really needed.
monkeypatch.setattr('mockredis.script.Script._import_lua_dependencies',
staticmethod(lambda *args, **kwargs: None))
cl = QueueClient(zone='testzone')
cl.redis = mockredis.MockRedis()
return cl
def get_redis_connection():
if settings.TEST:
import mockredis
if not getattr(get_redis_connection, 'redis_mock', None):
setattr(get_redis_connection, 'redis_mock', mockredis.MockRedis())
return get_redis_connection.redis_mock
return django_rq.get_connection()
def get_client():
global _client
if not _client:
if env.mode in [env.DISTRIBUTED, env.WORK, env.API]:
_config = dict(env.REDIS_CONFIG)
_config['decode_responses'] = True
_client = redis.Redis(**_config)
else:
# local or unittest
_client = mockredis.MockRedis(decode_responses=True)
return _client
def _setup(self):
"""
Instantiate the Redis interface. This called the first time Redis is
used.
"""
if settings.REDIS_URI is None:
import mockredis
self._wrapped = mockredis.MockRedis(strict=True)
else:
self._wrapped = redis.StrictRedis.from_url(settings.REDIS_URI,
decode_responses=True,
encoding='utf-8')
def db():
return mockredis.MockRedis(encoding="utf-8", decode_responses=True)
def set_up_redis():
return mockredis.MockRedis()
def mock_cache(monkeypatch):
clock = ChangableClock()
redis = mockredis.MockRedis(clock=clock)
monkeypatch.setattr(cache, "redis", redis)
return redis, clock