示例#1
0
def testBulkreader(tmpdir):
    q = PriorityEnqueue(str(tmpdir))
    q.queue([dict(id=i, p='L', expected=3) for i in xrange(2000)])
    q.close()
    # count how many files it created
    fns = tmpdir.join('3').listdir()
    nfiles = len(fns)

    dq = PriorityDequeue(str(tmpdir))
    count = 0
    while nfiles > 0:
        reader = dq.bulkreader()
        assert reader is not None
        for curi in reader:
            count += 1
        nfiles -= 1
    assert count == 2000
    reader = dq.bulkreader()
    assert reader is None