Пример #1
0
def test_concurrent_producers_single_consumer():
    num_threads = 100
    num_items = 10
    q = Queue()

    def producer(i):
        @gen.coroutine
        def run():
            for j in range(num_items):
                yield q.put((i, j))

        IOLoop(make_current=False).run_sync(run)

    producers = [
        threading.Thread(target=producer, args=(i, ))
        for i in range(num_threads)
    ]

    for p in producers:
        p.start()

    items = defaultdict(lambda: [])
    for x in range(num_items * num_threads):
        i, j = yield q.get()
        items[i].append(j)

    for v in items.values():
        assert v == list(range(num_items))

    for p in producers:
        p.join()
Пример #2
0
def test_concurrent_producers_single_consumer():
    num_threads = 100
    num_items = 10
    q = Queue()

    def producer(i):

        @gen.coroutine
        def run():
            for j in range(num_items):
                yield q.put((i, j))

        IOLoop(make_current=False).run_sync(run)

    producers = [
        threading.Thread(target=producer, args=(i,))
        for i in range(num_threads)
    ]

    for p in producers:
        p.start()

    items = defaultdict(lambda: [])
    for x in range(num_items * num_threads):
        i, j = yield q.get()
        items[i].append(j)

    for v in items.values():
        assert v == list(range(num_items))

    for p in producers:
        p.join()
Пример #3
0
def test_get_then_put(items):
    queue = Queue()

    got_futures = []
    for i in range(len(items)):
        got_futures.append(queue.get())

    for item in items:
        yield queue.put(item)

    got = yield got_futures
    assert got == items
Пример #4
0
def test_get_then_put(items):
    queue = Queue()

    got_futures = []
    for i in range(len(items)):
        got_futures.append(queue.get())

    for item in items:
        yield queue.put(item)

    got = yield got_futures
    assert got == items
Пример #5
0
def test_put_then_get_nowait(items):
    queue = Queue()

    for item in items:
        yield queue.put(item)

    got = []
    for i in range(len(items)):
        got.append(queue.get_nowait())

    assert got == items

    with pytest.raises(QueueEmpty):
        queue.get_nowait()

    future = queue.get()
    yield queue.put(42)
    assert 42 == (yield future)
Пример #6
0
def test_put_then_get_nowait(items):
    queue = Queue()

    for item in items:
        yield queue.put(item)

    got = []
    for i in range(len(items)):
        got.append(queue.get_nowait())

    assert got == items

    with pytest.raises(QueueEmpty):
        queue.get_nowait()

    future = queue.get()
    yield queue.put(42)
    assert 42 == (yield future)