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

Use registry fixtures in tests (m-n) (#118291)

Browse source
This commit is contained in:
epenet 2024-05-28 13:40:45 +02:00 • committed by GitHub
parent 8837c50da7
commit ead0e797c1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
22 changed files with 185 additions and 166 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/components/mqtt/test_diagnostics.py
View file

@ -152,6 +152,7 @@ async def test_entry_diagnostics(
async def test_redact_diagnostics(
hass: HomeAssistant,
device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry,
hass_client: ClientSessionGenerator,
mqtt_mock_entry: MqttMockHAClientGenerator,
) -> None:
@ -266,9 +267,10 @@ async def test_redact_diagnostics(
}
# Disable the entity and remove the state
ent_registry = er.async_get(hass)
device_tracker_entry = er.async_entries_for_device(ent_registry, device_entry.id)[0]
ent_registry.async_update_entity(
device_tracker_entry = er.async_entries_for_device(
entity_registry, device_entry.id
)[0]
entity_registry.async_update_entity(
device_tracker_entry.entity_id, disabled_by=er.RegistryEntryDisabler.USER
)
hass.states.async_remove(device_tracker_entry.entity_id)