rt/haco
1
0
Fork 0
mirror of https://github.com/fmtr/haco.git synced 2026-10-25 18:40:15 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add unit test to ensure device announce data included in control announce

Browse source
This commit is contained in:
Edward Brown 2026-05-17 13:42:50 +01:00
parent 57a5f5c329
commit aacb79ab61
1 changed files with 29 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

30
haco/tests/test_pydantic_models.py
View file

@ -31,6 +31,10 @@ def test_switch_announce_payload_matches_expected(client_stub):
"icon": "mdi:power",
"unique_id": "dev-main",
"availability_topic": "root/status",
"device": {
"name": "Dev",
"identifiers": ["dev"],
},
"state_topic": "root/client/dev/main/default/state",
"command_topic": "root/client/dev/main/default/command",
}
@ -59,10 +63,34 @@ def test_climate_announce_payload_has_expected_ha_keys(client_stub):
"power_state_topic",
}
assert expected_subset.issubset(set(data))
assert not any(key in data for key in {"capabilities", "subscriptions", "device", "parent"})
assert not any(key in data for key in {"capabilities", "subscriptions", "parent"})
assert not any(callable(value) for value in data.values())
def test_control_announce_includes_device_announce_data(client_stub):
switch = Switch(name="Main")
device = Device(
name="Dev",
manufacturer="ACME",
model="Model X",
sw_version="1.2.3",
controls=[switch],
)
device.set_parent(client_stub)
data = switch.get_announce()
assert data["device"] == {
"name": "Dev",
"manufacturer": "ACME",
"model": "Model X",
"sw_version": "1.2.3",
"identifiers": ["dev"],
}
assert "controls" not in data["device"]
assert "parent" not in data["device"]
assert "subscriptions" not in data["device"]
def test_model_dump_excludes_runtime_extra_attributes():
switch = Switch(name="Main")
switch.dynamic_callback = lambda value: value