from hbd.common.proto import dicttos, stodict, oldmtodict


def test_dicttos_and_stodict_roundtrip():
    msg = dicttos("HTB", {"name": "host.example", "interval": 10})
    assert msg.startswith(b"!HTB:")
    d = stodict(msg)
    assert d["ID"] == "HTB"
    assert d["name"] == "host.example"
    assert d["interval"] == 10


def test_dicttos_and_stodict_ack():
    d = stodict(dicttos("ACK", {"time": 12345}))
    assert d["ID"] == "ACK"
    assert d["time"] == 12345


def test_oldmtodict():
    msg = b"name=foo;interval=5"
    d = oldmtodict(msg)
    assert d["ID"].startswith("HTB")
    assert d["name"] == "foo"
    assert d["interval"] == 5