def test_InMemStream():
stream = InMemStream()
yield stream.write("1")
yield stream.write("2")
buf = yield stream.read()
assert buf == "12"
yield stream.write("3")
buf = yield stream.read()
assert buf == "3"
# check internal stream buffer.
assert len(stream._stream) == 0
stream.close()
with pytest.raises(StreamingError):
yield stream.write("4")
def test_error_during_stream(io_loop):
stream = InMemStream()
try:
1 / 0
except Exception as e:
stream.set_exception(e)
with pytest.raises(ZeroDivisionError):
yield stream.read()
with pytest.raises(ZeroDivisionError):
yield stream.write("a")
