fork/django-q2
1
0
Fork 0
mirror of https://github.com/django-q2/django-q2.git synced 2026-10-04 23:08:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

Adds bulk option to disque broker

Browse Source 
adds tests
This commit is contained in:
Ilan Steemers
2015-09-08 14:06:38 +02:00
parent 83548dcfbb
commit d849774c86
2 changed files with 31 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
django_q/tests/test_brokers.py
+19 -2
View File
@@ -76,6 +76,14 @@ def test_disque():
# fail
task_id = broker.enqueue('test')
broker.fail(task_id)
# bulk test
for i in range(5):
broker.enqueue('test')
Conf.BULK = 5
for i in range(5):
task = broker.dequeue()
assert task is not None
broker.acknowledge(task[0])
# delete queue
broker.enqueue('test')
broker.enqueue('test')
@@ -89,8 +97,8 @@ def test_disque():
reason="requires IronMQ credentials")
def test_ironmq():
Conf.IRON_MQ = {'host': os.getenv('IRON_MQ_HOST'),
'token': os.getenv('IRON_MQ_TOKEN'),
'project_id': os.getenv('IRON_MQ_PROJECT_ID')}
'token': os.getenv('IRON_MQ_TOKEN'),
'project_id': os.getenv('IRON_MQ_PROJECT_ID')}
# check broker
broker = get_broker(list_key='djangoQ')
assert broker.ping() is True
@@ -122,11 +130,20 @@ def test_ironmq():
# fail
task_id = broker.enqueue('test')
broker.fail(task_id)
# bulk test
for i in range(5):
broker.enqueue('test')
Conf.BULK = 5
for i in range(5):
task = broker.dequeue()
assert task is not None
broker.acknowledge(task[0])
# delete queue
broker.enqueue('test')
broker.enqueue('test')
broker.purge_queue()
assert broker.queue_size() == 0
broker.delete_queue()
# back to django-redis
Conf.IRON_MQ = None
Conf.DJANGO_REDIS = 'default'