Resets broker configuration in case a third party broker crashes

This commit is contained in:
Ilan Steemers
2015-09-08 15:50:32 +02:00
parent c946a16b99
commit 8eb1431413
3 changed files with 5 additions and 7 deletions
+2 -2
View File
@@ -155,8 +155,8 @@ def get_broker(list_key=Conf.PREFIX):
from brokers import disque from brokers import disque
return disque.Disque(list_key=list_key) return disque.Disque(list_key=list_key)
elif Conf.IRON_MQ: elif Conf.IRON_MQ:
from brokers import ironmq_broker from brokers import ironmq
return ironmq_broker.IronMQBroker(list_key=list_key) return ironmq.IronMQBroker(list_key=list_key)
# default to redis # default to redis
else: else:
from brokers import redis_broker from brokers import redis_broker
+3 -5
View File
@@ -18,11 +18,6 @@ from django_q.status import Stat
from django_q.brokers import get_broker from django_q.brokers import get_broker
from .tasks import multiply from .tasks import multiply
Conf.DISQUE_NODES = None
Conf.IRON_MQ = None
Conf.DJANGO_REDIS = 'default'
class WordClass(object): class WordClass(object):
def __init__(self): def __init__(self):
self.word_list = DEFAULT_WORDLIST self.word_list = DEFAULT_WORDLIST
@@ -33,6 +28,9 @@ class WordClass(object):
@pytest.fixture @pytest.fixture
def broker(): def broker():
Conf.DISQUE_NODES = None
Conf.IRON_MQ = None
Conf.DJANGO_REDIS = 'default'
return get_broker() return get_broker()