Updates test for new wait and count options

This commit is contained in:
Ilan Steemers
2015-10-02 20:03:38 +02:00
parent 7911419642
commit 9243731a9c
2 changed files with 15 additions and 3 deletions
+8 -3
View File
@@ -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
+7
View File
@@ -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))