Add light unit tests for switchbot (#140436)

tests/components/switchbot/__init__.py

@@ -463,3 +463,28 @@ HUMIDIFIER_SERVICE_INFO = BluetoothServiceInfoBleak(
     connectable=True,
     tx_power=-127,
 )
+
+
+WOSTRIP_SERVICE_INFO = BluetoothServiceInfoBleak(
+    name="WoStrip",
+    address="AA:BB:CC:DD:EE:FF",
+    manufacturer_data={
+        2409: b'\x84\xf7\x03\xb3?\xde\x04\xe4"\x0c\x00\x00\x00\x00\x00\x00'
+    },
+    service_data={"00000d00-0000-1000-8000-00805f9b34fb": b"r\x00d"},
+    service_uuids=["cba20d00-224d-11e6-9fb8-0002a5d5c51b"],
+    rssi=-60,
+    source="local",
+    advertisement=generate_advertisement_data(
+        local_name="WoStrip",
+        manufacturer_data={
+            2409: b'\x84\xf7\x03\xb3?\xde\x04\xe4"\x0c\x00\x00\x00\x00\x00\x00'
+        },
+        service_data={"00000d00-0000-1000-8000-00805f9b34fb": b"r\x00d"},
+        service_uuids=["cba20d00-224d-11e6-9fb8-0002a5d5c51b"],
+    ),
+    device=generate_ble_device("AA:BB:CC:DD:EE:FF", "WoStrip"),
+    time=0,
+    connectable=True,
+    tx_power=-127,
+)

tests/components/switchbot/test_light.py

@@ -0,0 +1,139 @@
+"""Test the switchbot lights."""
+
+from collections.abc import Callable
+from typing import Any
+from unittest.mock import AsyncMock, patch
+
+import pytest
+from switchbot import ColorMode as switchbotColorMode
+
+from homeassistant.components.light import (
+    ATTR_BRIGHTNESS,
+    ATTR_COLOR_TEMP_KELVIN,
+    ATTR_RGB_COLOR,
+    DOMAIN as LIGHT_DOMAIN,
+    SERVICE_TURN_OFF,
+    SERVICE_TURN_ON,
+)
+from homeassistant.const import ATTR_ENTITY_ID
+from homeassistant.core import HomeAssistant
+
+from . import WOSTRIP_SERVICE_INFO
+
+from tests.common import MockConfigEntry
+from tests.components.bluetooth import inject_bluetooth_service_info
+
+
+@pytest.mark.parametrize(
+    (
+        "service",
+        "service_data",
+        "mock_method",
+        "expected_args",
+        "color_modes",
+        "color_mode",
+    ),
+    [
+        (
+            SERVICE_TURN_OFF,
+            {},
+            "turn_off",
+            (),
+            {switchbotColorMode.RGB},
+            switchbotColorMode.RGB,
+        ),
+        (
+            SERVICE_TURN_ON,
+            {},
+            "turn_on",
+            (),
+            {switchbotColorMode.RGB},
+            switchbotColorMode.RGB,
+        ),
+        (
+            SERVICE_TURN_ON,
+            {ATTR_BRIGHTNESS: 128},
+            "set_brightness",
+            (round(128 / 255 * 100),),
+            {switchbotColorMode.RGB},
+            switchbotColorMode.RGB,
+        ),
+        (
+            SERVICE_TURN_ON,
+            {ATTR_RGB_COLOR: (255, 0, 0)},
+            "set_rgb",
+            (round(255 / 255 * 100), 255, 0, 0),
+            {switchbotColorMode.RGB},
+            switchbotColorMode.RGB,
+        ),
+        (
+            SERVICE_TURN_ON,
+            {ATTR_COLOR_TEMP_KELVIN: 4000},
+            "set_color_temp",
+            (100, 4000),
+            {switchbotColorMode.COLOR_TEMP},
+            switchbotColorMode.COLOR_TEMP,
+        ),
+    ],
+)
+async def test_light_strip_services(
+    hass: HomeAssistant,
+    mock_entry_factory: Callable[[str], MockConfigEntry],
+    service: str,
+    service_data: dict,
+    mock_method: str,
+    expected_args: Any,
+    color_modes: set | None,
+    color_mode: switchbotColorMode | None,
+) -> None:
+    """Test all SwitchBot light strip services with proper parameters."""
+    inject_bluetooth_service_info(hass, WOSTRIP_SERVICE_INFO)
+
+    entry = mock_entry_factory(sensor_type="light_strip")
+    entry.add_to_hass(hass)
+    entity_id = "light.test_name"
+
+    with (
+        patch("switchbot.SwitchbotLightStrip.color_modes", new=color_modes),
+        patch("switchbot.SwitchbotLightStrip.color_mode", new=color_mode),
+        patch(
+            "switchbot.SwitchbotLightStrip.turn_on",
+            new=AsyncMock(return_value=True),
+        ) as mock_turn_on,
+        patch(
+            "switchbot.SwitchbotLightStrip.turn_off",
+            new=AsyncMock(return_value=True),
+        ) as mock_turn_off,
+        patch(
+            "switchbot.SwitchbotLightStrip.set_brightness",
+            new=AsyncMock(return_value=True),
+        ) as mock_set_brightness,
+        patch(
+            "switchbot.SwitchbotLightStrip.set_rgb",
+            new=AsyncMock(return_value=True),
+        ) as mock_set_rgb,
+        patch(
+            "switchbot.SwitchbotLightStrip.set_color_temp",
+            new=AsyncMock(return_value=True),
+        ) as mock_set_color_temp,
+        patch("switchbot.SwitchbotLightStrip.update", new=AsyncMock(return_value=None)),
+    ):
+        assert await hass.config_entries.async_setup(entry.entry_id)
+        await hass.async_block_till_done()
+
+        await hass.services.async_call(
+            LIGHT_DOMAIN,
+            service,
+            {**service_data, ATTR_ENTITY_ID: entity_id},
+            blocking=True,
+        )
+
+        mock_map = {
+            "turn_off": mock_turn_off,
+            "turn_on": mock_turn_on,
+            "set_brightness": mock_set_brightness,
+            "set_rgb": mock_set_rgb,
+            "set_color_temp": mock_set_color_temp,
+        }
+        mock_instance = mock_map[mock_method]
+        mock_instance.assert_awaited_once_with(*expected_args)