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

Use the new device helpers in Integral (#120157)

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

88
tests/components/integration/test_init.py
View file

@ -121,3 +121,91 @@ 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(input_entry)
assert config_entry.entry_id in _get_device_config_entries(valid_entry)
async def test_device_cleaning(
hass: HomeAssistant,
device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry,
) -> None:
"""Test for source entity device for Integration."""
# 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 Integration
integration_config_entry = MockConfigEntry(
data={},
domain=DOMAIN,
options={
"method": "trapezoidal",
"name": "Integration",
"round": 1.0,
"source": "sensor.test_source",
"unit_prefix": "k",
"unit_time": "min",
"max_sub_interval": {"minutes": 1},
},
title="Integration",
)
integration_config_entry.add_to_hass(hass)
assert await hass.config_entries.async_setup(integration_config_entry.entry_id)
await hass.async_block_till_done()
# Confirm the link between the source entity device and the integration sensor
integration_entity = entity_registry.async_get("sensor.integration")
assert integration_entity is not None
assert integration_entity.device_id == source_entity.device_id
# Device entry incorrectly linked to Integration config entry
device_registry.async_get_or_create(
config_entry_id=integration_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=integration_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(
integration_config_entry.entry_id
)
assert len(devices_before_reload) == 3
# Config entry reload
await hass.config_entries.async_reload(integration_config_entry.entry_id)
await hass.async_block_till_done()
# Confirm the link between the source entity device and the integration sensor after reload
integration_entity = entity_registry.async_get("sensor.integration")
assert integration_entity is not None
assert integration_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(
integration_config_entry.entry_id
)
assert len(devices_after_reload) == 1