Skip to content
Draft
Show file tree
Hide file tree
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions homeassistant/components/switchbot/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -192,6 +192,7 @@
Platform.SENSOR,
],
SupportedModels.WEATHER_STATION.value: [Platform.SENSOR],
SupportedModels.UNIVERSAL_REMOTE.value: [Platform.SENSOR],
}
CLASS_BY_DEVICE = {
SupportedModels.CEILING_LIGHT.value: switchbot.SwitchbotCeilingLight,
Expand All @@ -209,6 +210,7 @@
SupportedModels.ROLLER_SHADE.value: switchbot.SwitchbotRollerShade,
SupportedModels.CIRCULATOR_FAN.value: switchbot.SwitchbotFan,
SupportedModels.STANDING_FAN.value: switchbot.SwitchbotStandingFan,
SupportedModels.UNIVERSAL_REMOTE.value: switchbot.SwitchbotUniversalRemote,
SupportedModels.S10_VACUUM.value: switchbot.SwitchbotVacuum,
SupportedModels.S20_VACUUM.value: switchbot.SwitchbotVacuum,
SupportedModels.K10_VACUUM.value: switchbot.SwitchbotVacuum,
Expand Down
2 changes: 2 additions & 0 deletions homeassistant/components/switchbot/const.py
Original file line number Diff line number Diff line change
Expand Up @@ -72,6 +72,7 @@ class SupportedModels(StrEnum):
LOCK_PRO_WIFI = "lock_pro_wifi"
WEATHER_STATION = "weather_station"
STANDING_FAN = "standing_fan"
UNIVERSAL_REMOTE = "universal_remote"


CONNECTABLE_SUPPORTED_MODEL_TYPES = {
Expand Down Expand Up @@ -122,6 +123,7 @@ class SupportedModels(StrEnum):
SwitchbotModel.LOCK_VISION: SupportedModels.LOCK_VISION,
SwitchbotModel.LOCK_PRO_WIFI: SupportedModels.LOCK_PRO_WIFI,
SwitchbotModel.STANDING_FAN: SupportedModels.STANDING_FAN,
SwitchbotModel.UNIVERSAL_REMOTE: SupportedModels.UNIVERSAL_REMOTE,
}

NON_CONNECTABLE_SUPPORTED_MODEL_TYPES = {
Expand Down
25 changes: 25 additions & 0 deletions tests/components/switchbot/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -589,6 +589,31 @@ def make_advertisement(
)


UNIVERSAL_REMOTE_SERVICE_INFO = BluetoothServiceInfoBleak(
name="Universal Remote",
manufacturer_data={
2409: b"\xaa\xbb\xcc\xdd\xee\xff\x00\x00\x00\x50\x00\x00\x00\x00\x00\x00"
},
service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"'\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="Universal Remote",
manufacturer_data={
2409: b"\xaa\xbb\xcc\xdd\xee\xff\x00\x00\x00\x50\x00\x00\x00\x00\x00\x00"
},
service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"'\x00"},
service_uuids=["cba20d00-224d-11e6-9fb8-0002a5d5c51b"],
),
device=generate_ble_device("AA:BB:CC:DD:EE:FF", "Universal Remote"),
time=0,
connectable=True,
tx_power=-127,
)


K20_VACUUM_SERVICE_INFO = BluetoothServiceInfoBleak(
name="K20 Vacuum",
manufacturer_data={
Expand Down
40 changes: 40 additions & 0 deletions tests/components/switchbot/test_sensor.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,6 +41,7 @@
PRESENCE_SENSOR_SERVICE_INFO,
RELAY_SWITCH_2PM_SERVICE_INFO,
REMOTE_SERVICE_INFO,
UNIVERSAL_REMOTE_SERVICE_INFO,
WEATHER_STATION_SERVICE_INFO,
WOHAND_SERVICE_INFO,
WOHUB2_SERVICE_INFO,
Expand Down Expand Up @@ -293,6 +294,45 @@ async def test_remote(hass: HomeAssistant) -> None:
await hass.async_block_till_done()


@pytest.mark.usefixtures("entity_registry_enabled_by_default")
async def test_universal_remote_sensor(hass: HomeAssistant) -> None:
"""Test setting up the Universal Remote battery sensor."""
await async_setup_component(hass, DOMAIN, {})
inject_bluetooth_service_info(hass, UNIVERSAL_REMOTE_SERVICE_INFO)

entry = MockConfigEntry(
domain=DOMAIN,
data={
CONF_ADDRESS: "AA:BB:CC:DD:EE:FF",
CONF_NAME: "test-name",
CONF_SENSOR_TYPE: "universal_remote",
},
unique_id="aabbccddeeff",
)
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")) == 2

battery_sensor = hass.states.get("sensor.test_name_battery")
battery_sensor_attrs = battery_sensor.attributes
assert battery_sensor.state == "80"
assert battery_sensor_attrs[ATTR_FRIENDLY_NAME] == "test-name Battery"
assert battery_sensor_attrs[ATTR_UNIT_OF_MEASUREMENT] == "%"
assert battery_sensor_attrs[ATTR_STATE_CLASS] == "measurement"

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 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_hub2_sensor(hass: HomeAssistant) -> None:
"""Test setting up creates the sensor for WoHub2."""
Expand Down