def test_outside_source_stream_discard():
stream = OutsideSourceReplayStream()
stream.set_header("header")
stream.feed_data(b"abcdefgh")
stream.discard(2)
with pytest.raises(IndexError):
stream.data.bytes()
assert stream.data[2:] == b"cdefgh"
assert stream.future_data[2:] == b"cdefgh"
assert stream.data[-2:-1] == b"g"
assert stream.future_data[-2:-1] == b"g"
assert stream.data[3] == 100
assert stream.future_data[3] == 100
for bad_range in [1, slice(1, 5, 1), slice(-7, 5, 1), slice(5, 1, 1)]:
with pytest.raises(IndexError):
stream.data[bad_range]
with pytest.raises(IndexError):
stream.future_data[bad_range]
assert len(stream.data) == 8
assert len(stream.future_data) == 8
v = stream.data.view(2)
assert v == b"cdefgh"
v.release()
with pytest.raises(IndexError):
stream.data.view()
with pytest.raises(IndexError):
stream.data.view(1, 3)
async def test_outside_source_stream_immediate_data():
stream = OutsideSourceReplayStream()
f1 = asyncio.ensure_future(stream.wait_for_header())
f2 = asyncio.ensure_future(stream.wait_for_data(0))
stream.set_header("header")
stream.feed_data(b"Lorem")
assert await f1 == "header"
assert await f2 == 5
assert stream.header == "header"
assert stream.data.bytes() == b"Lorem"
async def test_outside_source_stream_read(event_loop):
stream = OutsideSourceReplayStream()
f = asyncio.ensure_future(stream.wait_for_data(0))
stream.set_header("header")
await exhaust_callbacks(event_loop)
assert not f.done()
stream.feed_data(b"Lorem")
assert await f == 5
assert stream.data.bytes() == b"Lorem"
stream.finish()
await stream.wait_for_ended()
async def test_outside_source_stream_wait_until_position(event_loop):
stream = OutsideSourceReplayStream()
f = asyncio.ensure_future(stream.wait_for_data(3))
stream.set_header("header")
stream.feed_data(b"a")
exhaust_callbacks(event_loop)
assert not f.done()
stream.feed_data(b"aa")
exhaust_callbacks(event_loop)
assert not f.done()
stream.feed_data(b"ccc")
assert await f == 3
assert stream.data[3:6] == b"ccc"