jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-04-27 02:37:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Device change test improvements in Template (#120884)

Browse Source
This commit is contained in:
dougiteixeira 2024-07-05 09:47:31 -03:00 committed by GitHub
parent b9cfd4e8ce
commit 4289efb506
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 66 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

100
tests/components/template/test_init.py
View File

@ -1,4 +1,4 @@
"""The test for the Template sensor platform.""" """Test for Template helper."""
from datetime import timedelta from datetime import timedelta
from unittest.mock import patch from unittest.mock import patch
@ -271,13 +271,48 @@ async def async_yaml_patch_helper(hass, filename):
await hass.async_block_till_done() await hass.async_block_till_done()
@pytest.mark.parametrize(
(
"config_entry_options",
"config_user_input",
),
[
(
{
"name": "My template",
"state": "{{10}}",
"template_type": "sensor",
},
{
"state": "{{12}}",
},
),
(
{
"template_type": "binary_sensor",
"name": "My template",
"state": "{{1 == 1}}",
},
{
"state": "{{1 == 2}}",
},
),
],
)
async def test_change_device( async def test_change_device(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
config_entry_options: dict[str, str],
config_user_input: dict[str, str],
) -> None: ) -> None:
"""Test remove the device registry configuration entry when the device changes.""" """Test the link between the device and the config entry.
# Configure a device registry Test, for each platform, that the device was linked to the
config entry and the link was removed when the device is
changed in the integration options.
"""
# Configure devices registry
entry_device1 = MockConfigEntry() entry_device1 = MockConfigEntry()
entry_device1.add_to_hass(hass) entry_device1.add_to_hass(hass)
device1 = device_registry.async_get_or_create( device1 = device_registry.async_get_or_create(
@ -300,60 +335,57 @@ async def test_change_device(
device_id2 = device2.id device_id2 = device2.id
assert device_id2 is not None assert device_id2 is not None
# Setup the config entry (binary_sensor) # Setup the config entry
sensor_config_entry = MockConfigEntry( template_config_entry = MockConfigEntry(
data={}, data={},
domain=DOMAIN, domain=DOMAIN,
options={ options=config_entry_options | {"device_id": device_id1},
"template_type": "binary_sensor", title="Template",
"name": "Teste",
"state": "{{15}}",
"device_id": device_id1,
},
title="Binary sensor template",
) )
sensor_config_entry.add_to_hass(hass) template_config_entry.add_to_hass(hass)
assert await hass.config_entries.async_setup(sensor_config_entry.entry_id) assert await hass.config_entries.async_setup(template_config_entry.entry_id)
await hass.async_block_till_done() await hass.async_block_till_done()
# Confirm that the configuration entry has been added to the device 1 registry (current) # Confirm that the config entry has been added to the device 1 registry (current)
current_device = device_registry.async_get(device_id=device_id1) current_device = device_registry.async_get(device_id=device_id1)
assert sensor_config_entry.entry_id in current_device.config_entries assert template_config_entry.entry_id in current_device.config_entries
# Change configuration options to use device 2 and reload the integration # Change config options to use device 2 and reload the integration
result = await hass.config_entries.options.async_init(sensor_config_entry.entry_id) result = await hass.config_entries.options.async_init(
template_config_entry.entry_id
)
result = await hass.config_entries.options.async_configure( result = await hass.config_entries.options.async_configure(
result["flow_id"], result["flow_id"],
user_input={ user_input=config_user_input | {"device_id": device_id2},
"state": "{{15}}",
"device_id": device_id2,
},
) )
await hass.async_block_till_done() await hass.async_block_till_done()
# Confirm that the configuration entry has been removed from the device 1 registry (previous) # Confirm that the config entry has been removed from the device 1 registry
previous_device = device_registry.async_get(device_id=device_id1) previous_device = device_registry.async_get(device_id=device_id1)
assert sensor_config_entry.entry_id not in previous_device.config_entries assert template_config_entry.entry_id not in previous_device.config_entries
# Confirm that the configuration entry has been added to the device 2 registry (current) # Confirm that the config entry has been added to the device 2 registry (current)
current_device = device_registry.async_get(device_id=device_id2) current_device = device_registry.async_get(device_id=device_id2)
assert sensor_config_entry.entry_id in current_device.config_entries assert template_config_entry.entry_id in current_device.config_entries
result = await hass.config_entries.options.async_init(sensor_config_entry.entry_id) # Change the config options to remove the device and reload the integration
result = await hass.config_entries.options.async_init(
template_config_entry.entry_id
)
result = await hass.config_entries.options.async_configure( result = await hass.config_entries.options.async_configure(
result["flow_id"], result["flow_id"],
user_input={ user_input=config_user_input,
"state": "{{15}}",
},
) )
await hass.async_block_till_done() await hass.async_block_till_done()
# Confirm that the configuration entry has been removed from the device 2 registry (previous) # Confirm that the config entry has been removed from the device 2 registry
previous_device = device_registry.async_get(device_id=device_id2) previous_device = device_registry.async_get(device_id=device_id2)
assert sensor_config_entry.entry_id not in previous_device.config_entries assert template_config_entry.entry_id not in previous_device.config_entries
# Confirm that there is no device with the helper configuration entry # Confirm that there is no device with the helper config entry
assert ( assert (
dr.async_entries_for_config_entry(device_registry, sensor_config_entry.entry_id) dr.async_entries_for_config_entry(
device_registry, template_config_entry.entry_id
)
== [] == []
) )