mirror of
https://github.com/home-assistant/core.git
synced 2025-07-18 18:57:06 +00:00
Add Govee BLE integration (#75631)
* Add Govee BLE integration * add missing files * remove test file not needed yet * fix * add bbq sensors * fixed lib * bump again to fix the names * fix discovery of the newer bbq devices * fix the test to test the right thing * verify no outstanding flows * only accept entities that match the platform * refactor * refactor * refactor * Refactor PassiveBluetoothDataUpdateCoordinator to support multiple platforms * cover * Update for new model * Update for new model * Update tests/components/govee_ble/test_sensor.py Co-authored-by: Martin Hjelmare <marhje52@gmail.com> * purge dead code * backmerge from integration * Update docstring * Update docstring Co-authored-by: Martin Hjelmare <marhje52@gmail.com>
This commit is contained in:
parent
7075032bf7
commit
ba71a3c24d
@ -411,6 +411,8 @@ build.json @home-assistant/supervisor
|
|||||||
/homeassistant/components/google_cloud/ @lufton
|
/homeassistant/components/google_cloud/ @lufton
|
||||||
/homeassistant/components/google_travel_time/ @eifinger
|
/homeassistant/components/google_travel_time/ @eifinger
|
||||||
/tests/components/google_travel_time/ @eifinger
|
/tests/components/google_travel_time/ @eifinger
|
||||||
|
/homeassistant/components/govee_ble/ @bdraco
|
||||||
|
/tests/components/govee_ble/ @bdraco
|
||||||
/homeassistant/components/gpsd/ @fabaff
|
/homeassistant/components/gpsd/ @fabaff
|
||||||
/homeassistant/components/gree/ @cmroche
|
/homeassistant/components/gree/ @cmroche
|
||||||
/tests/components/gree/ @cmroche
|
/tests/components/gree/ @cmroche
|
||||||
|
40
homeassistant/components/govee_ble/__init__.py
Normal file
40
homeassistant/components/govee_ble/__init__.py
Normal file
@ -0,0 +1,40 @@
|
|||||||
|
"""The Govee Bluetooth BLE integration."""
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
import logging
|
||||||
|
|
||||||
|
from homeassistant.components.bluetooth.passive_update_processor import (
|
||||||
|
PassiveBluetoothProcessorCoordinator,
|
||||||
|
)
|
||||||
|
from homeassistant.config_entries import ConfigEntry
|
||||||
|
from homeassistant.const import Platform
|
||||||
|
from homeassistant.core import HomeAssistant
|
||||||
|
|
||||||
|
from .const import DOMAIN
|
||||||
|
|
||||||
|
PLATFORMS: list[Platform] = [Platform.SENSOR]
|
||||||
|
|
||||||
|
_LOGGER = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
|
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
||||||
|
"""Set up Govee BLE device from a config entry."""
|
||||||
|
address = entry.unique_id
|
||||||
|
assert address is not None
|
||||||
|
hass.data.setdefault(DOMAIN, {})[
|
||||||
|
entry.entry_id
|
||||||
|
] = PassiveBluetoothProcessorCoordinator(
|
||||||
|
hass,
|
||||||
|
_LOGGER,
|
||||||
|
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
|
93
homeassistant/components/govee_ble/config_flow.py
Normal file
93
homeassistant/components/govee_ble/config_flow.py
Normal file
@ -0,0 +1,93 @@
|
|||||||
|
"""Config flow for govee ble integration."""
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
from typing import Any
|
||||||
|
|
||||||
|
from govee_ble import GoveeBluetoothDeviceData 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 GoveeConfigFlow(ConfigFlow, domain=DOMAIN):
|
||||||
|
"""Handle a config flow for govee."""
|
||||||
|
|
||||||
|
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)}
|
||||||
|
),
|
||||||
|
)
|
3
homeassistant/components/govee_ble/const.py
Normal file
3
homeassistant/components/govee_ble/const.py
Normal file
@ -0,0 +1,3 @@
|
|||||||
|
"""Constants for the Govee Bluetooth integration."""
|
||||||
|
|
||||||
|
DOMAIN = "govee_ble"
|
27
homeassistant/components/govee_ble/manifest.json
Normal file
27
homeassistant/components/govee_ble/manifest.json
Normal file
@ -0,0 +1,27 @@
|
|||||||
|
{
|
||||||
|
"domain": "govee_ble",
|
||||||
|
"name": "Govee Bluetooth",
|
||||||
|
"config_flow": true,
|
||||||
|
"documentation": "https://www.home-assistant.io/integrations/govee_ble",
|
||||||
|
"bluetooth": [
|
||||||
|
{ "local_name": "Govee*" },
|
||||||
|
{ "local_name": "GVH5*" },
|
||||||
|
{ "local_name": "B5178*" },
|
||||||
|
{
|
||||||
|
"manufacturer_id": 26589,
|
||||||
|
"service_uuid": "00008351-0000-1000-8000-00805f9b34fb"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"manufacturer_id": 18994,
|
||||||
|
"service_uuid": "00008551-0000-1000-8000-00805f9b34fb"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"manufacturer_id": 14474,
|
||||||
|
"service_uuid": "00008151-0000-1000-8000-00805f9b34fb"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"requirements": ["govee-ble==0.12.3"],
|
||||||
|
"dependencies": ["bluetooth"],
|
||||||
|
"codeowners": ["@bdraco"],
|
||||||
|
"iot_class": "local_push"
|
||||||
|
}
|
157
homeassistant/components/govee_ble/sensor.py
Normal file
157
homeassistant/components/govee_ble/sensor.py
Normal file
@ -0,0 +1,157 @@
|
|||||||
|
"""Support for govee ble sensors."""
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
from typing import Optional, Union
|
||||||
|
|
||||||
|
from govee_ble import (
|
||||||
|
DeviceClass,
|
||||||
|
DeviceKey,
|
||||||
|
GoveeBluetoothDeviceData,
|
||||||
|
SensorDeviceInfo,
|
||||||
|
SensorUpdate,
|
||||||
|
Units,
|
||||||
|
)
|
||||||
|
|
||||||
|
from homeassistant import config_entries
|
||||||
|
from homeassistant.components.bluetooth.passive_update_processor import (
|
||||||
|
PassiveBluetoothDataProcessor,
|
||||||
|
PassiveBluetoothDataUpdate,
|
||||||
|
PassiveBluetoothEntityKey,
|
||||||
|
PassiveBluetoothProcessorCoordinator,
|
||||||
|
PassiveBluetoothProcessorEntity,
|
||||||
|
)
|
||||||
|
from homeassistant.components.sensor import (
|
||||||
|
SensorDeviceClass,
|
||||||
|
SensorEntity,
|
||||||
|
SensorEntityDescription,
|
||||||
|
SensorStateClass,
|
||||||
|
)
|
||||||
|
from homeassistant.const import (
|
||||||
|
ATTR_MANUFACTURER,
|
||||||
|
ATTR_MODEL,
|
||||||
|
ATTR_NAME,
|
||||||
|
PERCENTAGE,
|
||||||
|
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.BATTERY, Units.PERCENTAGE): SensorEntityDescription(
|
||||||
|
key=f"{DeviceClass.BATTERY}_{Units.PERCENTAGE}",
|
||||||
|
device_class=SensorDeviceClass.BATTERY,
|
||||||
|
native_unit_of_measurement=PERCENTAGE,
|
||||||
|
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 hass 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 Govee BLE sensors."""
|
||||||
|
coordinator: PassiveBluetoothProcessorCoordinator = hass.data[DOMAIN][
|
||||||
|
entry.entry_id
|
||||||
|
]
|
||||||
|
data = GoveeBluetoothDeviceData()
|
||||||
|
processor = PassiveBluetoothDataProcessor(
|
||||||
|
lambda service_info: sensor_update_to_bluetooth_data_update(
|
||||||
|
data.update(service_info)
|
||||||
|
)
|
||||||
|
)
|
||||||
|
entry.async_on_unload(coordinator.async_register_processor(processor))
|
||||||
|
entry.async_on_unload(
|
||||||
|
processor.async_add_entities_listener(
|
||||||
|
GoveeBluetoothSensorEntity, async_add_entities
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class GoveeBluetoothSensorEntity(
|
||||||
|
PassiveBluetoothProcessorEntity[
|
||||||
|
PassiveBluetoothDataProcessor[Optional[Union[float, int]]]
|
||||||
|
],
|
||||||
|
SensorEntity,
|
||||||
|
):
|
||||||
|
"""Representation of a govee ble sensor."""
|
||||||
|
|
||||||
|
@property
|
||||||
|
def native_value(self) -> int | float | None:
|
||||||
|
"""Return the native value."""
|
||||||
|
return self.processor.entity_data.get(self.entity_key)
|
21
homeassistant/components/govee_ble/strings.json
Normal file
21
homeassistant/components/govee_ble/strings.json
Normal file
@ -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%]"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
21
homeassistant/components/govee_ble/translations/en.json
Normal file
21
homeassistant/components/govee_ble/translations/en.json
Normal file
@ -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"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
@ -7,6 +7,33 @@ from __future__ import annotations
|
|||||||
# fmt: off
|
# fmt: off
|
||||||
|
|
||||||
BLUETOOTH: list[dict[str, str | int | list[int]]] = [
|
BLUETOOTH: list[dict[str, str | int | list[int]]] = [
|
||||||
|
{
|
||||||
|
"domain": "govee_ble",
|
||||||
|
"local_name": "Govee*"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"domain": "govee_ble",
|
||||||
|
"local_name": "GVH5*"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"domain": "govee_ble",
|
||||||
|
"local_name": "B5178*"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"domain": "govee_ble",
|
||||||
|
"manufacturer_id": 26589,
|
||||||
|
"service_uuid": "00008351-0000-1000-8000-00805f9b34fb"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"domain": "govee_ble",
|
||||||
|
"manufacturer_id": 18994,
|
||||||
|
"service_uuid": "00008551-0000-1000-8000-00805f9b34fb"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"domain": "govee_ble",
|
||||||
|
"manufacturer_id": 14474,
|
||||||
|
"service_uuid": "00008151-0000-1000-8000-00805f9b34fb"
|
||||||
|
},
|
||||||
{
|
{
|
||||||
"domain": "homekit_controller",
|
"domain": "homekit_controller",
|
||||||
"manufacturer_id": 76,
|
"manufacturer_id": 76,
|
||||||
|
@ -138,6 +138,7 @@ FLOWS = {
|
|||||||
"goodwe",
|
"goodwe",
|
||||||
"google",
|
"google",
|
||||||
"google_travel_time",
|
"google_travel_time",
|
||||||
|
"govee_ble",
|
||||||
"gpslogger",
|
"gpslogger",
|
||||||
"gree",
|
"gree",
|
||||||
"growatt_server",
|
"growatt_server",
|
||||||
|
@ -760,6 +760,9 @@ googlemaps==2.5.1
|
|||||||
# homeassistant.components.slide
|
# homeassistant.components.slide
|
||||||
goslide-api==0.5.1
|
goslide-api==0.5.1
|
||||||
|
|
||||||
|
# homeassistant.components.govee_ble
|
||||||
|
govee-ble==0.12.3
|
||||||
|
|
||||||
# homeassistant.components.remote_rpi_gpio
|
# homeassistant.components.remote_rpi_gpio
|
||||||
gpiozero==1.6.2
|
gpiozero==1.6.2
|
||||||
|
|
||||||
|
@ -560,6 +560,9 @@ google-nest-sdm==2.0.0
|
|||||||
# homeassistant.components.google_travel_time
|
# homeassistant.components.google_travel_time
|
||||||
googlemaps==2.5.1
|
googlemaps==2.5.1
|
||||||
|
|
||||||
|
# homeassistant.components.govee_ble
|
||||||
|
govee-ble==0.12.3
|
||||||
|
|
||||||
# homeassistant.components.gree
|
# homeassistant.components.gree
|
||||||
greeclimate==1.2.0
|
greeclimate==1.2.0
|
||||||
|
|
||||||
|
38
tests/components/govee_ble/__init__.py
Normal file
38
tests/components/govee_ble/__init__.py
Normal file
@ -0,0 +1,38 @@
|
|||||||
|
"""Tests for the Govee BLE integration."""
|
||||||
|
|
||||||
|
|
||||||
|
from homeassistant.helpers.service_info.bluetooth import BluetoothServiceInfo
|
||||||
|
|
||||||
|
NOT_GOVEE_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",
|
||||||
|
)
|
||||||
|
|
||||||
|
GVH5075_SERVICE_INFO = BluetoothServiceInfo(
|
||||||
|
name="GVH5075_2762",
|
||||||
|
address="61DE521B-F0BF-9F44-64D4-75BBE1738105",
|
||||||
|
rssi=-63,
|
||||||
|
manufacturer_data={
|
||||||
|
60552: b"\x00\x03A\xc2d\x00L\x00\x02\x15INTELLI_ROCKS_HWPu\xf2\xff\x0c"
|
||||||
|
},
|
||||||
|
service_uuids=["0000ec88-0000-1000-8000-00805f9b34fb"],
|
||||||
|
service_data={},
|
||||||
|
source="local",
|
||||||
|
)
|
||||||
|
|
||||||
|
GVH5177_SERVICE_INFO = BluetoothServiceInfo(
|
||||||
|
name="GVH5177_2EC8",
|
||||||
|
address="4125DDBA-2774-4851-9889-6AADDD4CAC3D",
|
||||||
|
rssi=-56,
|
||||||
|
manufacturer_data={
|
||||||
|
1: b"\x01\x01\x036&dL\x00\x02\x15INTELLI_ROCKS_HWQw\xf2\xff\xc2"
|
||||||
|
},
|
||||||
|
service_uuids=["0000ec88-0000-1000-8000-00805f9b34fb"],
|
||||||
|
service_data={},
|
||||||
|
source="local",
|
||||||
|
)
|
8
tests/components/govee_ble/conftest.py
Normal file
8
tests/components/govee_ble/conftest.py
Normal file
@ -0,0 +1,8 @@
|
|||||||
|
"""Govee session fixtures."""
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture(autouse=True)
|
||||||
|
def mock_bluetooth(enable_bluetooth):
|
||||||
|
"""Auto mock bluetooth."""
|
170
tests/components/govee_ble/test_config_flow.py
Normal file
170
tests/components/govee_ble/test_config_flow.py
Normal file
@ -0,0 +1,170 @@
|
|||||||
|
"""Test the Govee config flow."""
|
||||||
|
|
||||||
|
from unittest.mock import patch
|
||||||
|
|
||||||
|
from homeassistant import config_entries
|
||||||
|
from homeassistant.components.govee_ble.const import DOMAIN
|
||||||
|
from homeassistant.data_entry_flow import FlowResultType
|
||||||
|
|
||||||
|
from . import GVH5075_SERVICE_INFO, GVH5177_SERVICE_INFO, NOT_GOVEE_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=GVH5075_SERVICE_INFO,
|
||||||
|
)
|
||||||
|
assert result["type"] == FlowResultType.FORM
|
||||||
|
assert result["step_id"] == "bluetooth_confirm"
|
||||||
|
with patch(
|
||||||
|
"homeassistant.components.govee_ble.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"] == "H5075_2762"
|
||||||
|
assert result2["data"] == {}
|
||||||
|
assert result2["result"].unique_id == "61DE521B-F0BF-9F44-64D4-75BBE1738105"
|
||||||
|
|
||||||
|
|
||||||
|
async def test_async_step_bluetooth_not_govee(hass):
|
||||||
|
"""Test discovery via bluetooth not govee."""
|
||||||
|
result = await hass.config_entries.flow.async_init(
|
||||||
|
DOMAIN,
|
||||||
|
context={"source": config_entries.SOURCE_BLUETOOTH},
|
||||||
|
data=NOT_GOVEE_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.govee_ble.config_flow.async_discovered_service_info",
|
||||||
|
return_value=[GVH5177_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.govee_ble.async_setup_entry", return_value=True
|
||||||
|
):
|
||||||
|
result2 = await hass.config_entries.flow.async_configure(
|
||||||
|
result["flow_id"],
|
||||||
|
user_input={"address": "4125DDBA-2774-4851-9889-6AADDD4CAC3D"},
|
||||||
|
)
|
||||||
|
assert result2["type"] == FlowResultType.CREATE_ENTRY
|
||||||
|
assert result2["title"] == "H5177_2EC8"
|
||||||
|
assert result2["data"] == {}
|
||||||
|
assert result2["result"].unique_id == "4125DDBA-2774-4851-9889-6AADDD4CAC3D"
|
||||||
|
|
||||||
|
|
||||||
|
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="4125DDBA-2774-4851-9889-6AADDD4CAC3D",
|
||||||
|
)
|
||||||
|
entry.add_to_hass(hass)
|
||||||
|
|
||||||
|
with patch(
|
||||||
|
"homeassistant.components.govee_ble.config_flow.async_discovered_service_info",
|
||||||
|
return_value=[GVH5177_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"
|
||||||
|
|
||||||
|
|
||||||
|
async def test_async_step_bluetooth_devices_already_setup(hass):
|
||||||
|
"""Test we can't start a flow if there is already a config entry."""
|
||||||
|
entry = MockConfigEntry(
|
||||||
|
domain=DOMAIN,
|
||||||
|
unique_id="4125DDBA-2774-4851-9889-6AADDD4CAC3D",
|
||||||
|
)
|
||||||
|
entry.add_to_hass(hass)
|
||||||
|
|
||||||
|
result = await hass.config_entries.flow.async_init(
|
||||||
|
DOMAIN,
|
||||||
|
context={"source": config_entries.SOURCE_BLUETOOTH},
|
||||||
|
data=GVH5177_SERVICE_INFO,
|
||||||
|
)
|
||||||
|
assert result["type"] == FlowResultType.ABORT
|
||||||
|
assert result["reason"] == "already_configured"
|
||||||
|
|
||||||
|
|
||||||
|
async def test_async_step_bluetooth_already_in_progress(hass):
|
||||||
|
"""Test we can't start a flow for the same device twice."""
|
||||||
|
result = await hass.config_entries.flow.async_init(
|
||||||
|
DOMAIN,
|
||||||
|
context={"source": config_entries.SOURCE_BLUETOOTH},
|
||||||
|
data=GVH5177_SERVICE_INFO,
|
||||||
|
)
|
||||||
|
assert result["type"] == FlowResultType.FORM
|
||||||
|
assert result["step_id"] == "bluetooth_confirm"
|
||||||
|
|
||||||
|
result = await hass.config_entries.flow.async_init(
|
||||||
|
DOMAIN,
|
||||||
|
context={"source": config_entries.SOURCE_BLUETOOTH},
|
||||||
|
data=GVH5177_SERVICE_INFO,
|
||||||
|
)
|
||||||
|
assert result["type"] == FlowResultType.ABORT
|
||||||
|
assert result["reason"] == "already_in_progress"
|
||||||
|
|
||||||
|
|
||||||
|
async def test_async_step_user_takes_precedence_over_discovery(hass):
|
||||||
|
"""Test manual setup takes precedence over discovery."""
|
||||||
|
result = await hass.config_entries.flow.async_init(
|
||||||
|
DOMAIN,
|
||||||
|
context={"source": config_entries.SOURCE_BLUETOOTH},
|
||||||
|
data=GVH5177_SERVICE_INFO,
|
||||||
|
)
|
||||||
|
assert result["type"] == FlowResultType.FORM
|
||||||
|
assert result["step_id"] == "bluetooth_confirm"
|
||||||
|
|
||||||
|
with patch(
|
||||||
|
"homeassistant.components.govee_ble.config_flow.async_discovered_service_info",
|
||||||
|
return_value=[GVH5177_SERVICE_INFO],
|
||||||
|
):
|
||||||
|
result = await hass.config_entries.flow.async_init(
|
||||||
|
DOMAIN,
|
||||||
|
context={"source": config_entries.SOURCE_USER},
|
||||||
|
)
|
||||||
|
assert result["type"] == FlowResultType.FORM
|
||||||
|
|
||||||
|
with patch(
|
||||||
|
"homeassistant.components.govee_ble.async_setup_entry", return_value=True
|
||||||
|
):
|
||||||
|
result2 = await hass.config_entries.flow.async_configure(
|
||||||
|
result["flow_id"],
|
||||||
|
user_input={"address": "4125DDBA-2774-4851-9889-6AADDD4CAC3D"},
|
||||||
|
)
|
||||||
|
assert result2["type"] == FlowResultType.CREATE_ENTRY
|
||||||
|
assert result2["title"] == "H5177_2EC8"
|
||||||
|
assert result2["data"] == {}
|
||||||
|
assert result2["result"].unique_id == "4125DDBA-2774-4851-9889-6AADDD4CAC3D"
|
||||||
|
|
||||||
|
# Verify the original one was aborted
|
||||||
|
assert not hass.config_entries.flow.async_progress(DOMAIN)
|
50
tests/components/govee_ble/test_sensor.py
Normal file
50
tests/components/govee_ble/test_sensor.py
Normal file
@ -0,0 +1,50 @@
|
|||||||
|
"""Test the Govee BLE sensors."""
|
||||||
|
|
||||||
|
from unittest.mock import patch
|
||||||
|
|
||||||
|
from homeassistant.components.bluetooth import BluetoothChange
|
||||||
|
from homeassistant.components.govee_ble.const import DOMAIN
|
||||||
|
from homeassistant.components.sensor import ATTR_STATE_CLASS
|
||||||
|
from homeassistant.const import ATTR_FRIENDLY_NAME, ATTR_UNIT_OF_MEASUREMENT
|
||||||
|
|
||||||
|
from . import GVH5075_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.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(GVH5075_SERVICE_INFO, BluetoothChange.ADVERTISEMENT)
|
||||||
|
await hass.async_block_till_done()
|
||||||
|
assert len(hass.states.async_all()) == 3
|
||||||
|
|
||||||
|
temp_sensor = hass.states.get("sensor.h5075_2762_temperature")
|
||||||
|
temp_sensor_attribtes = temp_sensor.attributes
|
||||||
|
assert temp_sensor.state == "21.3442"
|
||||||
|
assert temp_sensor_attribtes[ATTR_FRIENDLY_NAME] == "H5075_2762 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()
|
Loading…
x
Reference in New Issue
Block a user