evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use the new device helpers in Threshold (#120158)

Browse source
This commit is contained in:
dougiteixeira 2024-06-22 10:38:58 -03:00 • committed by GitHub
parent 0b5c533669
commit 61931d15cd
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 103 additions and 38 deletions
Download patch file Download diff file Expand all files Collapse all files

86
tests/components/threshold/test_init.py
View file

@ -122,3 +122,89 @@ async def test_entry_changed(hass: HomeAssistant, platform) -> None:
# Check that the config entry association has updated
assert config_entry.entry_id not in _get_device_config_entries(run1_entry)
assert config_entry.entry_id in _get_device_config_entries(run2_entry)
async def test_device_cleaning(
hass: HomeAssistant,
device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry,
) -> None:
"""Test for source entity device for Threshold."""
# Source entity device config entry
source_config_entry = MockConfigEntry()
source_config_entry.add_to_hass(hass)
# Device entry of the source entity
source_device1_entry = device_registry.async_get_or_create(
config_entry_id=source_config_entry.entry_id,
identifiers={("sensor", "identifier_test1")},
connections={("mac", "30:31:32:33:34:01")},
)
# Source entity registry
source_entity = entity_registry.async_get_or_create(
"sensor",
"test",
"source",
config_entry=source_config_entry,
device_id=source_device1_entry.id,
)
await hass.async_block_till_done()
assert entity_registry.async_get("sensor.test_source") is not None
# Configure the configuration entry for Threshold
threshold_config_entry = MockConfigEntry(
data={},
domain=DOMAIN,
options={
"entity_id": "sensor.test_source",
"hysteresis": 0.0,
"lower": -2.0,
"name": "Threshold",
"upper": None,
},
title="Threshold",
)
threshold_config_entry.add_to_hass(hass)
assert await hass.config_entries.async_setup(threshold_config_entry.entry_id)
await hass.async_block_till_done()
# Confirm the link between the source entity device and the threshold sensor
threshold_entity = entity_registry.async_get("binary_sensor.threshold")
assert threshold_entity is not None
assert threshold_entity.device_id == source_entity.device_id
# Device entry incorrectly linked to Threshold config entry
device_registry.async_get_or_create(
config_entry_id=threshold_config_entry.entry_id,
identifiers={("sensor", "identifier_test2")},
connections={("mac", "30:31:32:33:34:02")},
)
device_registry.async_get_or_create(
config_entry_id=threshold_config_entry.entry_id,
identifiers={("sensor", "identifier_test3")},
connections={("mac", "30:31:32:33:34:03")},
)
await hass.async_block_till_done()
# Before reloading the config entry, two devices are expected to be linked
devices_before_reload = device_registry.devices.get_devices_for_config_entry_id(
threshold_config_entry.entry_id
)
assert len(devices_before_reload) == 3
# Config entry reload
await hass.config_entries.async_reload(threshold_config_entry.entry_id)
await hass.async_block_till_done()
# Confirm the link between the source entity device and the threshold sensor after reload
threshold_entity = entity_registry.async_get("binary_sensor.threshold")
assert threshold_entity is not None
assert threshold_entity.device_id == source_entity.device_id
# After reloading the config entry, only one linked device is expected
devices_after_reload = device_registry.devices.get_devices_for_config_entry_id(
threshold_config_entry.entry_id
)
assert len(devices_after_reload) == 1