srne-mqtt/tests/test_protocol.py

from io import BytesIO

from srnemqtt.protocol import write as protocol_write


def test_write():
    fh = BytesIO()
    protocol_write(fh, b"Hello, World!")
    fh.seek(0)

    assert fh.read() == b"Hello, World!\x4E\x11"