def test_multi_threaded_insertion():
global QUEUE_NAME
for qtype in QTYPES:
tq = TaskQueue(n_threads=40, queue_name=QUEUE_NAME, queue_server=qtype, qurl=QURL)
n_inserts = 100
tq.purge()
tq.wait()
if qtype != 'aws':
assert tq.enqueued == 0
for _ in range(n_inserts):
task = PrintTask()
tq.insert(task)
tq.wait()
if qtype == 'aws':
list_len = 10
else:
list_len = 100
lst = tq.list()
assert len(lst) == list_len # task list api only lists 100 items at a time
assert all(map(lambda x: type(x) == PrintTask, lst))
tq.purge()
if qtype != 'aws':
assert tq.enqueued == 0
def test_single_threaded_insertion():
global QUEUE_NAME
for qtype in QTYPES:
tq = TaskQueue(n_threads=0, queue_name=QUEUE_NAME, queue_server=qtype, qurl=QURL)
tq.purge()
if qtype != 'aws':
assert tq.enqueued == 0
n_inserts = 5
for _ in range(n_inserts):
task = PrintTask()
tq.insert(task)
tq.wait()
if qtype != 'aws':
assert tq.enqueued == n_inserts
assert all(map(lambda x: type(x) == PrintTask, tq.list()))
tq.purge()
if qtype != 'aws':
assert tq.enqueued == 0
def test_single_threaded_insertion(sqs, protocol):
path = getpath(protocol)
tq = TaskQueue(path, n_threads=0)
tq.purge()
n_inserts = 5
tq.insert((PrintTask() for i in range(n_inserts)))
assert all(map(lambda x: type(x) == PrintTask, tq.list()))
tq.purge()
def test_single_threaded_insertion_fns(sqs, protocol): path = getpath(protocol) tq = TaskQueue(path, n_threads=0) tq.purge() n_inserts = 5 tq.insert(( partial(printfn, "hello world " + str(i)) for i in range(n_inserts) )) assert all(map(lambda x: isinstance(x, FunctionTask), tq.list())) tq.purge()
