diff --git a/django_q/tests/test_brokers.py b/django_q/tests/test_brokers.py index c0d6b93..a95e17f 100644 --- a/django_q/tests/test_brokers.py +++ b/django_q/tests/test_brokers.py @@ -1,9 +1,11 @@ -from time import sleep -import pytest import os +from time import sleep + +import pytest import redis -from django_q.conf import Conf + from django_q.brokers import get_broker, Broker +from django_q.conf import Conf from django_q.humanhash import uuid @@ -47,6 +49,15 @@ def test_redis(): Conf.DJANGO_REDIS = 'default' +def test_custom(): + Conf.BROKER = 'brokers.redis_broker.Redis' + broker = get_broker() + assert broker.ping() is True + assert broker.info() is not None + assert broker.__class__.__name == 'Redis' + Conf.BROKER = None + + def test_disque(): Conf.DISQUE_NODES = ['127.0.0.1:7711'] # check broker @@ -132,15 +143,15 @@ def test_ironmq(): broker.acknowledge(task[0]) assert broker.dequeue() is None # Retry test - #Conf.RETRY = 1 - #broker.enqueue('test') - #assert broker.dequeue() is not None - #sleep(3) + # Conf.RETRY = 1 + # broker.enqueue('test') # assert broker.dequeue() is not None - #task = broker.dequeue()[0] - #assert len(task) > 0 - #broker.acknowledge(task[0]) - #sleep(3) + # sleep(3) + # assert broker.dequeue() is not None + # task = broker.dequeue()[0] + # assert len(task) > 0 + # broker.acknowledge(task[0]) + # sleep(3) # delete job task_id = broker.enqueue('test') broker.delete(task_id)