From d8a4a97ee01de330b4c1ac33e1d150c54b89922c Mon Sep 17 00:00:00 2001
From: Martin Hjelmare <marhje52@gmail.com>
Date: Thu, 20 Mar 2025 12:19:14 +0100
Subject: [PATCH] Allow patching Z-Wave platforms specifically in tests
 (#140987)

---
 tests/components/zwave_js/conftest.py   | 14 ++++++++++++--
 tests/components/zwave_js/test_siren.py |  9 ++++++++-
 2 files changed, 20 insertions(+), 3 deletions(-)

diff --git a/tests/components/zwave_js/conftest.py b/tests/components/zwave_js/conftest.py
index 1917ebedd34..ce7b0e0109e 100644
--- a/tests/components/zwave_js/conftest.py
+++ b/tests/components/zwave_js/conftest.py
@@ -13,7 +13,9 @@ from zwave_js_server.model.node import Node
 from zwave_js_server.model.node.data_model import NodeDataType
 from zwave_js_server.version import VersionInfo
 
+from homeassistant.components.zwave_js import PLATFORMS
 from homeassistant.components.zwave_js.const import DOMAIN
+from homeassistant.const import Platform
 from homeassistant.core import HomeAssistant
 from homeassistant.util.json import JsonArrayType
 
@@ -828,18 +830,26 @@ def nortek_thermostat_removed_event_fixture(client) -> Node:
 async def integration_fixture(
     hass: HomeAssistant,
     client: MagicMock,
+    platforms: list[Platform],
 ) -> MockConfigEntry:
     """Set up the zwave_js integration."""
     entry = MockConfigEntry(domain="zwave_js", data={"url": "ws://test.org"})
     entry.add_to_hass(hass)
-    await hass.config_entries.async_setup(entry.entry_id)
-    await hass.async_block_till_done()
+    with patch("homeassistant.components.zwave_js.PLATFORMS", platforms):
+        await hass.config_entries.async_setup(entry.entry_id)
+        await hass.async_block_till_done()
 
     client.async_send_command.reset_mock()
 
     return entry
 
 
+@pytest.fixture
+def platforms() -> list[Platform]:
+    """Fixture to specify platforms to test."""
+    return PLATFORMS
+
+
 @pytest.fixture(name="chain_actuator_zws12")
 def window_cover_fixture(client, chain_actuator_zws12_state) -> Node:
     """Mock a window cover node."""
diff --git a/tests/components/zwave_js/test_siren.py b/tests/components/zwave_js/test_siren.py
index 4eb872954d1..d932338f9dc 100644
--- a/tests/components/zwave_js/test_siren.py
+++ b/tests/components/zwave_js/test_siren.py
@@ -1,5 +1,6 @@
 """Test the Z-Wave JS siren platform."""
 
+import pytest
 from zwave_js_server.event import Event
 
 from homeassistant.components.siren import (
@@ -7,7 +8,7 @@ from homeassistant.components.siren import (
     ATTR_TONE,
     ATTR_VOLUME_LEVEL,
 )
-from homeassistant.const import STATE_OFF, STATE_ON, STATE_UNKNOWN
+from homeassistant.const import STATE_OFF, STATE_ON, STATE_UNKNOWN, Platform
 from homeassistant.core import HomeAssistant
 
 SIREN_ENTITY = "siren.indoor_siren_6_play_tone_2"
@@ -64,6 +65,12 @@ TONE_ID_VALUE_ID = {
 }
 
 
+@pytest.fixture
+def platforms() -> list[str]:
+    """Fixture to specify platforms to test."""
+    return [Platform.SIREN]
+
+
 async def test_siren(
     hass: HomeAssistant, client, aeotec_zw164_siren, integration
 ) -> None: