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")