Fix imgw_pib tests (#135913)

pull/135915/head
Josef Zweck 2025-01-18 13:07:28 +01:00 committed by GitHub
parent 81b7d01a7d
commit f878465a9a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 4 deletions

View File

@ -55,6 +55,7 @@ async def test_remove_binary_sensor_entity(
) -> None:
"""Test removing a binary_sensor entity."""
entity_id = "binary_sensor.river_name_station_name_flood_alarm"
await init_integration(hass, mock_config_entry)
entity_registry.async_get_or_create(
BINARY_SENSOR_PLATFORM,
@ -64,6 +65,4 @@ async def test_remove_binary_sensor_entity(
config_entry=mock_config_entry,
)
await init_integration(hass, mock_config_entry)
assert hass.states.get(entity_id) is None

View File

@ -76,6 +76,7 @@ async def test_remove_entity(
) -> None:
"""Test removing entity."""
entity_id = "sensor.river_name_station_name_flood_alarm_level"
await init_integration(hass, mock_config_entry)
entity_registry.async_get_or_create(
SENSOR_PLATFORM,
@ -85,6 +86,4 @@ async def test_remove_entity(
config_entry=mock_config_entry,
)
await init_integration(hass, mock_config_entry)
assert hass.states.get(entity_id) is None