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