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
Exemple #2
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
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