from hbd.udp import parse_message
from hbd.proto import dicttos


def test_parse_message_uncompressed():
    raw = dicttos("HTB", {"name": "host", "interval": 1})
    m = parse_message(raw)
    assert m["ID"].startswith("HTB")


def test_parse_message_compressed():
    raw = dicttos("ACK", {"time": 1}, compress=True)
    m = parse_message(raw)
    assert "ID" in m