From 857f2e1d869302c4b8f871dcdedd514ed1350ded Mon Sep 17 00:00:00 2001
From: Joost Lekkerkerker <joostlek@outlook.com>
Date: Thu, 19 Oct 2023 11:33:31 +0200
Subject: [PATCH] Patch platform in Withings sensor test (#102155)

* Patch platform in Withings sensor test

* Fix feedback
---
 tests/components/withings/test_sensor.py | 20 ++++++++++----------
 1 file changed, 10 insertions(+), 10 deletions(-)

diff --git a/tests/components/withings/test_sensor.py b/tests/components/withings/test_sensor.py
index 8351598df69..2e5be3e74aa 100644
--- a/tests/components/withings/test_sensor.py
+++ b/tests/components/withings/test_sensor.py
@@ -1,6 +1,6 @@
 """Tests for the Withings component."""
 from datetime import timedelta
-from unittest.mock import AsyncMock
+from unittest.mock import AsyncMock, patch
 
 from aiowithings import MeasurementGroup
 from freezegun.api import FrozenDateTimeFactory
@@ -29,16 +29,16 @@ async def test_all_entities(
     polling_config_entry: MockConfigEntry,
 ) -> None:
     """Test all entities."""
-    await setup_integration(hass, polling_config_entry)
-    entity_registry = er.async_get(hass)
-    entities = er.async_entries_for_config_entry(
-        entity_registry, polling_config_entry.entry_id
-    )
+    with patch("homeassistant.components.withings.PLATFORMS", [Platform.SENSOR]):
+        await setup_integration(hass, polling_config_entry)
+        entity_registry = er.async_get(hass)
+        entity_entries = er.async_entries_for_config_entry(
+            entity_registry, polling_config_entry.entry_id
+        )
 
-    for entity in entities:
-        if entity.domain == Platform.SENSOR:
-            assert hass.states.get(entity.entity_id) == snapshot
-    assert entities
+        assert entity_entries
+        for entity_entry in entity_entries:
+            assert hass.states.get(entity_entry.entity_id) == snapshot
 
 
 async def test_update_failed(