def test_buffer_frame_too_large() -> None:
buffer_ = WebsocketBuffer(2)
with pytest.raises(FrameTooLarge):
buffer_.extend(
TextMessage(data="abc",
frame_finished=False,
message_finished=True))
def test_buffer() -> None:
buffer_ = WebsocketBuffer(10)
buffer_.extend(TextMessage(data="abc", frame_finished=False, message_finished=True))
assert buffer_.to_message() == {"type": "websocket.receive", "bytes": None, "text": "abc"}
buffer_.clear()
buffer_.extend(BytesMessage(data=b"abc", frame_finished=False, message_finished=True))
assert buffer_.to_message() == {"type": "websocket.receive", "bytes": b"abc", "text": None}
def test_buffer_mixed_types(data: list) -> None:
buffer_ = WebsocketBuffer(10)
buffer_.extend(data[0])
with pytest.raises(TypeError):
buffer_.extend(data[1])