def test_append_over_kinesis_record_size():
acc = RecordAccumulator(RawBuffer, CONFIG)
success = acc.try_append(b'-' * (1024 * 1024))
assert not success
acc.flush()
success = acc.try_append(b'-')
assert success
def test_append_timeout():
acc = RecordAccumulator(RawBuffer, CONFIG)
acc.try_append(b'-')
time.sleep(0.2)
assert acc.is_ready()
acc.flush()
assert not acc.is_ready()
def test_append():
acc = RecordAccumulator(RawBuffer, CONFIG)
success = acc.try_append(b'-')
assert success
acc.flush()
success = acc.try_append(b'-')
assert success
def test_flush():
acc = RecordAccumulator(RawBuffer, CONFIG)
acc.try_append(b'123')
acc.try_append(b'456')
acc.try_append(b'789')
assert acc.flush() == b'123X456X789X'
acc.try_append(b'ABC')
assert acc.flush() == b'ABCX'
def test_has_record():
acc = RecordAccumulator(RawBuffer, CONFIG)
assert not acc.has_records()
acc.try_append(b'-')
assert acc.has_records()
acc.flush()
assert not acc.has_records()
acc.try_append(b'-')
assert acc.has_records()