mirror of
https://github.com/django-q2/django-q2.git
synced 2026-10-01 01:08:11 +08:00
Ilan Steemers
d6b9b10e3e
* test for redis and django-redis * test for disque TODO add disque to Travis
62 lines
1.6 KiB
Python
62 lines
1.6 KiB
Python
from time import sleep
|
|
import pytest
|
|
from django_q.conf import Conf
|
|
from django_q.brokers import get_broker, Broker
|
|
|
|
|
|
def test_broker():
|
|
broker = Broker()
|
|
broker.enqueue('test')
|
|
broker.dequeue()
|
|
broker.queue_size()
|
|
broker.delete_queue()
|
|
broker.acknowledge('test')
|
|
broker.ping()
|
|
assert broker.get_stat('test_1') is None
|
|
broker.set_stat('test_1', 'test', 3)
|
|
assert broker.get_stat('test_1') == 'test'
|
|
assert broker.get_stats('test:*')[0] == 'test'
|
|
|
|
|
|
def test_redis():
|
|
Conf.DJANGO_REDIS = None
|
|
broker = get_broker()
|
|
assert broker.ping() is True
|
|
Conf.REDIS = {'host': '127.0.0.1', 'port': 7712}
|
|
broker = get_broker()
|
|
with pytest.raises(Exception):
|
|
broker.ping()
|
|
|
|
|
|
def test_disque():
|
|
Conf.DISQUE = ['127.0.0.1:7711']
|
|
broker = get_broker()
|
|
assert broker.ping() is True
|
|
broker.enqueue('test')
|
|
assert broker.queue_size() == 1
|
|
task = broker.dequeue()
|
|
assert task[1] == 'test'
|
|
broker.acknowledge(task[0])
|
|
assert broker.queue_size() == 0
|
|
# Retry test
|
|
Conf.RETRY = 1
|
|
broker.enqueue('test')
|
|
assert broker.queue_size() == 1
|
|
broker.dequeue()
|
|
assert broker.queue_size() == 0
|
|
sleep(1.5)
|
|
assert broker.queue_size() == 1
|
|
task = broker.dequeue()
|
|
assert broker.queue_size() == 0
|
|
broker.acknowledge(task[0])
|
|
sleep(1.5)
|
|
assert broker.queue_size() == 0
|
|
# errors
|
|
with pytest.raises(NotImplementedError):
|
|
broker.delete_queue()
|
|
Conf.DISQUE = ['127.0.0.1:7712', '127.0.0.1:7713']
|
|
with pytest.raises(ConnectionError):
|
|
broker.get_connection()
|
|
# back to djangoredis
|
|
Conf.DJANGO_REDIS = 'default'
|