def test_buffer_heap():
"""RecordBuffer directly exposes its heap"""
record = local_record(now(), "1")
shard = new_shard()
buffer = RecordBuffer()
buffer.push(record, shard)
# [(sort, record, shard)]
assert buffer.heap[0][2] is shard
def test_push_all():
"""Bulk push is slightly more efficient"""
now_ = now()
records = [local_record(now_, str(i)) for i in reversed(range(100))]
shard = new_shard()
buffer = RecordBuffer()
pairs = [(record, shard) for record in records]
buffer.push_all(pairs)
same_records = [buffer.pop()[0] for _ in range(len(records))]
same_records.reverse()
assert records == same_records
def test_empty_buffer():
"""Trying to access an empty buffer raises IndexError"""
buffer = RecordBuffer()
assert not buffer
with pytest.raises(IndexError):
buffer.pop()
with pytest.raises(IndexError):
buffer.peek()
def test_clear():
record = local_record(now(), "1")
shard = new_shard()
buffer = RecordBuffer()
buffer.push(record, shard)
assert buffer
buffer.clear()
assert not buffer
def test_sort_every_push():
"""Push high to low, retrieve low to high"""
now_ = now()
records = [local_record(now_, str(i)) for i in reversed(range(15))]
shard = new_shard()
buffer = RecordBuffer()
for record in records:
buffer.push(record, shard)
# inserting high to low, every record should be at the front
assert buffer.peek()[0] is record
same_records = [buffer.pop()[0] for _ in range(len(records))]
same_records.reverse()
assert records == same_records
def test_single_record():
"""Push a record, peek at it, then get the same thing back"""
record = local_record(now(), "1")
shard = new_shard()
buffer = RecordBuffer()
buffer.push(record, shard)
assert buffer
same_record, same_shard = buffer.peek()
also_same_record, also_same_shard = buffer.pop()
assert not buffer
assert record is same_record is also_same_record
assert shard is same_shard is also_same_shard