Set an aggressive timeout on the redis connections to allow the unit tests to run when communication with redis is disabled.

This commit is contained in:
Jake Moshenko 2014-09-22 18:42:22 -04:00
parent 3259cda000
commit a6225ad34a
2 changed files with 3 additions and 3 deletions

View file

@ -17,7 +17,7 @@ class RedisBuildLogs(object):
PHASE = 'phase'
def __init__(self, redis_host):
self._redis = redis.StrictRedis(host=redis_host)
self._redis = redis.StrictRedis(host=redis_host, socket_timeout=5)
@staticmethod
def _logs_key(build_id):

View file

@ -44,7 +44,7 @@ class UserEvent(object):
as backed by Redis.
"""
def __init__(self, redis_host, username):
self._redis = redis.StrictRedis(host=redis_host)
self._redis = redis.StrictRedis(host=redis_host, socket_timeout=5)
self._username = username
@staticmethod
@ -77,7 +77,7 @@ class UserEventListener(object):
def __init__(self, redis_host, username, events=set([])):
channels = [self._user_event_key(username, e) for e in events]
self._redis = redis.StrictRedis(host=redis_host)
self._redis = redis.StrictRedis(host=redis_host, socket_timeout=5)
self._pubsub = self._redis.pubsub()
self._pubsub.subscribe(channels)