def test_clearing_non_empty_queue():
queue = MultiQueue(['q1', 'q2'])
queue.put(1, 'q1')
queue.clear(['q2'])
assert_false(queue.empty(['q1']))
queue.clear(['q1'])
assert_true(queue.empty(['q1']))
def test_select_queue_to_get_from():
queue = MultiQueue(['q1', 'q2'])
queue.put(1, 'q1')
queue.put(2, 'q2')
value, from_queue = queue.get(['q2'])
assert_equal(value, 2)
assert_equal(from_queue, 'q2')
def test_select_queue_to_get_from():
queue = MultiQueue(['q1', 'q2'])
queue.put(1, 'q1')
queue.put(2, 'q2')
value, from_queue = queue.get(['q2'])
assert_equal(value, 2)
assert_equal(from_queue, 'q2')
def test_clearing_non_empty_queue():
queue = MultiQueue(['q1', 'q2'])
queue.put(1, 'q1')
queue.clear(['q2'])
assert_false(queue.empty(['q1']))
queue.clear(['q1'])
assert_true(queue.empty(['q1']))
def test_bidirectional_get():
queue = MultiQueue()
queue.put(1)
queue.put(2)
queue.put(3)
assert_equal(queue.get(), (1, 'default'))
assert_equal(queue.popleft(), (3, 'default'))
assert_equal(queue.pop(), (2, 'default'))
def test_noarg_multiqueue_works_like_standard_queue():
queue = MultiQueue()
queue.put(1)
queue.put(2)
value, _ = queue.get()
assert_equal(value, 1)
value, _ = queue.get()
assert_equal(value, 2)
def test_noarg_multiqueue_works_like_standard_queue():
queue = MultiQueue()
queue.put(1)
queue.put(2)
value, _ = queue.get()
assert_equal(value, 1)
value, _ = queue.get()
assert_equal(value, 2)
def test_bidirectional_get():
queue = MultiQueue()
queue.put(1)
queue.put(2)
queue.put(3)
assert_equal(queue.get(), (1, 'default'))
assert_equal(queue.popleft(), (3, 'default'))
assert_equal(queue.pop(), (2, 'default'))
def test_get_from_many_queues():
queue = MultiQueue(['q1', 'q2'])
queue.put(1, 'q1')
queue.put(2, 'q2')
value, from_queue = queue.get(['q1', 'q2'])
assert_equal(value, 1)
assert_equal(from_queue, 'q1')
value, from_queue = queue.get(['q1', 'q2'])
assert_equal(value, 2)
assert_equal(from_queue, 'q2')
def test_get_from_many_queues():
queue = MultiQueue(['q1', 'q2'])
queue.put(1, 'q1')
queue.put(2, 'q2')
value, from_queue = queue.get(['q1', 'q2'])
assert_equal(value, 1)
assert_equal(from_queue, 'q1')
value, from_queue = queue.get(['q1', 'q2'])
assert_equal(value, 2)
assert_equal(from_queue, 'q2')
def test_blocked_if_selected_queue_is_empty():
queue = MultiQueue(['q1', 'q2'])
output = []
def consumer():
output.append(queue.get(['q2']))
queue.put(1, 'q1')
tconsumer = threading.Thread(target=consumer)
tconsumer.start()
time.sleep(0.5)
assert_equal(len(output), 0)
queue.put(2, 'q2')
time.sleep(0.5)
assert_equal(len(output), 1)
value, from_queue = output[0]
assert_equal(value, 2)
assert_equal(from_queue, 'q2')
tconsumer.join()
def test_blocked_if_selected_queue_is_empty():
queue = MultiQueue(['q1', 'q2'])
output = []
def consumer():
output.append(queue.get(['q2']))
queue.put(1, 'q1')
tconsumer = threading.Thread(target=consumer)
tconsumer.start()
time.sleep(0.5)
assert_equal(len(output), 0)
queue.put(2, 'q2')
time.sleep(0.5)
assert_equal(len(output), 1)
value, from_queue = output[0]
assert_equal(value, 2)
assert_equal(from_queue, 'q2')
tconsumer.join()
def test_put_to_nonexistent_queue():
queue = MultiQueue(['q1'])
queue.put(1, 'q2')
def test_successful_nonblocking_get():
queue = MultiQueue()
queue.put(1)
assert_equal(queue.get(blocking=False), (1, 'default'))
def test_queue_non_emptiness():
queue = MultiQueue(['q1', 'q2'])
queue.put(1, 'q1')
assert_false(queue.empty(['q1']))
assert_true(queue.empty(['q2']))
assert_false(queue.empty(['q1', 'q2']))
def test_access_specific_queue():
queue = MultiQueue(['myqueue'])
queue.put(1, 'myqueue')
value, from_queue = queue.get(['myqueue'])
assert_equal(value, 1)
assert_equal(from_queue, 'myqueue')
def test_access_specific_queue():
queue = MultiQueue(['myqueue'])
queue.put(1, 'myqueue')
value, from_queue = queue.get(['myqueue'])
assert_equal(value, 1)
assert_equal(from_queue, 'myqueue')
def test_successful_nonblocking_get():
queue = MultiQueue()
queue.put(1)
assert_equal(queue.get(blocking=False), (1, 'default'))
def test_put_to_nonexistent_queue():
queue = MultiQueue(['q1'])
queue.put(1, 'q2')
def test_queue_non_emptiness():
queue = MultiQueue(['q1', 'q2'])
queue.put(1, 'q1')
assert_false(queue.empty(['q1']))
assert_true(queue.empty(['q2']))
assert_false(queue.empty(['q1', 'q2']))
