Add plug mini eu for switchbot integration (home-assistant#151130)

Author: zerzhang

homeassistant/components/switchbot/__init__.py

@@ -97,6 +97,7 @@
     SupportedModels.STRIP_LIGHT_3.value: [Platform.LIGHT, Platform.SENSOR],
     SupportedModels.RGBICWW_FLOOR_LAMP.value: [Platform.LIGHT, Platform.SENSOR],
     SupportedModels.RGBICWW_STRIP_LIGHT.value: [Platform.LIGHT, Platform.SENSOR],
+    SupportedModels.PLUG_MINI_EU.value: [Platform.SWITCH, Platform.SENSOR],
 }
 CLASS_BY_DEVICE = {
     SupportedModels.CEILING_LIGHT.value: switchbot.SwitchbotCeilingLight,
@@ -127,6 +128,7 @@
     SupportedModels.STRIP_LIGHT_3.value: switchbot.SwitchbotStripLight3,
     SupportedModels.RGBICWW_FLOOR_LAMP.value: switchbot.SwitchbotRgbicLight,
     SupportedModels.RGBICWW_STRIP_LIGHT.value: switchbot.SwitchbotRgbicLight,
+    SupportedModels.PLUG_MINI_EU.value: switchbot.SwitchbotRelaySwitch,
 }
 
 

homeassistant/components/switchbot/const.py

@@ -53,6 +53,7 @@ class SupportedModels(StrEnum):
     STRIP_LIGHT_3 = "strip_light_3"
     RGBICWW_STRIP_LIGHT = "rgbicww_strip_light"
     RGBICWW_FLOOR_LAMP = "rgbicww_floor_lamp"
+    PLUG_MINI_EU = "plug_mini_eu"
 
 
 CONNECTABLE_SUPPORTED_MODEL_TYPES = {
@@ -85,6 +86,7 @@ class SupportedModels(StrEnum):
     SwitchbotModel.STRIP_LIGHT_3: SupportedModels.STRIP_LIGHT_3,
     SwitchbotModel.RGBICWW_STRIP_LIGHT: SupportedModels.RGBICWW_STRIP_LIGHT,
     SwitchbotModel.RGBICWW_FLOOR_LAMP: SupportedModels.RGBICWW_FLOOR_LAMP,
+    SwitchbotModel.PLUG_MINI_EU: SupportedModels.PLUG_MINI_EU,
 }
 
 NON_CONNECTABLE_SUPPORTED_MODEL_TYPES = {
@@ -118,6 +120,7 @@ class SupportedModels(StrEnum):
     SwitchbotModel.STRIP_LIGHT_3,
     SwitchbotModel.RGBICWW_STRIP_LIGHT,
     SwitchbotModel.RGBICWW_FLOOR_LAMP,
+    SwitchbotModel.PLUG_MINI_EU,
 }
 
 ENCRYPTED_SWITCHBOT_MODEL_TO_CLASS: dict[
@@ -136,6 +139,7 @@ class SupportedModels(StrEnum):
     SwitchbotModel.STRIP_LIGHT_3: switchbot.SwitchbotStripLight3,
     SwitchbotModel.RGBICWW_STRIP_LIGHT: switchbot.SwitchbotRgbicLight,
     SwitchbotModel.RGBICWW_FLOOR_LAMP: switchbot.SwitchbotRgbicLight,
+    SwitchbotModel.PLUG_MINI_EU: switchbot.SwitchbotRelaySwitch,
 }
 
 HASS_SENSOR_TYPE_TO_SWITCHBOT_MODEL = {

tests/components/switchbot/__init__.py

@@ -1056,3 +1056,27 @@ def make_advertisement(
     connectable=True,
     tx_power=-127,
 )
+
+PLUG_MINI_EU_SERVICE_INFO = BluetoothServiceInfoBleak(
+    name="Plug Mini (EU)",
+    manufacturer_data={
+        2409: b"\x94\xa9\x90T\x85^?\xa1\x00\x00\x04\xe6\x00\x00\x00\x00",
+    },
+    service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"?\x00\x00\x00"},
+    service_uuids=["cba20d00-224d-11e6-9fb8-0002a5d5c51b"],
+    address="AA:BB:CC:DD:EE:FF",
+    rssi=-60,
+    source="local",
+    advertisement=generate_advertisement_data(
+        local_name="Plug Mini (EU)",
+        manufacturer_data={
+            2409: b"\x94\xa9\x90T\x85^?\xa1\x00\x00\x04\xe6\x00\x00\x00\x00",
+        },
+        service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"?\x00\x00\x00"},
+        service_uuids=["cba20d00-224d-11e6-9fb8-0002a5d5c51b"],
+    ),
+    device=generate_ble_device("AA:BB:CC:DD:EE:FF", "Plug Mini (EU)"),
+    time=0,
+    connectable=True,
+    tx_power=-127,
+)

tests/components/switchbot/test_sensor.py

@@ -28,6 +28,7 @@
     HUB3_SERVICE_INFO,
     HUBMINI_MATTER_SERVICE_INFO,
     LEAK_SERVICE_INFO,
+    PLUG_MINI_EU_SERVICE_INFO,
     REMOTE_SERVICE_INFO,
     WOHAND_SERVICE_INFO,
     WOHUB2_SERVICE_INFO,
@@ -542,3 +543,77 @@ async def test_evaporative_humidifier_sensor(hass: HomeAssistant) -> None:
 
         assert await hass.config_entries.async_unload(entry.entry_id)
         await hass.async_block_till_done()
+
+
+@pytest.mark.usefixtures("entity_registry_enabled_by_default")
+async def test_plug_mini_eu_sensor(hass: HomeAssistant) -> None:
+    """Test setting up creates the plug mini eu sensor."""
+    await async_setup_component(hass, DOMAIN, {})
+    inject_bluetooth_service_info(hass, PLUG_MINI_EU_SERVICE_INFO)
+
+    with patch(
+        "homeassistant.components.switchbot.switch.switchbot.SwitchbotRelaySwitch.get_basic_info",
+        new=AsyncMock(
+            return_value={
+                "power": 500,
+                "current": 0.5,
+                "voltage": 230,
+                "energy": 0.4,
+            }
+        ),
+    ):
+        entry = MockConfigEntry(
+            domain=DOMAIN,
+            data={
+                CONF_ADDRESS: "aa:bb:cc:dd:ee:ff",
+                CONF_NAME: "test-name",
+                CONF_SENSOR_TYPE: "plug_mini_eu",
+                CONF_KEY_ID: "ff",
+                CONF_ENCRYPTION_KEY: "ffffffffffffffffffffffffffffffff",
+            },
+            unique_id="aabbccddeeaa",
+        )
+        entry.add_to_hass(hass)
+
+        assert await hass.config_entries.async_setup(entry.entry_id)
+        await hass.async_block_till_done()
+
+    assert len(hass.states.async_all("sensor")) == 5
+
+    power_sensor = hass.states.get("sensor.test_name_power")
+    power_sensor_attrs = power_sensor.attributes
+    assert power_sensor.state == "500"
+    assert power_sensor_attrs[ATTR_FRIENDLY_NAME] == "test-name Power"
+    assert power_sensor_attrs[ATTR_UNIT_OF_MEASUREMENT] == "W"
+    assert power_sensor_attrs[ATTR_STATE_CLASS] == "measurement"
+
+    voltage_sensor = hass.states.get("sensor.test_name_voltage")
+    voltage_sensor_attrs = voltage_sensor.attributes
+    assert voltage_sensor.state == "230"
+    assert voltage_sensor_attrs[ATTR_FRIENDLY_NAME] == "test-name Voltage"
+    assert voltage_sensor_attrs[ATTR_UNIT_OF_MEASUREMENT] == "V"
+    assert voltage_sensor_attrs[ATTR_STATE_CLASS] == "measurement"
+
+    current_sensor = hass.states.get("sensor.test_name_current")
+    current_sensor_attrs = current_sensor.attributes
+    assert current_sensor.state == "0.5"
+    assert current_sensor_attrs[ATTR_FRIENDLY_NAME] == "test-name Current"
+    assert current_sensor_attrs[ATTR_UNIT_OF_MEASUREMENT] == "A"
+    assert current_sensor_attrs[ATTR_STATE_CLASS] == "measurement"
+
+    energy_sensor = hass.states.get("sensor.test_name_energy")
+    energy_sensor_attrs = energy_sensor.attributes
+    assert energy_sensor.state == "0.4"
+    assert energy_sensor_attrs[ATTR_FRIENDLY_NAME] == "test-name Energy"
+    assert energy_sensor_attrs[ATTR_UNIT_OF_MEASUREMENT] == "kWh"
+    assert energy_sensor_attrs[ATTR_STATE_CLASS] == "total_increasing"
+
+    rssi_sensor = hass.states.get("sensor.test_name_bluetooth_signal")
+    rssi_sensor_attrs = rssi_sensor.attributes
+    assert rssi_sensor.state == "-60"
+    assert rssi_sensor_attrs[ATTR_FRIENDLY_NAME] == "test-name Bluetooth signal"
+    assert rssi_sensor_attrs[ATTR_UNIT_OF_MEASUREMENT] == "dBm"
+    assert rssi_sensor_attrs[ATTR_STATE_CLASS] == "measurement"
+
+    assert await hass.config_entries.async_unload(entry.entry_id)
+    await hass.async_block_till_done()

tests/components/switchbot/test_switch.py

@@ -6,6 +6,7 @@
 import pytest
 from switchbot.devices.device import SwitchbotOperationError
 
+from homeassistant.components.bluetooth import BluetoothServiceInfoBleak
 from homeassistant.components.switch import (
     DOMAIN as SWITCH_DOMAIN,
     SERVICE_TURN_OFF,
@@ -16,7 +17,7 @@
 from homeassistant.core import HomeAssistant, State
 from homeassistant.exceptions import HomeAssistantError
 
-from . import WOHAND_SERVICE_INFO
+from . import PLUG_MINI_EU_SERVICE_INFO, WOHAND_SERVICE_INFO
 
 from tests.common import MockConfigEntry, mock_restore_cache
 from tests.components.bluetooth import inject_bluetooth_service_info
@@ -103,3 +104,51 @@ async def test_exception_handling_switch(
                 {ATTR_ENTITY_ID: entity_id},
                 blocking=True,
             )
+
+
+@pytest.mark.parametrize(
+    ("sensor_type", "service_info"),
+    [
+        ("plug_mini_eu", PLUG_MINI_EU_SERVICE_INFO),
+    ],
+)
+@pytest.mark.parametrize(
+    ("service", "mock_method"),
+    [
+        (SERVICE_TURN_ON, "turn_on"),
+        (SERVICE_TURN_OFF, "turn_off"),
+    ],
+)
+async def test_relay_switch_control(
+    hass: HomeAssistant,
+    mock_entry_encrypted_factory: Callable[[str], MockConfigEntry],
+    sensor_type: str,
+    service_info: BluetoothServiceInfoBleak,
+    service: str,
+    mock_method: str,
+) -> None:
+    """Test Relay Switch control."""
+    inject_bluetooth_service_info(hass, service_info)
+
+    entry = mock_entry_encrypted_factory(sensor_type=sensor_type)
+    entry.add_to_hass(hass)
+
+    mocked_instance = AsyncMock(return_value=True)
+    with patch.multiple(
+        "homeassistant.components.switchbot.switch.switchbot.SwitchbotRelaySwitch",
+        update=AsyncMock(return_value=None),
+        **{mock_method: mocked_instance},
+    ):
+        assert await hass.config_entries.async_setup(entry.entry_id)
+        await hass.async_block_till_done()
+
+        entity_id = "switch.test_name"
+
+        await hass.services.async_call(
+            SWITCH_DOMAIN,
+            service,
+            {ATTR_ENTITY_ID: entity_id},
+            blocking=True,
+        )
+
+        mocked_instance.assert_awaited_once()