def test_capped_stream_short_reads():
stream = util.UnbufferedStream([b"1" * 123, b"2" * 456])
capped_stream = web.CappedStream(stream, 1024)
assert capped_stream.read(1024) == b"1" * 123
assert capped_stream.read(1024) == b"2" * 456
assert capped_stream.read(1024) == b""
def test_capped_stream_read_size():
stream = io.BytesIO(b"x" * 1024)
capped_stream = web.CappedStream(stream, 768)
assert capped_stream.read(123) == b"x" * 123
def test_capped_stream_buffer_size():
stream = io.BytesIO(b"x" * 8192)
buffer_size = 4096
capped_stream = web.CappedStream(stream, 5120, buffer_size=buffer_size)
chunks = list(capped_stream)
assert chunks == [b"x" * buffer_size, b"x" * 1024]
def test_capped_stream_read_default():
stream = io.BytesIO(b"x" * 8192)
buffer_size = 4096
capped_stream = web.CappedStream(stream, 5120, buffer_size=buffer_size)
assert capped_stream.read() == b"x" * buffer_size
assert capped_stream.read() == b"x" * 1024
def test_capped_stream_iter():
stream = io.BytesIO(b"x" * 148 * 1024)
max_bytes = 138 * 1024
capped_stream = web.CappedStream(stream, max_bytes, buffer_size=128 * 1024)
data = b"".join(capped_stream)
assert data == b"x" * max_bytes