From 9243731a9c8c91e2772d1792e535bd4a1f42bc2d Mon Sep 17 00:00:00 2001 From: Ilan Steemers Date: Fri, 2 Oct 2015 20:03:38 +0200 Subject: [PATCH] Updates test for new wait and count options --- django_q/tests/test_cached.py | 11 ++++++++--- django_q/tests/test_cluster.py | 7 +++++++ 2 files changed, 15 insertions(+), 3 deletions(-) diff --git a/django_q/tests/test_cached.py b/django_q/tests/test_cached.py index 9a9f879..7e8e618 100644 --- a/django_q/tests/test_cached.py +++ b/django_q/tests/test_cached.py @@ -33,8 +33,13 @@ def test_cached(broker): async('math.copysign', 1, -1, cached=True, broker=broker, group=group) async('math.popysign', 1, -1, cached=True, broker=broker, group=group) # test wait on cache - assert result(task_id, wait=1, cached=True) is None - assert fetch(task_id, wait=1, cached=True) is None + # test wait timeout + assert result(task_id, wait=10, cached=True) is None + assert fetch(task_id, wait=10, cached=True) is None + assert result_group(group, wait=10, cached=True) is None + assert result_group(group, count=2, wait=10, cached=True) is None + assert fetch_group(group, wait=10, cached=True) is None + assert fetch_group(group, count=2, wait=10, cached=True) is None # run a single cluster start_event = Event() stop_event = Event() @@ -42,7 +47,7 @@ def test_cached(broker): Sentinel(stop_event, start_event, broker=broker) # assert results assert result(task_id, wait=500, cached=True) == -1 - assert fetch(task_id,wait=500, cached=True).result == -1 + assert fetch(task_id, wait=500, cached=True).result == -1 # make sure it's not in the db backend assert fetch(task_id) is None # assert group diff --git a/django_q/tests/test_cluster.py b/django_q/tests/test_cluster.py index 37dc79a..5b1350e 100644 --- a/django_q/tests/test_cluster.py +++ b/django_q/tests/test_cluster.py @@ -160,6 +160,13 @@ def test_async(broker, admin_user): assert broker.queue_size() == 0 assert task_queue.qsize() == task_count task_queue.put('STOP') + # test wait timeout + assert result(j, wait=10) is None + assert fetch(j, wait=10) is None + assert result_group('test_j', wait=10) is None + assert result_group('test_j', count=2, wait=10) is None + assert fetch_group('test_j', wait=10) is None + assert fetch_group('test_j', count=2, wait=10) is None # let a worker handle them result_queue = Queue() worker(task_queue, result_queue, Value('f', -1))