Fix imgw_pib tests (#135913)
parent
81b7d01a7d
commit
f878465a9a
|
@ -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
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue