def test_peek_eof():
source = b'Hello, world!\n'
source_stream = io.BytesIO(source)
stream = BufferedReader(source_stream.read, len(source) - 1)
assert stream.peek(0) == b''
assert stream.peek(1) == b'H'
assert stream.peek(2) == b'He'
assert stream.peek(16) == b'Hello, world!'
assert stream.peek(32) == b'Hello, world!'
assert source_stream.read() == b'\n'
def test_read_until_shared_boundary(chunk_size):
source = b'-boundary-like-' * 4 + b'--some junk--\n' + b'\n' * 1024
source_stream = io.BytesIO(source)
stream = BufferedReader(source_stream.read, len(source), chunk_size)
assert stream.read_until(b'-boundary-like---') == b'-boundary-like-' * 3
assert stream.peek(17) == b'-boundary-like---'