def test_newtask_queue(redis_conn): newtask_queue = NewTaskQueue(redis_conn, 'test_newtask_queue') newtask_queue.clear() assert len(newtask_queue) == 0 newtask_queue.put('task_1') newtask_queue.put({'task': '任务2'}) newtask_queue.put({'task': '任务3'}, ['task_4']) assert len(newtask_queue) == 4 assert newtask_queue.get() == 'task_1' assert newtask_queue.get() == {'task': u'任务2'} assert newtask_queue.get() == {'task': u'任务3'} assert newtask_queue.get() == ['task_4'] assert len(newtask_queue) == 0
def newtask_queue(redis_conn): from fulmar.message_queue.redis_queue import NewTaskQueue newtask_queue = NewTaskQueue(redis_conn, 'test_newtask_queue') newtask_queue.clear() return newtask_queue