From f878465a9aacff0af8248e225727891e7c184669 Mon Sep 17 00:00:00 2001 From: Josef Zweck Date: Sat, 18 Jan 2025 13:07:28 +0100 Subject: [PATCH] Fix imgw_pib tests (#135913) --- tests/components/imgw_pib/test_init.py | 3 +-- tests/components/imgw_pib/test_sensor.py | 3 +-- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/tests/components/imgw_pib/test_init.py b/tests/components/imgw_pib/test_init.py index 834fd505497..6ee3528422d 100644 --- a/tests/components/imgw_pib/test_init.py +++ b/tests/components/imgw_pib/test_init.py @@ -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 diff --git a/tests/components/imgw_pib/test_sensor.py b/tests/components/imgw_pib/test_sensor.py index c38c8e6773e..aebe4d2f86e 100644 --- a/tests/components/imgw_pib/test_sensor.py +++ b/tests/components/imgw_pib/test_sensor.py @@ -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