From 67e16d77e8627b88b3dd811f9b3328cac2873d58 Mon Sep 17 00:00:00 2001
From: "J. Nick Koston" <nick@koston.org>
Date: Thu, 21 Jul 2022 20:31:23 -0500
Subject: [PATCH] Add SensorPush BLE integration (#75531)
---
CODEOWNERS | 2 +
.../components/sensorpush/__init__.py | 56 +++++++
.../components/sensorpush/config_flow.py | 93 ++++++++++++
homeassistant/components/sensorpush/const.py | 3 +
.../components/sensorpush/manifest.json | 15 ++
homeassistant/components/sensorpush/sensor.py | 143 ++++++++++++++++++
.../components/sensorpush/strings.json | 21 +++
.../sensorpush/translations/en.json | 21 +++
homeassistant/generated/bluetooth.py | 4 +
homeassistant/generated/config_flows.py | 1 +
requirements_all.txt | 3 +
requirements_test_all.txt | 3 +
tests/components/sensorpush/__init__.py | 34 +++++
tests/components/sensorpush/conftest.py | 8 +
.../components/sensorpush/test_config_flow.py | 98 ++++++++++++
tests/components/sensorpush/test_sensor.py | 50 ++++++
16 files changed, 555 insertions(+)
create mode 100644 homeassistant/components/sensorpush/__init__.py
create mode 100644 homeassistant/components/sensorpush/config_flow.py
create mode 100644 homeassistant/components/sensorpush/const.py
create mode 100644 homeassistant/components/sensorpush/manifest.json
create mode 100644 homeassistant/components/sensorpush/sensor.py
create mode 100644 homeassistant/components/sensorpush/strings.json
create mode 100644 homeassistant/components/sensorpush/translations/en.json
create mode 100644 tests/components/sensorpush/__init__.py
create mode 100644 tests/components/sensorpush/conftest.py
create mode 100644 tests/components/sensorpush/test_config_flow.py
create mode 100644 tests/components/sensorpush/test_sensor.py
diff --git a/CODEOWNERS b/CODEOWNERS
index dbbb21f0d6c..2a3999779fc 100644
--- a/CODEOWNERS
+++ b/CODEOWNERS
@@ -922,6 +922,8 @@ build.json @home-assistant/supervisor
/tests/components/sensibo/ @andrey-git @gjohansson-ST
/homeassistant/components/sensor/ @home-assistant/core
/tests/components/sensor/ @home-assistant/core
+/homeassistant/components/sensorpush/ @bdraco
+/tests/components/sensorpush/ @bdraco
/homeassistant/components/sentry/ @dcramer @frenck
/tests/components/sentry/ @dcramer @frenck
/homeassistant/components/senz/ @milanmeu
diff --git a/homeassistant/components/sensorpush/__init__.py b/homeassistant/components/sensorpush/__init__.py
new file mode 100644
index 00000000000..89e2287be70
--- /dev/null
+++ b/homeassistant/components/sensorpush/__init__.py
@@ -0,0 +1,56 @@
+"""The SensorPush Bluetooth integration."""
+from __future__ import annotations
+
+import logging
+
+from sensorpush_ble import SensorPushBluetoothDeviceData
+
+from homeassistant.components.bluetooth.passive_update_coordinator import (
+ PassiveBluetoothDataUpdate,
+ PassiveBluetoothDataUpdateCoordinator,
+)
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import Platform
+from homeassistant.core import HomeAssistant, callback
+from homeassistant.helpers.service_info.bluetooth import BluetoothServiceInfo
+
+from .const import DOMAIN
+from .sensor import sensor_update_to_bluetooth_data_update
+
+PLATFORMS: list[Platform] = [Platform.SENSOR]
+
+_LOGGER = logging.getLogger(__name__)
+
+
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+ """Set up SensorPush BLE device from a config entry."""
+ address = entry.unique_id
+ assert address is not None
+
+ data = SensorPushBluetoothDeviceData()
+
+ @callback
+ def _async_update_data(
+ service_info: BluetoothServiceInfo,
+ ) -> PassiveBluetoothDataUpdate:
+ """Update data from SensorPush Bluetooth."""
+ return sensor_update_to_bluetooth_data_update(data.update(service_info))
+
+ hass.data.setdefault(DOMAIN, {})[
+ entry.entry_id
+ ] = PassiveBluetoothDataUpdateCoordinator(
+ hass,
+ _LOGGER,
+ update_method=_async_update_data,
+ address=address,
+ )
+ await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
+ return True
+
+
+async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+ """Unload a config entry."""
+ if unload_ok := await hass.config_entries.async_unload_platforms(entry, PLATFORMS):
+ hass.data[DOMAIN].pop(entry.entry_id)
+
+ return unload_ok
diff --git a/homeassistant/components/sensorpush/config_flow.py b/homeassistant/components/sensorpush/config_flow.py
new file mode 100644
index 00000000000..1a8e8b47abe
--- /dev/null
+++ b/homeassistant/components/sensorpush/config_flow.py
@@ -0,0 +1,93 @@
+"""Config flow for sensorpush integration."""
+from __future__ import annotations
+
+from typing import Any
+
+from sensorpush_ble import SensorPushBluetoothDeviceData as DeviceData
+import voluptuous as vol
+
+from homeassistant.components.bluetooth import (
+ BluetoothServiceInfo,
+ async_discovered_service_info,
+)
+from homeassistant.config_entries import ConfigFlow
+from homeassistant.const import CONF_ADDRESS
+from homeassistant.data_entry_flow import FlowResult
+
+from .const import DOMAIN
+
+
+class SensorPushConfigFlow(ConfigFlow, domain=DOMAIN):
+ """Handle a config flow for sensorpush."""
+
+ VERSION = 1
+
+ def __init__(self) -> None:
+ """Initialize the config flow."""
+ self._discovery_info: BluetoothServiceInfo | None = None
+ self._discovered_device: DeviceData | None = None
+ self._discovered_devices: dict[str, str] = {}
+
+ async def async_step_bluetooth(
+ self, discovery_info: BluetoothServiceInfo
+ ) -> FlowResult:
+ """Handle the bluetooth discovery step."""
+ await self.async_set_unique_id(discovery_info.address)
+ self._abort_if_unique_id_configured()
+ device = DeviceData()
+ if not device.supported(discovery_info):
+ return self.async_abort(reason="not_supported")
+ self._discovery_info = discovery_info
+ self._discovered_device = device
+ return await self.async_step_bluetooth_confirm()
+
+ async def async_step_bluetooth_confirm(
+ self, user_input: dict[str, Any] | None = None
+ ) -> FlowResult:
+ """Confirm discovery."""
+ assert self._discovered_device is not None
+ device = self._discovered_device
+ assert self._discovery_info is not None
+ discovery_info = self._discovery_info
+ title = device.title or device.get_device_name() or discovery_info.name
+ if user_input is not None:
+ return self.async_create_entry(title=title, data={})
+
+ self._set_confirm_only()
+ placeholders = {"name": title}
+ self.context["title_placeholders"] = placeholders
+ return self.async_show_form(
+ step_id="bluetooth_confirm", description_placeholders=placeholders
+ )
+
+ async def async_step_user(
+ self, user_input: dict[str, Any] | None = None
+ ) -> FlowResult:
+ """Handle the user step to pick discovered device."""
+ if user_input is not None:
+ address = user_input[CONF_ADDRESS]
+ await self.async_set_unique_id(address, raise_on_progress=False)
+ return self.async_create_entry(
+ title=self._discovered_devices[address], data={}
+ )
+
+ current_addresses = self._async_current_ids()
+ for discovery_info in async_discovered_service_info(self.hass):
+ address = discovery_info.address
+ if address in current_addresses or address in self._discovered_devices:
+ continue
+ device = DeviceData()
+ if device.supported(discovery_info):
+ self._discovered_devices[address] = (
+ device.title or device.get_device_name() or discovery_info.name
+ )
+
+ if not self._discovered_devices:
+ return self.async_abort(reason="no_devices_found")
+
+ return self.async_show_form(
+ step_id="user",
+ data_schema=vol.Schema(
+ {vol.Required(CONF_ADDRESS): vol.In(self._discovered_devices)}
+ ),
+ )
diff --git a/homeassistant/components/sensorpush/const.py b/homeassistant/components/sensorpush/const.py
new file mode 100644
index 00000000000..8f566c72d07
--- /dev/null
+++ b/homeassistant/components/sensorpush/const.py
@@ -0,0 +1,3 @@
+"""Constants for the SensorPush Bluetooth integration."""
+
+DOMAIN = "sensorpush"
diff --git a/homeassistant/components/sensorpush/manifest.json b/homeassistant/components/sensorpush/manifest.json
new file mode 100644
index 00000000000..08212f88132
--- /dev/null
+++ b/homeassistant/components/sensorpush/manifest.json
@@ -0,0 +1,15 @@
+{
+ "domain": "sensorpush",
+ "name": "SensorPush",
+ "config_flow": true,
+ "documentation": "https://www.home-assistant.io/integrations/sensorpush",
+ "bluetooth": [
+ {
+ "local_name": "SensorPush*"
+ }
+ ],
+ "requirements": ["sensorpush-ble==1.4.2"],
+ "dependencies": ["bluetooth"],
+ "codeowners": ["@bdraco"],
+ "iot_class": "local_push"
+}
diff --git a/homeassistant/components/sensorpush/sensor.py b/homeassistant/components/sensorpush/sensor.py
new file mode 100644
index 00000000000..147555990e4
--- /dev/null
+++ b/homeassistant/components/sensorpush/sensor.py
@@ -0,0 +1,143 @@
+"""Support for sensorpush ble sensors."""
+from __future__ import annotations
+
+from typing import Optional, Union
+
+from sensorpush_ble import DeviceClass, DeviceKey, SensorDeviceInfo, SensorUpdate, Units
+
+from homeassistant import config_entries
+from homeassistant.components.bluetooth.passive_update_coordinator import (
+ PassiveBluetoothCoordinatorEntity,
+ PassiveBluetoothDataUpdate,
+ PassiveBluetoothDataUpdateCoordinator,
+ PassiveBluetoothEntityKey,
+)
+from homeassistant.components.sensor import (
+ SensorDeviceClass,
+ SensorEntity,
+ SensorEntityDescription,
+ SensorStateClass,
+)
+from homeassistant.const import (
+ ATTR_MANUFACTURER,
+ ATTR_MODEL,
+ ATTR_NAME,
+ PERCENTAGE,
+ PRESSURE_MBAR,
+ SIGNAL_STRENGTH_DECIBELS_MILLIWATT,
+ TEMP_CELSIUS,
+)
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity import DeviceInfo
+from homeassistant.helpers.entity_platform import AddEntitiesCallback
+
+from .const import DOMAIN
+
+SENSOR_DESCRIPTIONS = {
+ (DeviceClass.TEMPERATURE, Units.TEMP_CELSIUS): SensorEntityDescription(
+ key=f"{DeviceClass.TEMPERATURE}_{Units.TEMP_CELSIUS}",
+ device_class=SensorDeviceClass.TEMPERATURE,
+ native_unit_of_measurement=TEMP_CELSIUS,
+ state_class=SensorStateClass.MEASUREMENT,
+ ),
+ (DeviceClass.HUMIDITY, Units.PERCENTAGE): SensorEntityDescription(
+ key=f"{DeviceClass.HUMIDITY}_{Units.PERCENTAGE}",
+ device_class=SensorDeviceClass.HUMIDITY,
+ native_unit_of_measurement=PERCENTAGE,
+ state_class=SensorStateClass.MEASUREMENT,
+ ),
+ (DeviceClass.PRESSURE, Units.PRESSURE_MBAR): SensorEntityDescription(
+ key=f"{DeviceClass.PRESSURE}_{Units.PRESSURE_MBAR}",
+ device_class=SensorDeviceClass.PRESSURE,
+ native_unit_of_measurement=PRESSURE_MBAR,
+ state_class=SensorStateClass.MEASUREMENT,
+ ),
+ (
+ DeviceClass.SIGNAL_STRENGTH,
+ Units.SIGNAL_STRENGTH_DECIBELS_MILLIWATT,
+ ): SensorEntityDescription(
+ key=f"{DeviceClass.SIGNAL_STRENGTH}_{Units.SIGNAL_STRENGTH_DECIBELS_MILLIWATT}",
+ device_class=SensorDeviceClass.SIGNAL_STRENGTH,
+ native_unit_of_measurement=SIGNAL_STRENGTH_DECIBELS_MILLIWATT,
+ state_class=SensorStateClass.MEASUREMENT,
+ entity_registry_enabled_default=False,
+ ),
+}
+
+
+def _device_key_to_bluetooth_entity_key(
+ device_key: DeviceKey,
+) -> PassiveBluetoothEntityKey:
+ """Convert a device key to an entity key."""
+ return PassiveBluetoothEntityKey(device_key.key, device_key.device_id)
+
+
+def _sensor_device_info_to_hass(
+ sensor_device_info: SensorDeviceInfo,
+) -> DeviceInfo:
+ """Convert a sensor device info to a sensor device info."""
+ hass_device_info = DeviceInfo({})
+ if sensor_device_info.name is not None:
+ hass_device_info[ATTR_NAME] = sensor_device_info.name
+ if sensor_device_info.manufacturer is not None:
+ hass_device_info[ATTR_MANUFACTURER] = sensor_device_info.manufacturer
+ if sensor_device_info.model is not None:
+ hass_device_info[ATTR_MODEL] = sensor_device_info.model
+ return hass_device_info
+
+
+def sensor_update_to_bluetooth_data_update(
+ sensor_update: SensorUpdate,
+) -> PassiveBluetoothDataUpdate:
+ """Convert a sensor update to a bluetooth data update."""
+ return PassiveBluetoothDataUpdate(
+ devices={
+ device_id: _sensor_device_info_to_hass(device_info)
+ for device_id, device_info in sensor_update.devices.items()
+ },
+ entity_descriptions={
+ _device_key_to_bluetooth_entity_key(device_key): SENSOR_DESCRIPTIONS[
+ (description.device_class, description.native_unit_of_measurement)
+ ]
+ for device_key, description in sensor_update.entity_descriptions.items()
+ if description.device_class and description.native_unit_of_measurement
+ },
+ entity_data={
+ _device_key_to_bluetooth_entity_key(device_key): sensor_values.native_value
+ for device_key, sensor_values in sensor_update.entity_values.items()
+ },
+ entity_names={
+ _device_key_to_bluetooth_entity_key(device_key): sensor_values.name
+ for device_key, sensor_values in sensor_update.entity_values.items()
+ },
+ )
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ entry: config_entries.ConfigEntry,
+ async_add_entities: AddEntitiesCallback,
+) -> None:
+ """Set up the SensorPush BLE sensors."""
+ coordinator: PassiveBluetoothDataUpdateCoordinator = hass.data[DOMAIN][
+ entry.entry_id
+ ]
+ entry.async_on_unload(
+ coordinator.async_add_entities_listener(
+ SensorPushBluetoothSensorEntity, async_add_entities
+ )
+ )
+
+
+class SensorPushBluetoothSensorEntity(
+ PassiveBluetoothCoordinatorEntity[
+ PassiveBluetoothDataUpdateCoordinator[Optional[Union[float, int]]]
+ ],
+ SensorEntity,
+):
+ """Representation of a sensorpush ble sensor."""
+
+ @property
+ def native_value(self) -> int | float | None:
+ """Return the native value."""
+ return self.coordinator.entity_data.get(self.entity_key)
diff --git a/homeassistant/components/sensorpush/strings.json b/homeassistant/components/sensorpush/strings.json
new file mode 100644
index 00000000000..7111626cca1
--- /dev/null
+++ b/homeassistant/components/sensorpush/strings.json
@@ -0,0 +1,21 @@
+{
+ "config": {
+ "flow_title": "[%key:component::bluetooth::config::flow_title%]",
+ "step": {
+ "user": {
+ "description": "[%key:component::bluetooth::config::step::user::description%]",
+ "data": {
+ "address": "[%key:component::bluetooth::config::step::user::data::address%]"
+ }
+ },
+ "bluetooth_confirm": {
+ "description": "[%key:component::bluetooth::config::step::bluetooth_confirm::description%]"
+ }
+ },
+ "abort": {
+ "no_devices_found": "[%key:common::config_flow::abort::no_devices_found%]",
+ "already_in_progress": "[%key:common::config_flow::abort::already_in_progress%]",
+ "already_configured": "[%key:common::config_flow::abort::already_configured_device%]"
+ }
+ }
+}
diff --git a/homeassistant/components/sensorpush/translations/en.json b/homeassistant/components/sensorpush/translations/en.json
new file mode 100644
index 00000000000..d24df64f135
--- /dev/null
+++ b/homeassistant/components/sensorpush/translations/en.json
@@ -0,0 +1,21 @@
+{
+ "config": {
+ "abort": {
+ "already_configured": "Device is already configured",
+ "already_in_progress": "Configuration flow is already in progress",
+ "no_devices_found": "No devices found on the network"
+ },
+ "flow_title": "{name}",
+ "step": {
+ "bluetooth_confirm": {
+ "description": "Do you want to setup {name}?"
+ },
+ "user": {
+ "data": {
+ "address": "Device"
+ },
+ "description": "Choose a device to setup"
+ }
+ }
+ }
+}
\ No newline at end of file
diff --git a/homeassistant/generated/bluetooth.py b/homeassistant/generated/bluetooth.py
index 03486b6043c..e3a4edc11d4 100644
--- a/homeassistant/generated/bluetooth.py
+++ b/homeassistant/generated/bluetooth.py
@@ -14,6 +14,10 @@ BLUETOOTH: list[dict[str, str | int | list[int]]] = [
6
]
},
+ {
+ "domain": "sensorpush",
+ "local_name": "SensorPush*"
+ },
{
"domain": "switchbot",
"service_uuid": "cba20d00-224d-11e6-9fb8-0002a5d5c51b"
diff --git a/homeassistant/generated/config_flows.py b/homeassistant/generated/config_flows.py
index 3c02842e856..95050ea6dbd 100644
--- a/homeassistant/generated/config_flows.py
+++ b/homeassistant/generated/config_flows.py
@@ -308,6 +308,7 @@ FLOWS = {
"sense",
"senseme",
"sensibo",
+ "sensorpush",
"sentry",
"senz",
"sharkiq",
diff --git a/requirements_all.txt b/requirements_all.txt
index c88df779c25..f125a49c0a4 100644
--- a/requirements_all.txt
+++ b/requirements_all.txt
@@ -2154,6 +2154,9 @@ sendgrid==6.8.2
# homeassistant.components.sense
sense_energy==0.10.4
+# homeassistant.components.sensorpush
+sensorpush-ble==1.4.2
+
# homeassistant.components.sentry
sentry-sdk==1.7.2
diff --git a/requirements_test_all.txt b/requirements_test_all.txt
index 57787b0cb72..605eb8352ea 100644
--- a/requirements_test_all.txt
+++ b/requirements_test_all.txt
@@ -1438,6 +1438,9 @@ securetar==2022.2.0
# homeassistant.components.sense
sense_energy==0.10.4
+# homeassistant.components.sensorpush
+sensorpush-ble==1.4.2
+
# homeassistant.components.sentry
sentry-sdk==1.7.2
diff --git a/tests/components/sensorpush/__init__.py b/tests/components/sensorpush/__init__.py
new file mode 100644
index 00000000000..0fe9ced64df
--- /dev/null
+++ b/tests/components/sensorpush/__init__.py
@@ -0,0 +1,34 @@
+"""Tests for the SensorPush integration."""
+
+
+from homeassistant.helpers.service_info.bluetooth import BluetoothServiceInfo
+
+NOT_SENSOR_PUSH_SERVICE_INFO = BluetoothServiceInfo(
+ name="Not it",
+ address="61DE521B-F0BF-9F44-64D4-75BBE1738105",
+ rssi=-63,
+ manufacturer_data={3234: b"\x00\x01"},
+ service_data={},
+ service_uuids=[],
+ source="local",
+)
+
+HTW_SERVICE_INFO = BluetoothServiceInfo(
+ name="SensorPush HT.w 0CA1",
+ address="61DE521B-F0BF-9F44-64D4-75BBE1738105",
+ rssi=-63,
+ manufacturer_data={11271: b"\xfe\x00\x01"},
+ service_data={},
+ service_uuids=["ef090000-11d6-42ba-93b8-9dd7ec090ab0"],
+ source="local",
+)
+
+HTPWX_SERVICE_INFO = BluetoothServiceInfo(
+ name="SensorPush HTP.xw F4D",
+ address="4125DDBA-2774-4851-9889-6AADDD4CAC3D",
+ rssi=-56,
+ manufacturer_data={7168: b"\xcd=!\xd1\xb9"},
+ service_data={},
+ service_uuids=["ef090000-11d6-42ba-93b8-9dd7ec090ab0"],
+ source="local",
+)
diff --git a/tests/components/sensorpush/conftest.py b/tests/components/sensorpush/conftest.py
new file mode 100644
index 00000000000..c6497a4e76d
--- /dev/null
+++ b/tests/components/sensorpush/conftest.py
@@ -0,0 +1,8 @@
+"""SensorPush session fixtures."""
+
+import pytest
+
+
+@pytest.fixture(autouse=True)
+def auto_mock_bleak_scanner_start(mock_bleak_scanner_start):
+ """Auto mock bleak scanner start."""
diff --git a/tests/components/sensorpush/test_config_flow.py b/tests/components/sensorpush/test_config_flow.py
new file mode 100644
index 00000000000..662841af3f1
--- /dev/null
+++ b/tests/components/sensorpush/test_config_flow.py
@@ -0,0 +1,98 @@
+"""Test the SensorPush config flow."""
+
+from unittest.mock import patch
+
+from homeassistant import config_entries
+from homeassistant.components.sensorpush.const import DOMAIN
+from homeassistant.data_entry_flow import FlowResultType
+
+from . import HTPWX_SERVICE_INFO, HTW_SERVICE_INFO, NOT_SENSOR_PUSH_SERVICE_INFO
+
+from tests.common import MockConfigEntry
+
+
+async def test_async_step_bluetooth_valid_device(hass):
+ """Test discovery via bluetooth with a valid device."""
+ result = await hass.config_entries.flow.async_init(
+ DOMAIN,
+ context={"source": config_entries.SOURCE_BLUETOOTH},
+ data=HTPWX_SERVICE_INFO,
+ )
+ assert result["type"] == FlowResultType.FORM
+ assert result["step_id"] == "bluetooth_confirm"
+ with patch(
+ "homeassistant.components.sensorpush.async_setup_entry", return_value=True
+ ):
+ result2 = await hass.config_entries.flow.async_configure(
+ result["flow_id"], user_input={}
+ )
+ assert result2["type"] == FlowResultType.CREATE_ENTRY
+ assert result2["title"] == "HTP.xw F4D"
+ assert result2["data"] == {}
+ assert result2["result"].unique_id == "4125DDBA-2774-4851-9889-6AADDD4CAC3D"
+
+
+async def test_async_step_bluetooth_not_sensorpush(hass):
+ """Test discovery via bluetooth not sensorpush."""
+ result = await hass.config_entries.flow.async_init(
+ DOMAIN,
+ context={"source": config_entries.SOURCE_BLUETOOTH},
+ data=NOT_SENSOR_PUSH_SERVICE_INFO,
+ )
+ assert result["type"] == FlowResultType.ABORT
+ assert result["reason"] == "not_supported"
+
+
+async def test_async_step_user_no_devices_found(hass):
+ """Test setup from service info cache with no devices found."""
+ result = await hass.config_entries.flow.async_init(
+ DOMAIN,
+ context={"source": config_entries.SOURCE_USER},
+ )
+ assert result["type"] == FlowResultType.ABORT
+ assert result["reason"] == "no_devices_found"
+
+
+async def test_async_step_user_with_found_devices(hass):
+ """Test setup from service info cache with devices found."""
+ with patch(
+ "homeassistant.components.sensorpush.config_flow.async_discovered_service_info",
+ return_value=[HTW_SERVICE_INFO],
+ ):
+ result = await hass.config_entries.flow.async_init(
+ DOMAIN,
+ context={"source": config_entries.SOURCE_USER},
+ )
+ assert result["type"] == FlowResultType.FORM
+ assert result["step_id"] == "user"
+ with patch(
+ "homeassistant.components.sensorpush.async_setup_entry", return_value=True
+ ):
+ result2 = await hass.config_entries.flow.async_configure(
+ result["flow_id"],
+ user_input={"address": "61DE521B-F0BF-9F44-64D4-75BBE1738105"},
+ )
+ assert result2["type"] == FlowResultType.CREATE_ENTRY
+ assert result2["title"] == "HT.w 0CA1"
+ assert result2["data"] == {}
+ assert result2["result"].unique_id == "61DE521B-F0BF-9F44-64D4-75BBE1738105"
+
+
+async def test_async_step_user_with_found_devices_already_setup(hass):
+ """Test setup from service info cache with devices found."""
+ entry = MockConfigEntry(
+ domain=DOMAIN,
+ unique_id="61DE521B-F0BF-9F44-64D4-75BBE1738105",
+ )
+ entry.add_to_hass(hass)
+
+ with patch(
+ "homeassistant.components.sensorpush.config_flow.async_discovered_service_info",
+ return_value=[HTW_SERVICE_INFO],
+ ):
+ result = await hass.config_entries.flow.async_init(
+ DOMAIN,
+ context={"source": config_entries.SOURCE_USER},
+ )
+ assert result["type"] == FlowResultType.ABORT
+ assert result["reason"] == "no_devices_found"
diff --git a/tests/components/sensorpush/test_sensor.py b/tests/components/sensorpush/test_sensor.py
new file mode 100644
index 00000000000..c48b8bc3407
--- /dev/null
+++ b/tests/components/sensorpush/test_sensor.py
@@ -0,0 +1,50 @@
+"""Test the SensorPush config flow."""
+
+from unittest.mock import patch
+
+from homeassistant.components.bluetooth import BluetoothChange
+from homeassistant.components.sensor import ATTR_STATE_CLASS
+from homeassistant.components.sensorpush.const import DOMAIN
+from homeassistant.const import ATTR_FRIENDLY_NAME, ATTR_UNIT_OF_MEASUREMENT
+
+from . import HTPWX_SERVICE_INFO
+
+from tests.common import MockConfigEntry
+
+
+async def test_sensors(hass):
+ """Test setting up creates the sensors."""
+ entry = MockConfigEntry(
+ domain=DOMAIN,
+ unique_id="61DE521B-F0BF-9F44-64D4-75BBE1738105",
+ )
+ entry.add_to_hass(hass)
+
+ saved_callback = None
+
+ def _async_register_callback(_hass, _callback, _matcher):
+ nonlocal saved_callback
+ saved_callback = _callback
+ return lambda: None
+
+ with patch(
+ "homeassistant.components.bluetooth.passive_update_coordinator.async_register_callback",
+ _async_register_callback,
+ ):
+ assert await hass.config_entries.async_setup(entry.entry_id)
+ await hass.async_block_till_done()
+
+ assert len(hass.states.async_all()) == 0
+ saved_callback(HTPWX_SERVICE_INFO, BluetoothChange.ADVERTISEMENT)
+ await hass.async_block_till_done()
+ assert len(hass.states.async_all()) == 3
+
+ temp_sensor = hass.states.get("sensor.htp_xw_f4d_temperature")
+ temp_sensor_attribtes = temp_sensor.attributes
+ assert temp_sensor.state == "20.11"
+ assert temp_sensor_attribtes[ATTR_FRIENDLY_NAME] == "HTP.xw F4D Temperature"
+ assert temp_sensor_attribtes[ATTR_UNIT_OF_MEASUREMENT] == "°C"
+ assert temp_sensor_attribtes[ATTR_STATE_CLASS] == "measurement"
+
+ assert await hass.config_entries.async_unload(entry.entry_id)
+ await hass.async_block_till_done()
--
GitLab