Update CHANGELOG.md

### Fix
* Bumping pyowletapi to 2023.11.1
* Sensors and binary sensors are now only created where the sock contains that property, this stops errors where different sock versions have different properties

.github/workflows/cron.yml

@@ -0,0 +1,25 @@
+name: Validate with hassfest
+
+on:
+  push:
+  pull_request:
+  schedule:
+    - cron:  '0 0 * * *'
+
+jobs:
+  hassfest:
+    name: "Hassfest validation"
+    runs-on: "ubuntu-latest"
+    steps:
+        - uses: "actions/checkout@v3"
+        - uses: "home-assistant/actions/hassfest@master"
+  hacs:
+    name: "HACS Action"
+    runs-on: "ubuntu-latest"
+    steps:
+      - name: "HACS Action"
+        uses: "hacs/action@main"
+        with:
+            category: "integration"
+        
+        

CHANGELOG.md

@@ -1,12 +1,66 @@
 # Changelog
 
 <!--next-version-placeholder-->
-## 2023-05-2 (2023-05-16)
+## 2023.11.1 (2023-11-16)
+### Fix
+* Bumping pyowletapi to 2023.11.1 ([`3acf847`](https://github.com/ryanbdclark/owlet/commit/3acf8473526665382b44ef6325d708a6c62fff45))
+* Sensors and binary sensors are now only created where the sock contains that property, this stops errors where different sock versions have different properties ([`3acf847`](https://github.com/ryanbdclark/owlet/commit/3acf8473526665382b44ef6325d708a6c62fff45))
+
+## 2023.9.1 (2023-09-20)
+### Fix
+* Bumping pyowletapi to 2023.9.1 to allow for revisions ([`0a7f703`](https://github.com/ryanbdclark/owlet/commit/0a7f70310080a129c988e9607331baa2f6c691e0))
+* New revision of sock, revision 5 doesn't report all vitals as before, this would cause the integration to fail to update. Have adjusted the integration to detect the revision and ignore the vitals that are no longer reported ([`0a7f703`](https://github.com/ryanbdclark/owlet/commit/0a7f70310080a129c988e9607331baa2f6c691e0))
+
+
+## 2023.8.1 (2023-08-21)
+### Feature
+* 2 new sensors, movement and movement bucket disabled by default, thanks [`@seanford`](https://github.com/seanford) ([`575b213`](https://github.com/ryanbdclark/owlet/commit/575b213ddd732779cd7938e575fc87c8881a69b0))
+
+### Fix
+* Various refactoring tasks completed to make this integration more inline with home assistants style guidelines ([`2cd46c1`](https://github.com/ryanbdclark/owlet/commit/c45959b123a6e5f77747475f11d3d3ab67859756))
+* Added new sensors to strings jsons ([`2cd46c1`](https://github.com/ryanbdclark/owlet/commit/c45959b123a6e5f77747475f11d3d3ab67859756))
+
+## 2023.7.2 (2023-07-04)
+### Fix
+* Bumping pyowletapi version to 2023.7.2 ([`c45959b`](https://github.com/ryanbdclark/owlet/commit/c45959b123a6e5f77747475f11d3d3ab67859756))
+
+## 2023.7.1 (2023-07-03)
+### Fix
+*  Bumping pyowletapi to 2023.7.1 ([`c693fef`](https://github.com/ryanbdclark/owlet/commit/c693fefbf3dba8f35802b87d064401dadbb211b5))
+  
+## 2023.5.7 (2023-05-30)
+### Fix
+* Fixed issue with binary sensors not loading, caused by change to way the coordinators are stored ([`8d17317`](https://github.com/ryanbdclark/owlet/commit/8d173174e286b0451cbb2c0d4ae3087028d1ea23))
+
+## 2023.5.6 (2023-05-30)
+### Fix
+* In light of submitting this as a pull request to the core of HA there have been some refactoring changes to comply with HA's style requirements
+* Sensor names now moved to strings file to allow for translations
+* Coordinator now properly handles multiple devices
+* Spelling of signal strength sensor corrected
+
+### Feature
+* Tests added
+
+## 2023.5.5 (2023-05-19)
+#### Fix
+* Owlet refresh token becomes invalid after 24 hours. Meant that after 1 day the integration would stop working. Moved to pyowletapi v2023.5.28 which uses different refresh token, should no longer need reconfiguring after 24 hours ([`dc58b19`](https://github.com/ryanbdclark/owlet/commit/0141f7d01a9ac9b3e1dcc74cabb896e19bd4a821))
+
+## 2023.5.4 (2023-05-17)
+#### Fix
+* Bumping to pyowletapi 2023.5.25
+
+## 2023.5.3 (2023-05-17)
+#### Fix
+* Bumping to pyowletapi 2023.5.24
+* Reauthing now no longer re adds users' password to config entry
+
+## 2023.5.2 (2023-05-16)
 #### Feature
 * Integration now makes use of refresh token from pyowletapi to reauthenticate, user password in no longer stored by integration ([`dc710a1`](https://github.com/ryanbdclark/owlet/commit/dc710a1783a4cad9d6cf355240fe12ac779a87ef))
 * New sensors create for baby sleep state ([`9b3392b`](https://github.com/ryanbdclark/owlet/commit/9b3392bdbcd82015ed31d3a50a517e4e22905684))
  
-## 2023-05-1 (2023-05-15)
+## 2023.5.1 (2023-05-15)
 #### Feature
 * Changed versioning to date based
 ### Fix

custom_components/owlet/__init__.py

@@ -1,29 +1,32 @@
 """The Owlet Smart Sock integration."""
 from __future__ import annotations
 
+import asyncio
 import logging
 
 from pyowletapi.api import OwletAPI
+from pyowletapi.exceptions import (
+    OwletAuthenticationError,
+    OwletConnectionError,
+    OwletDevicesError,
+    OwletEmailError,
+    OwletPasswordError,
+)
 from pyowletapi.sock import Sock
-from pyowletapi.exceptions import OwletAuthenticationError
 
-from homeassistant.config_entries import ConfigEntry, ConfigEntryAuthFailed
+from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import (
-    Platform,
-    CONF_REGION,
-    CONF_USERNAME,
-    CONF_PASSWORD,
-    CONF_SCAN_INTERVAL,
     CONF_API_TOKEN,
+    CONF_REGION,
+    CONF_SCAN_INTERVAL,
+    CONF_USERNAME,
+    Platform,
 )
 from homeassistant.core import HomeAssistant
+from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
 from homeassistant.helpers.aiohttp_client import async_get_clientsession
-from .const import (
-    DOMAIN,
-    CONF_OWLET_EXPIRY,
-    CONF_OWLET_REFRESH,
-    SUPPORTED_VERSIONS,
-)
+
+from .const import CONF_OWLET_EXPIRY, CONF_OWLET_REFRESH, DOMAIN, SUPPORTED_VERSIONS
 from .coordinator import OwletCoordinator
 
 PLATFORMS: list[Platform] = [Platform.BINARY_SENSOR, Platform.SENSOR]
@@ -44,30 +47,50 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     )
 
     try:
-        token = await owlet_api.authenticate()
-
-        if token:
+        if token := await owlet_api.authenticate():
             hass.config_entries.async_update_entry(entry, data={**entry.data, **token})
 
-        socks = {
-            device["device"]["dsn"]: Sock(owlet_api, device["device"])
-            for device in await owlet_api.get_devices(SUPPORTED_VERSIONS)
-        }
+        devices = await owlet_api.get_devices(SUPPORTED_VERSIONS)
 
-    except OwletAuthenticationError as err:
+    except (OwletAuthenticationError, OwletEmailError, OwletPasswordError) as err:
         _LOGGER.error("Credentials no longer valid, please setup owlet again")
         raise ConfigEntryAuthFailed(
             f"Credentials expired for {entry.data[CONF_USERNAME]}"
         ) from err
 
-    coordinators = [
-        OwletCoordinator(hass, sock, entry.options.get(CONF_SCAN_INTERVAL))
-        for sock in socks.values()
-    ]
+    except OwletConnectionError as err:
+        raise ConfigEntryNotReady(
+            f"Error connecting to {entry.data[CONF_USERNAME]}"
+        ) from err
 
-    for coordinator in coordinators:
-        await coordinator.async_config_entry_first_refresh()
-        hass.data[DOMAIN][entry.entry_id] = coordinator
+    except OwletDevicesError:
+        _LOGGER.error("No owlet devices found to set up")
+        return False
+
+    if devices["tokens"]:
+        hass.config_entries.async_update_entry(
+            entry, data={**entry.data, **devices["tokens"]}
+        )
+
+    socks = {
+        device["device"]["dsn"]: Sock(owlet_api, device["device"])
+        for device in devices["response"]
+    }
+
+    scan_interval = entry.options.get(CONF_SCAN_INTERVAL)
+    coordinators = {
+        serial: OwletCoordinator(hass, sock, scan_interval, entry)
+        for (serial, sock) in socks.items()
+    }
+
+    await asyncio.gather(
+        *(
+            coordinator.async_config_entry_first_refresh()
+            for coordinator in list(coordinators.values())
+        )
+    )
+
+    hass.data[DOMAIN][entry.entry_id] = coordinators
 
     await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
 

custom_components/owlet/binary_sensor.py

@@ -18,78 +18,59 @@ from .entity import OwletBaseEntity
 
 
 @dataclass
-class OwletBinarySensorEntityMixin:
-    """Owlet binary sensor element mixin"""
-
-    element: str
-
-
-@dataclass
-class OwletBinarySensorEntityDescription(
-    BinarySensorEntityDescription, OwletBinarySensorEntityMixin
-):
+class OwletBinarySensorEntityDescription(BinarySensorEntityDescription):
     """Represent the owlet binary sensor entity description."""
 
 
 SENSORS: tuple[OwletBinarySensorEntityDescription, ...] = (
     OwletBinarySensorEntityDescription(
         key="charging",
-        name="Charging",
+        translation_key="charging",
         device_class=BinarySensorDeviceClass.BATTERY_CHARGING,
-        element="charging",
     ),
     OwletBinarySensorEntityDescription(
-        key="highhr",
-        name="High heart rate alert",
+        key="high_heart_rate_alert",
+        translation_key="high_hr_alrt",
         device_class=BinarySensorDeviceClass.SOUND,
-        element="high_heart_rate_alert",
     ),
     OwletBinarySensorEntityDescription(
-        key="lowhr",
-        name="Low Heart Rate Alert",
+        key="low_heart_rate_alert",
+        translation_key="low_hr_alrt",
         device_class=BinarySensorDeviceClass.SOUND,
-        element="low_heart_rate_alert",
     ),
     OwletBinarySensorEntityDescription(
-        key="higho2",
-        name="High oxygen alert",
+        key="high_oxygen_alert",
+        translation_key="high_ox_alrt",
         device_class=BinarySensorDeviceClass.SOUND,
-        element="high_oxygen_alert",
     ),
     OwletBinarySensorEntityDescription(
-        key="lowo2",
-        name="Low oxygen alert",
+        key="low_oxygen_alert",
+        translation_key="low_ox_alrt",
         device_class=BinarySensorDeviceClass.SOUND,
-        element="low_oxygen_alert",
     ),
     OwletBinarySensorEntityDescription(
-        key="lowbattery",
-        name="Low Battery alert",
+        key="low_battery_alert",
+        translation_key="low_batt_alrt",
         device_class=BinarySensorDeviceClass.SOUND,
-        element="low_battery_alert",
     ),
     OwletBinarySensorEntityDescription(
-        key="lostpower",
-        name="Lost power alert",
+        key="lost_power_alert",
+        translation_key="lost_pwr_alrt",
         device_class=BinarySensorDeviceClass.SOUND,
-        element="lost_power_alert",
     ),
     OwletBinarySensorEntityDescription(
-        key="sockdisconnected",
-        name="Sock disconnected alert",
+        key="sock_disconnected",
+        translation_key="sock_discon_alrt",
         device_class=BinarySensorDeviceClass.SOUND,
-        element="sock_disconnected",
     ),
     OwletBinarySensorEntityDescription(
         key="sock_off",
-        name="Sock off",
+        translation_key="sock_off",
         device_class=BinarySensorDeviceClass.POWER,
-        element="sock_off",
     ),
     OwletBinarySensorEntityDescription(
-        key="awake",
-        name="Awake",
-        element="sleep_state",
+        key="sleep_state",
+        translation_key="awake",
         icon="mdi:sleep",
     ),
 )
@@ -102,12 +83,15 @@ async def async_setup_entry(
 ) -> None:
     """Set up the owlet sensors from config entry."""
 
-    coordinator: OwletCoordinator = hass.data[DOMAIN][config_entry.entry_id]
+    coordinators: OwletCoordinator = hass.data[DOMAIN][config_entry.entry_id].values()
 
-    entities = [OwletBinarySensor(coordinator, sensor) for sensor in SENSORS]
-
-    async_add_entities(entities)
+    sensors = []
+    for coordinator in coordinators:
+        for sensor in SENSORS:
+            if sensor.key in coordinator.sock.properties:
+                sensors.append(OwletBinarySensor(coordinator, sensor))
                 
+    async_add_entities(sensors)
 
 class OwletBinarySensor(OwletBaseEntity, BinarySensorEntity):
     """Representation of an Owlet binary sensor."""
@@ -115,19 +99,19 @@ class OwletBinarySensor(OwletBaseEntity, BinarySensorEntity):
     def __init__(
         self,
         coordinator: OwletCoordinator,
-        sensor_description: OwletBinarySensorEntityDescription,
+        description: OwletBinarySensorEntityDescription,
     ) -> None:
         """Initialize the binary sensor."""
         super().__init__(coordinator)
-        self.entity_description = sensor_description
-        self._attr_unique_id = f"{self.sock.serial}-{self.entity_description.name}"
+        self.entity_description = description
+        self._attr_unique_id = f"{self.sock.serial}-{description.key}"
 
     @property
     def is_on(self) -> bool:
         """Return true if the binary sensor is on."""
-        state = self.sock.properties[self.entity_description.element]
+        state = self.sock.properties[self.entity_description.key]
 
-        if self.entity_description.element == "sleep_state":
+        if self.entity_description.key == "sleep_state":
             if self.sock.properties["charging"]:
                 return None
             if state in [8, 15]:

custom_components/owlet/config_flow.py

@@ -1,47 +1,42 @@
 """Config flow for Owlet Smart Sock integration."""
 from __future__ import annotations
 
+from collections.abc import Mapping
 import logging
 from typing import Any
 
 from pyowletapi.api import OwletAPI
-from pyowletapi.sock import Sock
 from pyowletapi.exceptions import (
+    OwletCredentialsError,
     OwletDevicesError,
     OwletEmailError,
     OwletPasswordError,
 )
-
+from pyowletapi.sock import Sock
 import voluptuous as vol
 
 from homeassistant import config_entries, exceptions
-from homeassistant.data_entry_flow import FlowResult
 from homeassistant.config_entries import ConfigEntry
-from homeassistant.helpers.aiohttp_client import async_get_clientsession
-from homeassistant.core import callback
 from homeassistant.const import (
-    CONF_REGION,
-    CONF_USERNAME,
-    CONF_PASSWORD,
-    CONF_SCAN_INTERVAL,
     CONF_API_TOKEN,
+    CONF_PASSWORD,
+    CONF_REGION,
+    CONF_SCAN_INTERVAL,
+    CONF_USERNAME,
 )
+from homeassistant.core import callback
+from homeassistant.data_entry_flow import FlowResult
+from homeassistant.helpers.aiohttp_client import async_get_clientsession
 
-from .const import (
-    DOMAIN,
-    CONF_OWLET_EXPIRY,
-    POLLING_INTERVAL,
-    SUPPORTED_VERSIONS,
-    CONF_OWLET_REFRESH,
-)
+from .const import CONF_OWLET_EXPIRY, CONF_OWLET_REFRESH, DOMAIN, POLLING_INTERVAL
 
 _LOGGER = logging.getLogger(__name__)
 
 STEP_USER_DATA_SCHEMA = vol.Schema(
     {
-        vol.Required("region"): vol.In(["europe", "world"]),
-        vol.Required("username"): str,
-        vol.Required("password"): str,
+        vol.Required(CONF_REGION): vol.In(["europe", "world"]),
+        vol.Required(CONF_USERNAME): str,
+        vol.Required(CONF_PASSWORD): str,
     }
 )
 
@@ -50,14 +45,10 @@ class OwletConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
     """Handle a config flow for Owlet Smart Sock."""
 
     VERSION = 1
+    reauth_entry: ConfigEntry | None = None
 
     def __init__(self) -> None:
-        self._entry: ConfigEntry
-        self._region: str
-        self._username: str
-        self._password: str
-        self._devices: dict[str, Sock]
-        self.reauth_entry: ConfigEntry | None = None
+        """Initialise config flow."""
 
     async def async_step_user(
         self, user_input: dict[str, Any] | None = None
@@ -65,45 +56,43 @@ class OwletConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
         """Handle the initial step."""
         errors: dict[str, str] = {}
         if user_input is not None:
-            self._region = user_input[CONF_REGION]
-            self._username = user_input[CONF_USERNAME]
-            self._password = user_input[CONF_PASSWORD]
-
             owlet_api = OwletAPI(
-                self._region,
-                self._username,
-                self._password,
+                region=user_input[CONF_REGION],
+                user=user_input[CONF_USERNAME],
+                password=user_input[CONF_PASSWORD],
                 session=async_get_clientsession(self.hass),
             )
 
-            await self.async_set_unique_id(self._username.lower())
+            await self.async_set_unique_id(user_input[CONF_USERNAME].lower())
             self._abort_if_unique_id_configured()
 
             try:
                 token = await owlet_api.authenticate()
-                try:
-                    await owlet_api.get_devices(SUPPORTED_VERSIONS)
+                await owlet_api.validate_authentication()
+
+            except OwletDevicesError:
+                errors["base"] = "no_devices"
+            except OwletEmailError:
+                errors[CONF_USERNAME] = "invalid_email"
+            except OwletPasswordError:
+                errors[CONF_PASSWORD] = "invalid_password"
+            except OwletCredentialsError:
+                errors["base"] = "invalid_credentials"
+            except Exception:  # pylint: disable=broad-except
+                _LOGGER.exception("Unexpected exception")
+                errors["base"] = "unknown"
+            else:
                 return self.async_create_entry(
-                        title=self._username,
+                    title=user_input[CONF_USERNAME],
                     data={
-                            CONF_REGION: self._region,
-                            CONF_USERNAME: self._username,
+                        CONF_REGION: user_input[CONF_REGION],
+                        CONF_USERNAME: user_input[CONF_PASSWORD],
                         CONF_API_TOKEN: token[CONF_API_TOKEN],
                         CONF_OWLET_EXPIRY: token[CONF_OWLET_EXPIRY],
                         CONF_OWLET_REFRESH: token[CONF_OWLET_REFRESH],
                     },
                     options={CONF_SCAN_INTERVAL: POLLING_INTERVAL},
                 )
-                except OwletDevicesError:
-                    errors["base"] = "no_devices"
-
-            except OwletEmailError:
-                errors["base"] = "invalid_email"
-            except OwletPasswordError:
-                errors["base"] = "invalid_password"
-            except Exception:  # pylint: disable=broad-except
-                _LOGGER.exception("Unexpected exception")
-                errors["base"] = "unknown"
 
         return self.async_show_form(
             step_id="user", data_schema=STEP_USER_DATA_SCHEMA, errors=errors
@@ -111,19 +100,21 @@ class OwletConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
 
     @staticmethod
     @callback
-    def async_get_options_flow(config_entry):
+    def async_get_options_flow(config_entry: ConfigEntry) -> OptionsFlowHandler:
         """Get the options flow for this handler."""
         return OptionsFlowHandler(config_entry)
 
-    async def async_step_reauth(self, user_input=None):
-        """Handle reauth"""
+    async def async_step_reauth(self, user_input: Mapping[str, Any]) -> FlowResult:
+        """Handle reauth."""
         self.reauth_entry = self.hass.config_entries.async_get_entry(
             self.context["entry_id"]
         )
         return await self.async_step_reauth_confirm()
 
-    async def async_step_reauth_confirm(self, user_input=None):
-        """Dialog that informs the user that reauth is required"""
+    async def async_step_reauth_confirm(
+        self, user_input: dict[str, Any] | None = None
+    ) -> FlowResult:
+        """Dialog that informs the user that reauth is required."""
         assert self.reauth_entry is not None
         errors: dict[str, str] = {}
 
@@ -138,23 +129,18 @@ class OwletConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
             try:
                 token = await owlet_api.authenticate()
                 if token:
-                    user_input[CONF_API_TOKEN] = token[CONF_API_TOKEN]
-                    user_input[CONF_OWLET_EXPIRY] = token[CONF_OWLET_EXPIRY]
-                    user_input[CONF_OWLET_REFRESH] = token[CONF_OWLET_REFRESH]
                     self.hass.config_entries.async_update_entry(
-                    self.reauth_entry, data={**entry_data, **user_input}
+                        self.reauth_entry, data={**entry_data, **token}
                     )
 
                 await self.hass.config_entries.async_reload(self.reauth_entry.entry_id)
 
                 return self.async_abort(reason="reauth_successful")
 
-            except OwletEmailError:
-                errors["base"] = "invalid_email"
             except OwletPasswordError:
-                errors["base"] = "invalid_password"
+                errors[CONF_PASSWORD] = "invalid_password"
             except Exception:  # pylint: disable=broad-except
-                _LOGGER.exception("error reauthing")
+                _LOGGER.exception("Error reauthenticating")
 
         return self.async_show_form(
             step_id="reauth_confirm",
@@ -164,14 +150,16 @@ class OwletConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
 
 
 class OptionsFlowHandler(config_entries.OptionsFlow):
-    """Handle a options flow for owlet"""
+    """Handle a options flow for owlet."""
 
-    def __init__(self, config_entry: config_entries.ConfigEntry) -> None:
-        """Initialise options flow"""
+    def __init__(self, config_entry: ConfigEntry) -> None:
+        """Initialise options flow."""
         self.config_entry = config_entry
 
-    async def async_step_init(self, user_input=None):
-        """Handle options flow"""
+    async def async_step_init(
+        self, user_input: dict[str, Any] | None = None
+    ) -> FlowResult:
+        """Handle options flow."""
         if user_input is not None:
             return self.async_create_entry(title="", data=user_input)
 
@@ -188,4 +176,4 @@ class OptionsFlowHandler(config_entries.OptionsFlow):
 
 
 class InvalidAuth(exceptions.HomeAssistantError):
-    """Error to indiciate there is invalud auth"""
+    """Error to indicate there is invalid auth."""

custom_components/owlet/const.py

@@ -8,4 +8,4 @@ CONF_OWLET_REFRESH = "refresh"
 SUPPORTED_VERSIONS = [3]
 POLLING_INTERVAL = 5
 MANUFACTURER = "Owlet Baby Care"
-SLEEP_STATES = {1, "Awake", 8, "Light Sleep", 15, "Deep Sleep"}
+SLEEP_STATES = {0: "unknown", 1: "awake", 8: "light_sleep", 15: "deep_sleep"}

custom_components/owlet/coordinator.py

@@ -1,22 +1,23 @@
-"""Owlet integration."""
+"""Owlet integration coordinator class."""
 from __future__ import annotations
 
 from datetime import timedelta
 import logging
 
-from pyowletapi.sock import Sock
 from pyowletapi.exceptions import (
-    OwletError,
-    OwletConnectionError,
     OwletAuthenticationError,
+    OwletConnectionError,
+    OwletError,
 )
+from pyowletapi.sock import Sock
 
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_EMAIL
 from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity import DeviceInfo
+from homeassistant.exceptions import ConfigEntryAuthFailed
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
-from homeassistant.const import CONF_API_TOKEN
 
-from .const import DOMAIN, MANUFACTURER, CONF_OWLET_EXPIRY, CONF_OWLET_REFRESH
+from .const import DOMAIN
 
 _LOGGER = logging.getLogger(__name__)
 
@@ -24,7 +25,9 @@ _LOGGER = logging.getLogger(__name__)
 class OwletCoordinator(DataUpdateCoordinator):
     """Coordinator is responsible for querying the device at a specified route."""
 
-    def __init__(self, hass: HomeAssistant, sock: Sock, interval: int) -> None:
+    def __init__(
+        self, hass: HomeAssistant, sock: Sock, interval, entry: ConfigEntry
+    ) -> None:
         """Initialise a custom coordinator."""
         super().__init__(
             hass,
@@ -32,35 +35,21 @@ class OwletCoordinator(DataUpdateCoordinator):
             name=DOMAIN,
             update_interval=timedelta(seconds=interval),
         )
-        assert self.config_entry is not None
-        self._device_unique_id = sock.serial
-        self._model = sock.model
-        self._sw_version = sock.sw_version
-        self._hw_version = sock.version
         self.sock = sock
-        self.device_info = DeviceInfo(
-            identifiers={(DOMAIN, self._device_unique_id)},
-            name="Owlet Baby Care Sock",
-            manufacturer=MANUFACTURER,
-            model=self._model,
-            sw_version=self._sw_version,
-            hw_version=self._hw_version,
-        )
+        self.config_entry = entry
 
     async def _async_update_data(self) -> None:
         """Fetch the data from the device."""
         try:
-            await self.sock.update_properties()
-            tokens = await self.sock.api.tokens_changed(
-                {
-                    CONF_API_TOKEN: self.config_entry.data[CONF_API_TOKEN],
-                    CONF_OWLET_EXPIRY: self.config_entry.data[CONF_OWLET_EXPIRY],
-                    CONF_OWLET_REFRESH: self.config_entry.data[CONF_OWLET_REFRESH],
-                }
-            )
-            if tokens:
+            properties = await self.sock.update_properties()
+            if properties["tokens"]:
                 self.hass.config_entries.async_update_entry(
-                    self.config_entry, data={**self.config_entry.data, **tokens}
+                    self.config_entry,
+                    data={**self.config_entry.data, **properties["tokens"]},
                 )
-        except (OwletError, OwletConnectionError, OwletAuthenticationError) as err:
+        except OwletAuthenticationError as err:
+            raise ConfigEntryAuthFailed(
+                f"Authentication failed for {self.config_entry.data[CONF_EMAIL]}"
+            ) from err
+        except (OwletError, OwletConnectionError) as err:
             raise UpdateFailed(err) from err

custom_components/owlet/entity.py

@@ -2,13 +2,17 @@
 
 from homeassistant.helpers.entity import Entity
 from homeassistant.helpers.update_coordinator import CoordinatorEntity
+from homeassistant.helpers.entity import DeviceInfo
 
 from .coordinator import OwletCoordinator
+from .const import DOMAIN, MANUFACTURER
 
 
 class OwletBaseEntity(CoordinatorEntity[OwletCoordinator], Entity):
     """Base class for Owlet Sock entities."""
 
+    _attr_has_entity_name = True
+
     def __init__(
         self,
         coordinator: OwletCoordinator,
@@ -16,5 +20,15 @@ class OwletBaseEntity(CoordinatorEntity[OwletCoordinator], Entity):
         """Initialize the base entity."""
         super().__init__(coordinator)
         self.sock = coordinator.sock
-        self._attr_device_info = coordinator.device_info
-        self._attr_has_entity_name = True
+
+    @property
+    def device_info(self) -> DeviceInfo:
+        """Return the device info of the device"""
+        return DeviceInfo(
+            identifiers={(DOMAIN, self.sock.serial)},
+            name="Owlet Baby Care Sock",
+            manufacturer=MANUFACTURER,
+            model=self.sock.model,
+            sw_version=self.sock.sw_version,
+            hw_version=f"{self.sock.version}r{self.sock.revision}",
+        )

custom_components/owlet/manifest.json

@@ -5,12 +5,11 @@
     "@ryanbdclark"
   ],
   "config_flow": true,
-  "dependencies": [],
   "documentation": "https://www.home-assistant.io/integrations/owlet",
-  "homekit": {},
   "iot_class": "cloud_polling",
+  "issue_tracker": "https://github.com/ryanbdclark/owlet/issues",
   "requirements": [
-    "pyowletapi==2023.5.23"
+    "pyowletapi==2023.11.1"
   ],
-  "version":"2023.5.2"
+  "version": "2023.11.1"
 }

custom_components/owlet/sensor.py

@@ -1,5 +1,6 @@
-"""Support for Android IP Webcam binary sensors."""
+"""Support for Owlet sensors."""
 from __future__ import annotations
+
 from dataclasses import dataclass
 
 from homeassistant.components.sensor import (
@@ -9,14 +10,15 @@ from homeassistant.components.sensor import (
     SensorStateClass,
 )
 from homeassistant.config_entries import ConfigEntry
-from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from homeassistant.const import (
     PERCENTAGE,
     SIGNAL_STRENGTH_DECIBELS_MILLIWATT,
-    UnitOfTime,
     UnitOfTemperature,
+    UnitOfTime,
 )
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.typing import StateType
 
 from .const import DOMAIN, SLEEP_STATES
 from .coordinator import OwletCoordinator
@@ -24,75 +26,78 @@ from .entity import OwletBaseEntity
 
 
 @dataclass
-class OwletSensorEntityDescriptionMixin:
-    """Owlet sensor description mix in"""
-
-    element: str
-
-
-@dataclass
-class OwletSensorEntityDescription(
-    SensorEntityDescription, OwletSensorEntityDescriptionMixin
-):
+class OwletSensorEntityDescription(SensorEntityDescription):
     """Represent the owlet sensor entity description."""
 
 
 SENSORS: tuple[OwletSensorEntityDescription, ...] = (
     OwletSensorEntityDescription(
-        key="batterypercentage",
-        name="Battery",
+        key="battery_percentage",
+        translation_key="batterypercent",
         native_unit_of_measurement=PERCENTAGE,
         device_class=SensorDeviceClass.BATTERY,
         state_class=SensorStateClass.MEASUREMENT,
-        element="battery_percentage",
     ),
     OwletSensorEntityDescription(
-        key="oxygensaturation",
-        name="O2 Saturation",
+        key="oxygen_saturation",
+        translation_key="o2saturation",
         native_unit_of_measurement=PERCENTAGE,
         state_class=SensorStateClass.MEASUREMENT,
-        element="oxygen_saturation",
         icon="mdi:leaf",
     ),
     OwletSensorEntityDescription(
-        key="oxygensaturation10a",
-        name="O2 Saturation 10 Minute Average",
-        native_unit_of_measurement=PERCENTAGE,
-        state_class=SensorStateClass.MEASUREMENT,
-        element="oxygen_10_av",
-        icon="mdi:leaf",
-    ),
-    OwletSensorEntityDescription(
-        key="heartrate",
-        name="Heart rate",
+        key="heart_rate",
+        translation_key="heartrate",
         native_unit_of_measurement="bpm",
         state_class=SensorStateClass.MEASUREMENT,
-        element="heart_rate",
         icon="mdi:heart-pulse",
     ),
     OwletSensorEntityDescription(
-        key="batteryminutes",
-        name="Battery Remaining",
+        key="battery_minutes",
+        translation_key="batterymin",
         native_unit_of_measurement=UnitOfTime.MINUTES,
         device_class=SensorDeviceClass.DURATION,
         state_class=SensorStateClass.MEASUREMENT,
-        element="battery_minutes",
     ),
     OwletSensorEntityDescription(
-        key="signalstrength",
-        name="Singal Strength",
+        key="signal_strength",
+        translation_key="signalstrength",
         native_unit_of_measurement=SIGNAL_STRENGTH_DECIBELS_MILLIWATT,
         device_class=SensorDeviceClass.SIGNAL_STRENGTH,
         state_class=SensorStateClass.MEASUREMENT,
-        element="signal_strength",
     ),
     OwletSensorEntityDescription(
-        key="skintemp",
-        name="Skin Temperature",
+        key="skin_temperature",
+        translation_key="skintemp",
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
         device_class=SensorDeviceClass.TEMPERATURE,
         state_class=SensorStateClass.MEASUREMENT,
-        element="skin_temperature",
+    ),
+    OwletSensorEntityDescription(
+        key="sleep_state",
+        translation_key="sleepstate",
+        device_class=SensorDeviceClass.ENUM,
+    ),
+    OwletSensorEntityDescription(
+        key="movement",
+        translation_key="movement",
+        state_class=SensorStateClass.MEASUREMENT,
+        icon="mdi:cursor-move",
+        entity_registry_enabled_default=False,
+    ),
+    OwletSensorEntityDescription(
+        key="oxygen_10_av",
+        translation_key="o2saturation10a",
+        native_unit_of_measurement=PERCENTAGE,
+        state_class=SensorStateClass.MEASUREMENT,
+        icon="mdi:leaf",
+    ),
+    OwletSensorEntityDescription(
+        key="movement_bucket",
+        translation_key="movementbucket",
+        state_class=SensorStateClass.MEASUREMENT,
+        icon="mdi:bucket-outline",
+        entity_registry_enabled_default=False,
     ),
 )
 
@@ -104,12 +109,18 @@ async def async_setup_entry(
 ) -> None:
     """Set up the owlet sensors from config entry."""
 
-    coordinator: OwletCoordinator = hass.data[DOMAIN][config_entry.entry_id]
+    coordinators: list[OwletCoordinator] = list(
+        hass.data[DOMAIN][config_entry.entry_id].values()
+    )
 
-    entities = [OwletSensor(coordinator, sensor) for sensor in SENSORS]
-    entities.append(OwletSleepStateSensor(coordinator))
+    sensors = []
 
-    async_add_entities(entities)
+    for coordinator in coordinators:
+        for sensor in SENSORS:
+            if sensor.key in coordinator.sock.properties:
+                sensors.append(OwletSensor(coordinator, sensor))
+
+    async_add_entities(sensors)
 
 
 class OwletSensor(OwletBaseEntity, SensorEntity):
@@ -118,56 +129,39 @@ class OwletSensor(OwletBaseEntity, SensorEntity):
     def __init__(
         self,
         coordinator: OwletCoordinator,
-        sensor_description: OwletSensorEntityDescription,
+        description: OwletSensorEntityDescription,
     ) -> None:
         """Initialize the sensor."""
         super().__init__(coordinator)
-        self.entity_description = sensor_description
-        self._attr_unique_id = f"{self.sock.serial}-{self.entity_description.name}"
+        self.entity_description: OwletSensorEntityDescription = description
+        self._attr_unique_id = f"{self.sock.serial}-{description.key}"
 
     @property
-    def native_value(self):
-        """Return sensor value"""
+    def native_value(self) -> StateType:
+        """Return sensor value."""
 
         if (
-            self.entity_description.element
+            self.entity_description.key
             in [
                 "heart_rate",
                 "battery_minutes",
                 "oxygen_saturation",
                 "skin_temperature",
                 "oxygen_10_av",
+                "sleep_state",
             ]
             and self.sock.properties["charging"]
         ):
             return None
 
-        return self.sock.properties[self.entity_description.element]
-
-
-class OwletSleepStateSensor(OwletBaseEntity, SensorEntity):
-    """Representation of an Owlet sleep state sensor."""
-
-    def __init__(
-        self,
-        coordinator: OwletCoordinator,
-    ) -> None:
-        """Initialize the sensor."""
-        super().__init__(coordinator)
-        self._attr_unique_id = f"{self.sock.serial}-Sleep State"
-        self._attr_icon = "mdi:sleep"
-        self._attr_device_class = SensorDeviceClass.ENUM
-        self._attr_translation_key = "sleepstate"
-        self._attr_name = "Sleep State"
-
-    @property
-    def native_value(self):
-        """Return sensor value"""
-        if self.sock.properties["charging"]:
-            return None
-
+        if self.entity_description.key == "sleep_state":
             return SLEEP_STATES[self.sock.properties["sleep_state"]]
 
+        return self.sock.properties[self.entity_description.key]
+
     @property
     def options(self) -> list[str]:
-        return ["Awake", "Light Sleep", "Deep Sleep"]
+        """Set options for sleep state."""
+        if self.entity_description.key != "sleep_state":
+            return None
+        return list(SLEEP_STATES.values())

custom_components/owlet/strings.json

@@ -2,7 +2,6 @@
   "config": {
     "step": {
       "user": {
-        "title": "Enter login details",
         "data": {
           "region": "Region",
           "username": "Email",
@@ -20,10 +19,12 @@
       "cannot_connect": "[%key:common::config_flow::error::cannot_connect%]",
       "invalid_email": "Entered email address is incorrect",
       "invalid_password": "Entered password is incorrect",
-      "unknown": "[%key:common::config_flow::error::unknown%]"
+      "invalid_credentials": "Entered credentials are incorrect",
+      "unknown": "Unknown error occured"
     },
     "abort": {
-      "already_configured": "[%key:common::config_flow::abort::already_configured_device%]"
+      "already_configured": "Device already configured",
+      "reauth_successful": "Reauthentication successful"
     }
   },
   "options": {
@@ -35,5 +36,77 @@
         }
       }
     }
+  },
+  "entity": {
+    "binary_sensor": {
+      "charging": {
+        "name": "Charging"
+      },
+      "high_hr_alrt": {
+        "name": "High heart rate alert"
+      },
+      "low_hr_alrt": {
+        "name": "Low heart rate alert"
+      },
+      "high_ox_alrt": {
+        "name": "High oxygen alert"
+      },
+      "low_ox_alrt": {
+        "name": "Low oxygen alert"
+      },
+      "low_batt_alrt": {
+        "name": "Low battery alert"
+      },
+      "lost_pwr_alrt": {
+        "name": "Lost power alert"
+      },
+      "sock_discon_alrt": {
+        "name": "Sock diconnected alert"
+      },
+      "sock_off": {
+        "name": "Sock off"
+      },
+      "awake": {
+        "name": "Awake"
+      }
+    },
+    "sensor": {
+      "batterypercent": {
+        "name": "Battery percentage"
+      },
+      "signalstrength": {
+        "name": "Signal strength"
+      },
+      "o2saturation": {
+        "name": "O2 saturation"
+      },
+      "o2saturation10a": {
+        "name": "O2 saturation 10 minute average"
+      },
+      "heartrate": {
+        "name": "Heart rate"
+      },
+      "batterymin": {
+        "name": "Battery remaining"
+      },
+      "skintemp": {
+        "name": "Skin temperature"
+      },
+      "sleepstate": {
+        "name": "Sleep state",
+        "state": {
+          "unknown": "Unknown",
+          "awake": "Awake",
+          "light_sleep": "Light sleep",
+          "deep_sleep": "Deep sleep"
+        }
+      },
+      "movement": {
+        "name": "Movement"
+      },
+      "movementbucket": {
+        "name": "Movement bucket"
+      }
+    }
   }
 }

custom_components/owlet/translations/en.json

@@ -2,7 +2,6 @@
   "config": {
     "step": {
       "user": {
-        "title": "Enter login details",
         "data": {
           "region": "Region",
           "username": "Email",
@@ -20,10 +19,12 @@
       "cannot_connect": "[%key:common::config_flow::error::cannot_connect%]",
       "invalid_email": "Entered email address is incorrect",
       "invalid_password": "Entered password is incorrect",
-      "unknown": "[%key:common::config_flow::error::unknown%]"
+      "invalid_credentials": "Entered credentials are incorrect",
+      "unknown": "Unknown error occured"
     },
     "abort": {
-      "already_configured": "[%key:common::config_flow::abort::already_configured_device%]"
+      "already_configured": "Device already configured",
+      "reauth_successful": "Reauthentication successful"
     }
   },
   "options": {
@@ -35,5 +36,77 @@
         }
       }
     }
+  },
+  "entity": {
+    "binary_sensor": {
+      "charging": {
+        "name": "Charging"
+      },
+      "high_hr_alrt": {
+        "name": "High heart rate alert"
+      },
+      "low_hr_alrt": {
+        "name": "Low heart rate alert"
+      },
+      "high_ox_alrt": {
+        "name": "High oxygen alert"
+      },
+      "low_ox_alrt": {
+        "name": "Low oxygen alert"
+      },
+      "low_batt_alrt": {
+        "name": "Low battery alert"
+      },
+      "lost_pwr_alrt": {
+        "name": "Lost power alert"
+      },
+      "sock_discon_alrt": {
+        "name": "Sock diconnected alert"
+      },
+      "sock_off": {
+        "name": "Sock off"
+      },
+      "awake": {
+        "name": "Awake"
+      }
+    },
+    "sensor": {
+      "batterypercent": {
+        "name": "Battery percentage"
+      },
+      "signalstrength": {
+        "name": "Signal strength"
+      },
+      "o2saturation": {
+        "name": "O2 saturation"
+      },
+      "o2saturation10a": {
+        "name": "O2 saturation 10 minute average"
+      },
+      "heartrate": {
+        "name": "Heart rate"
+      },
+      "batterymin": {
+        "name": "Battery remaining"
+      },
+      "skintemp": {
+        "name": "Skin temperature"
+      },
+      "sleepstate": {
+        "name": "Sleep state",
+        "state": {
+          "unknown": "Unknown",
+          "awake": "Awake",
+          "light_sleep": "Light sleep",
+          "deep_sleep": "Deep sleep"
+        }
+      },
+      "movement": {
+        "name": "Movement"
+      },
+      "movementbucket": {
+        "name": "Movement bucket"
+      }
+    }
   }
 }

custom_components/owlet/translations/uk.json

@@ -2,7 +2,6 @@
   "config": {
     "step": {
       "user": {
-        "title": "Enter login details",
         "data": {
           "region": "Region",
           "username": "Email",
@@ -20,10 +19,12 @@
       "cannot_connect": "[%key:common::config_flow::error::cannot_connect%]",
       "invalid_email": "Entered email address is incorrect",
       "invalid_password": "Entered password is incorrect",
-      "unknown": "[%key:common::config_flow::error::unknown%]"
+      "invalid_credentials": "Entered credentials are incorrect",
+      "unknown": "Unknown error occured"
     },
     "abort": {
-      "already_configured": "[%key:common::config_flow::abort::already_configured_device%]"
+      "already_configured": "Device already configured",
+      "reauth_successful": "Reauthentication successful"
     }
   },
   "options": {
@@ -35,5 +36,77 @@
         }
       }
     }
+  },
+  "entity": {
+    "binary_sensor": {
+      "charging": {
+        "name": "Charging"
+      },
+      "high_hr_alrt": {
+        "name": "High heart rate alert"
+      },
+      "low_hr_alrt": {
+        "name": "Low heart rate alert"
+      },
+      "high_ox_alrt": {
+        "name": "High oxygen alert"
+      },
+      "low_ox_alrt": {
+        "name": "Low oxygen alert"
+      },
+      "low_batt_alrt": {
+        "name": "Low battery alert"
+      },
+      "lost_pwr_alrt": {
+        "name": "Lost power alert"
+      },
+      "sock_discon_alrt": {
+        "name": "Sock diconnected alert"
+      },
+      "sock_off": {
+        "name": "Sock off"
+      },
+      "awake": {
+        "name": "Awake"
+      }
+    },
+    "sensor": {
+      "batterypercent": {
+        "name": "Battery percentage"
+      },
+      "signalstrength": {
+        "name": "Signal strength"
+      },
+      "o2saturation": {
+        "name": "O2 saturation"
+      },
+      "o2saturation10a": {
+        "name": "O2 saturation 10 minute average"
+      },
+      "heartrate": {
+        "name": "Heart rate"
+      },
+      "batterymin": {
+        "name": "Battery remaining"
+      },
+      "skintemp": {
+        "name": "Skin temperature"
+      },
+      "sleepstate": {
+        "name": "Sleep state",
+        "state": {
+          "unknown": "Unknown",
+          "awake": "Awake",
+          "light_sleep": "Light sleep",
+          "deep_sleep": "Deep sleep"
+        }
+      },
+      "movement": {
+        "name": "Movement"
+      },
+      "movementbucket": {
+        "name": "Movement bucket"
+      }
+    }
   }
 }

tests/__init__.py

@@ -0,0 +1,60 @@
+"""Tests for the Owlet integration."""
+from __future__ import annotations
+
+import json
+from unittest.mock import patch
+
+from homeassistant.components.owlet.const import (
+    CONF_OWLET_EXPIRY,
+    CONF_OWLET_REFRESH,
+    DOMAIN,
+    POLLING_INTERVAL,
+)
+from homeassistant.const import (
+    CONF_API_TOKEN,
+    CONF_REGION,
+    CONF_SCAN_INTERVAL,
+    CONF_USERNAME,
+)
+from homeassistant.core import HomeAssistant
+
+from tests.common import MockConfigEntry, load_fixture
+
+
+async def async_init_integration(
+    hass: HomeAssistant,
+    skip_setup: bool = False,
+    properties_fixture: str = "update_properties_charging.json",
+    devices_fixture: str = "get_devices.json",
+) -> MockConfigEntry:
+    """Set up integration entry."""
+    entry = MockConfigEntry(
+        domain=DOMAIN,
+        title="sample@gmail.com",
+        unique_id="sample@gmail.com",
+        data={
+            CONF_REGION: "europe",
+            CONF_USERNAME: "sample@gmail.com",
+            CONF_API_TOKEN: "api_token",
+            CONF_OWLET_EXPIRY: 100,
+            CONF_OWLET_REFRESH: "refresh_token",
+        },
+        options={CONF_SCAN_INTERVAL: POLLING_INTERVAL},
+    )
+
+    entry.add_to_hass(hass)
+
+    if not skip_setup:
+        with patch(
+            "homeassistant.components.owlet.OwletAPI.get_properties",
+            return_value=json.loads(load_fixture(properties_fixture, "owlet")),
+        ), patch(
+            "homeassistant.components.owlet.OwletAPI.authenticate", return_value=None
+        ), patch(
+            "homeassistant.components.owlet.OwletAPI.get_devices",
+            return_value=json.loads(load_fixture(devices_fixture, "owlet")),
+        ):
+            await hass.config_entries.async_setup(entry.entry_id)
+            await hass.async_block_till_done()
+
+    return entry

tests/const.py

@@ -0,0 +1,14 @@
+"""Constants used for Owlet tests."""
+
+AUTH_RETURN = {
+    "api_token": "api_token",
+    "expiry": 100,
+    "refresh": "refresh_token",
+}
+
+
+CONF_INPUT = {
+    "region": "europe",
+    "username": "sample@gmail.com",
+    "password": "sample",
+}

tests/fixtures/get_devices.json

@@ -0,0 +1,32 @@
+{
+  "response": [
+    {
+      "device": {
+        "product_name": "Owlet Baby Monitors",
+        "model": "AY001MTL1",
+        "dsn": "SERIAL_NUMBER",
+        "oem_model": "SS3-OBL-EU",
+        "sw_version": "bc 2.9.7-beta 05/01/19 16:15:18 ID beb2858",
+        "template_id": 48775,
+        "mac": "MAC",
+        "unique_hardware_id": "None",
+        "hwsig": "123-123-123",
+        "lan_ip": "192.0.0.1",
+        "connected_at": "2023-05-23T01:21:29Z",
+        "key": 123456,
+        "lan_enabled": false,
+        "connection_priority": [],
+        "has_properties": true,
+        "product_class": "None",
+        "connection_status": "Online",
+        "lat": "0.0",
+        "lng": "0.0",
+        "locality": "QW12",
+        "device_type": "Wifi",
+        "dealer": "None",
+        "manuf_model": "LBEE5PA1LD-222"
+      }
+    }
+  ],
+  "tokens": ""
+}

tests/fixtures/get_devices_with_tokens.json

@@ -0,0 +1,36 @@
+{
+  "response": [
+    {
+      "device": {
+        "product_name": "Owlet Baby Monitors",
+        "model": "AY001MTL1",
+        "dsn": "SERIAL_NUMBER",
+        "oem_model": "SS3-OBL-EU",
+        "sw_version": "bc 2.9.7-beta 05/01/19 16:15:18 ID beb2858",
+        "template_id": 48775,
+        "mac": "MAC",
+        "unique_hardware_id": "None",
+        "hwsig": "123-123-123",
+        "lan_ip": "192.0.0.1",
+        "connected_at": "2023-05-23T01:21:29Z",
+        "key": 123456,
+        "lan_enabled": false,
+        "connection_priority": [],
+        "has_properties": true,
+        "product_class": "None",
+        "connection_status": "Online",
+        "lat": "0.0",
+        "lng": "0.0",
+        "locality": "QW12",
+        "device_type": "Wifi",
+        "dealer": "None",
+        "manuf_model": "LBEE5PA1LD-222"
+      }
+    }
+  ],
+  "tokens": {
+    "api_token": "new_api_token",
+    "expiry": 200,
+    "refresh": "new_refresh_token"
+  }
+}

tests/test_config_flow.py

@@ -0,0 +1,237 @@
+"""Test Owlet config flow."""
+from __future__ import annotations
+
+from unittest.mock import patch
+
+from pyowletapi.exceptions import (
+    OwletCredentialsError,
+    OwletDevicesError,
+    OwletEmailError,
+    OwletPasswordError,
+)
+
+from homeassistant import config_entries
+from homeassistant.components.owlet.const import DOMAIN, POLLING_INTERVAL
+from homeassistant.config_entries import SOURCE_REAUTH, SOURCE_USER
+from homeassistant.const import CONF_PASSWORD, CONF_SCAN_INTERVAL
+from homeassistant.core import HomeAssistant
+from homeassistant.data_entry_flow import FlowResultType
+
+from . import async_init_integration
+from .const import AUTH_RETURN, CONF_INPUT
+
+
+async def test_form(hass: HomeAssistant) -> None:
+    """Test that the form is served with no input."""
+    # await async_init_integration(hass)
+    result = await hass.config_entries.flow.async_init(
+        DOMAIN, context={"source": SOURCE_USER}
+    )
+
+    assert result["type"] == FlowResultType.FORM
+    assert result["step_id"] == "user"
+
+    with patch(
+        "homeassistant.components.owlet.config_flow.OwletAPI.authenticate",
+        return_value=AUTH_RETURN,
+    ), patch(
+        "homeassistant.components.owlet.config_flow.OwletAPI.validate_authentication"
+    ):
+        result = await hass.config_entries.flow.async_configure(
+            result["flow_id"],
+            user_input=CONF_INPUT,
+        )
+        await hass.async_block_till_done()
+
+        assert result["type"] == FlowResultType.CREATE_ENTRY
+        assert result["title"] == "sample@gmail.com"
+        assert result["data"] == {
+            "region": "europe",
+            "username": "sample@gmail.com",
+            "api_token": "api_token",
+            "expiry": 100,
+            "refresh": "refresh_token",
+        }
+        assert result["options"] == {"scan_interval": POLLING_INTERVAL}
+
+
+async def test_flow_wrong_password(hass: HomeAssistant) -> None:
+    """Test incorrect login throwing error."""
+    with patch(
+        "homeassistant.components.owlet.config_flow.OwletAPI.authenticate",
+        side_effect=OwletPasswordError(),
+    ):
+        result = await hass.config_entries.flow.async_init(
+            DOMAIN,
+            context={"source": config_entries.SOURCE_USER},
+        )
+        result = await hass.config_entries.flow.async_configure(
+            result["flow_id"],
+            user_input=CONF_INPUT,
+        )
+        assert result["type"] == FlowResultType.FORM
+        assert result["errors"] == {"base": "invalid_password"}
+
+
+async def test_flow_wrong_email(hass: HomeAssistant) -> None:
+    """Test incorrect login throwing error."""
+    with patch(
+        "homeassistant.components.owlet.config_flow.OwletAPI.authenticate",
+        side_effect=OwletEmailError(),
+    ):
+        result = await hass.config_entries.flow.async_init(
+            DOMAIN,
+            context={"source": config_entries.SOURCE_USER},
+        )
+        result = await hass.config_entries.flow.async_configure(
+            result["flow_id"],
+            user_input=CONF_INPUT,
+        )
+        assert result["type"] == FlowResultType.FORM
+        assert result["errors"] == {"base": "invalid_email"}
+
+
+async def test_flow_credentials_error(hass: HomeAssistant) -> None:
+    """Test incorrect login throwing error."""
+    with patch(
+        "homeassistant.components.owlet.config_flow.OwletAPI.authenticate",
+        side_effect=OwletCredentialsError(),
+    ):
+        result = await hass.config_entries.flow.async_init(
+            DOMAIN,
+            context={"source": config_entries.SOURCE_USER},
+        )
+        result = await hass.config_entries.flow.async_configure(
+            result["flow_id"],
+            user_input=CONF_INPUT,
+        )
+        assert result["type"] == FlowResultType.FORM
+        assert result["errors"] == {"base": "invalid_credentials"}
+
+
+async def test_flow_unknown_error(hass: HomeAssistant) -> None:
+    """Test unknown error throwing error."""
+    with patch(
+        "homeassistant.components.owlet.config_flow.OwletAPI.authenticate",
+        side_effect=Exception(),
+    ):
+        result = await hass.config_entries.flow.async_init(
+            DOMAIN,
+            context={"source": config_entries.SOURCE_USER},
+        )
+        result = await hass.config_entries.flow.async_configure(
+            result["flow_id"],
+            user_input=CONF_INPUT,
+        )
+        assert result["type"] == FlowResultType.FORM
+        assert result["errors"] == {"base": "unknown"}
+
+
+async def test_flow_no_devices(hass: HomeAssistant) -> None:
+    """Test unknown error throwing error."""
+    with patch(
+        "homeassistant.components.owlet.config_flow.OwletAPI.authenticate"
+    ), patch(
+        "homeassistant.components.owlet.config_flow.OwletAPI.validate_authentication",
+        side_effect=OwletDevicesError(),
+    ):
+        result = await hass.config_entries.flow.async_init(
+            DOMAIN,
+            context={"source": config_entries.SOURCE_USER},
+        )
+        result = await hass.config_entries.flow.async_configure(
+            result["flow_id"],
+            user_input=CONF_INPUT,
+        )
+        assert result["type"] == FlowResultType.FORM
+        assert result["errors"] == {"base": "no_devices"}
+
+
+async def test_reauth_success(hass: HomeAssistant) -> None:
+    """Test reauth form."""
+
+    entry = await async_init_integration(hass, skip_setup=True)
+
+    result = await hass.config_entries.flow.async_init(
+        DOMAIN, context={"source": SOURCE_REAUTH, "entry_id": entry.entry_id}
+    )
+
+    assert result["type"] == FlowResultType.FORM
+    assert result["step_id"] == "reauth_confirm"
+
+    with patch(
+        "homeassistant.components.owlet.config_flow.OwletAPI.authenticate",
+        return_value=AUTH_RETURN,
+    ):
+        result = await hass.config_entries.flow.async_configure(
+            result["flow_id"],
+            user_input={CONF_PASSWORD: "sample"},
+        )
+        await hass.async_block_till_done()
+
+        assert result["type"] == FlowResultType.ABORT
+        assert result["reason"] == "reauth_successful"
+
+        await hass.config_entries.async_unload(entry.entry_id)
+
+
+async def test_reauth_invalid_password(hass: HomeAssistant) -> None:
+    """Test reauth with invalid password errir."""
+    entry = await async_init_integration(hass, skip_setup=True)
+
+    result = await hass.config_entries.flow.async_init(
+        DOMAIN, context={"source": SOURCE_REAUTH, "entry_id": entry.entry_id}
+    )
+
+    with patch(
+        "homeassistant.components.owlet.config_flow.OwletAPI.authenticate",
+        side_effect=OwletPasswordError(),
+    ):
+        result = await hass.config_entries.flow.async_configure(
+            result["flow_id"], user_input={CONF_PASSWORD: "sample"}
+        )
+
+        assert result["type"] == FlowResultType.FORM
+        assert result["step_id"] == "reauth_confirm"
+        assert result["errors"] == {"base": "invalid_password"}
+
+
+async def test_reauth_unknown_error(hass: HomeAssistant) -> None:
+    """Test reauthing with an unknown error."""
+
+    entry = await async_init_integration(hass, skip_setup=True)
+
+    result = await hass.config_entries.flow.async_init(
+        DOMAIN, context={"source": SOURCE_REAUTH, "entry_id": entry.entry_id}
+    )
+
+    with patch(
+        "homeassistant.components.owlet.config_flow.OwletAPI.authenticate",
+        side_effect=Exception(),
+    ):
+        result = await hass.config_entries.flow.async_configure(
+            result["flow_id"], user_input={CONF_PASSWORD: "sample"}
+        )
+
+        assert result["type"] == FlowResultType.FORM
+        assert result["step_id"] == "reauth_confirm"
+
+
+async def test_options_flow(hass: HomeAssistant) -> None:
+    """Test that the form is served with no input."""
+    entry = await async_init_integration(hass, skip_setup=True)
+
+    result = await hass.config_entries.options.async_init(entry.entry_id)
+
+    assert result["type"] == FlowResultType.FORM
+    assert result["step_id"] == "init"
+
+    result = await hass.config_entries.options.async_configure(
+        result["flow_id"],
+        user_input={CONF_SCAN_INTERVAL: 10},
+    )
+    await hass.async_block_till_done()
+
+    assert result["type"] == FlowResultType.CREATE_ENTRY
+    assert result["title"] == ""
+    assert result["data"] == {CONF_SCAN_INTERVAL: 10}

tests/test_coordinator.py

@@ -0,0 +1,74 @@
+"""Test owlet coordinator."""
+from __future__ import annotations
+
+import json
+from unittest.mock import patch
+
+from pyowletapi.exceptions import OwletAuthenticationError, OwletConnectionError
+
+from homeassistant.config_entries import ConfigEntryState
+from homeassistant.const import Platform
+from homeassistant.core import HomeAssistant
+
+from . import async_init_integration
+
+from tests.common import load_fixture
+
+PLATFORMS: list[Platform] = [Platform.BINARY_SENSOR, Platform.SENSOR]
+
+
+async def test_coordinator_auth_error(hass: HomeAssistant) -> None:
+    """Test coordinator setup authentication error."""
+    entry = await async_init_integration(hass, skip_setup=True)
+
+    with patch(
+        "homeassistant.components.owlet.Sock.update_properties",
+        side_effect=OwletAuthenticationError(),
+    ), patch(
+        "homeassistant.components.owlet.OwletAPI.authenticate", return_value=None
+    ), patch(
+        "homeassistant.components.owlet.OwletAPI.get_devices",
+        return_value=json.loads(load_fixture("get_devices.json", "owlet")),
+    ):
+        await hass.config_entries.async_setup(entry.entry_id)
+        await hass.async_block_till_done()
+
+        assert entry.state == ConfigEntryState.SETUP_RETRY
+
+
+async def test_coordinator_connection_error(hass: HomeAssistant) -> None:
+    """Test coordinator setup connection error error."""
+    entry = await async_init_integration(hass, skip_setup=True)
+
+    with patch(
+        "homeassistant.components.owlet.Sock.update_properties",
+        side_effect=OwletConnectionError(),
+    ), patch(
+        "homeassistant.components.owlet.OwletAPI.authenticate", return_value=None
+    ), patch(
+        "homeassistant.components.owlet.OwletAPI.get_devices",
+        return_value=json.loads(load_fixture("get_devices.json", "owlet")),
+    ):
+        await hass.config_entries.async_setup(entry.entry_id)
+        await hass.async_block_till_done()
+
+        assert entry.state == ConfigEntryState.SETUP_RETRY
+
+
+async def test_coordinator_error(hass: HomeAssistant) -> None:
+    """Test coordinator setup generic error."""
+    entry = await async_init_integration(hass, skip_setup=True)
+
+    with patch(
+        "homeassistant.components.owlet.Sock.update_properties",
+        side_effect=Exception(),
+    ), patch(
+        "homeassistant.components.owlet.OwletAPI.authenticate", return_value=None
+    ), patch(
+        "homeassistant.components.owlet.OwletAPI.get_devices",
+        return_value=json.loads(load_fixture("get_devices.json", "owlet")),
+    ):
+        await hass.config_entries.async_setup(entry.entry_id)
+        await hass.async_block_till_done()
+
+        assert entry.state == ConfigEntryState.SETUP_RETRY

tests/test_init.py

@@ -0,0 +1,136 @@
+"""Test Owlet init."""
+from __future__ import annotations
+
+from unittest.mock import patch
+
+from pyowletapi.exceptions import (
+    OwletAuthenticationError,
+    OwletConnectionError,
+    OwletDevicesError,
+    OwletError,
+)
+
+from homeassistant.components.owlet.const import (
+    CONF_OWLET_EXPIRY,
+    CONF_OWLET_REFRESH,
+    DOMAIN,
+)
+from homeassistant.config_entries import ConfigEntryState
+from homeassistant.const import CONF_API_TOKEN, CONF_REGION, CONF_USERNAME, Platform
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers import device_registry as dr, entity_registry as er
+
+from . import async_init_integration
+
+PLATFORMS: list[Platform] = [Platform.BINARY_SENSOR, Platform.SENSOR]
+
+
+async def test_async_setup_entry(hass: HomeAssistant) -> None:
+    """Test setting up entry."""
+    entry = await async_init_integration(hass)
+
+    assert entry.state == ConfigEntryState.LOADED
+
+    device_registry = dr.async_get(hass)
+
+    device_entry = device_registry.async_get_or_create(
+        config_entry_id=entry.entry_id, identifiers={(DOMAIN, "SERIAL_NUMBER")}
+    )
+
+    assert device_entry.name == "Owlet Baby Care Sock"
+
+    entity_registry = er.async_get(hass)
+
+    entities = er.async_entries_for_device(entity_registry, device_entry.id)
+
+    assert len(entities) == 8
+
+    await entry.async_unload(hass)
+
+    assert entry.state == ConfigEntryState.NOT_LOADED
+
+
+async def test_async_setup_entry_new_tokens(hass: HomeAssistant) -> None:
+    """Test setting up entry and getting new tokens."""
+    entry = await async_init_integration(
+        hass, devices_fixture="get_devices_with_tokens.json"
+    )
+
+    assert entry.data == {
+        CONF_REGION: "europe",
+        CONF_USERNAME: "sample@gmail.com",
+        CONF_API_TOKEN: "new_api_token",
+        CONF_OWLET_EXPIRY: 200,
+        CONF_OWLET_REFRESH: "new_refresh_token",
+    }
+
+    assert entry.state == ConfigEntryState.LOADED
+
+    await entry.async_unload(hass)
+
+    assert entry.state == ConfigEntryState.NOT_LOADED
+
+
+async def test_async_setup_entry_auth_error(hass: HomeAssistant) -> None:
+    """Test setting up entry with auth error."""
+    entry = await async_init_integration(hass, skip_setup=True)
+
+    with patch(
+        "homeassistant.components.owlet.OwletAPI.authenticate",
+        side_effect=OwletAuthenticationError(),
+    ):
+        await hass.config_entries.async_setup(entry.entry_id)
+        await hass.async_block_till_done()
+
+        assert entry.state == ConfigEntryState.SETUP_ERROR
+
+        await entry.async_unload(hass)
+
+
+async def test_async_setup_entry_connection_error(hass: HomeAssistant) -> None:
+    """Test setting up entry with connection error."""
+    entry = await async_init_integration(hass, skip_setup=True)
+
+    with patch(
+        "homeassistant.components.owlet.OwletAPI.authenticate",
+        side_effect=OwletConnectionError(),
+    ):
+        await hass.config_entries.async_setup(entry.entry_id)
+        await hass.async_block_till_done()
+
+        assert entry.state == ConfigEntryState.SETUP_RETRY
+
+        await entry.async_unload(hass)
+
+
+async def test_async_setup_entry_devices_error(hass: HomeAssistant) -> None:
+    """Test setting up entry with device error."""
+    entry = await async_init_integration(hass, skip_setup=True)
+
+    with patch(
+        "homeassistant.components.owlet.OwletAPI.authenticate", return_value=None
+    ), patch(
+        "homeassistant.components.owlet.OwletAPI.get_devices",
+        side_effect=OwletDevicesError(),
+    ):
+        await hass.config_entries.async_setup(entry.entry_id)
+        await hass.async_block_till_done()
+
+        assert entry.state == ConfigEntryState.SETUP_ERROR
+        await entry.async_unload(hass)
+
+
+async def test_async_setup_entry_error(hass: HomeAssistant) -> None:
+    """Test setting up entry with unknown error."""
+    entry = await async_init_integration(hass, skip_setup=True)
+
+    with patch(
+        "homeassistant.components.owlet.OwletAPI.authenticate",
+        side_effect=OwletError(),
+    ):
+        await hass.config_entries.async_setup(entry.entry_id)
+        await hass.async_block_till_done()
+
+        assert entry.state == ConfigEntryState.SETUP_ERROR
+
+        await entry.async_unload(hass)

tests/test_sensor.py

@@ -0,0 +1,109 @@
+"""Test Owlet Sensor."""
+from __future__ import annotations
+
+from homeassistant.core import HomeAssistant
+
+from . import async_init_integration
+
+
+async def test_sensors_asleep(hass: HomeAssistant) -> None:
+    """Test sensor values."""
+    await async_init_integration(
+        hass, properties_fixture="update_properties_asleep.json"
+    )
+
+    assert len(hass.states.async_all("sensor")) == 8
+
+    assert (
+        hass.states.get("sensor.owlet_baby_care_sock_battery_percentage").state
+        == "50.0"
+    )
+    assert (
+        hass.states.get("sensor.owlet_baby_care_sock_battery_remaining").state
+        == "400.0"
+    )
+    assert hass.states.get("sensor.owlet_baby_care_sock_heart_rate").state == "97.0"
+    assert hass.states.get("sensor.owlet_baby_care_sock_o2_saturation").state == "99.0"
+    assert (
+        hass.states.get(
+            "sensor.owlet_baby_care_sock_o2_saturation_10_minute_average"
+        ).state
+        == "97.0"
+    )
+
+    assert (
+        hass.states.get("sensor.owlet_baby_care_sock_signal_strength").state == "30.0"
+    )
+    assert hass.states.get("sensor.owlet_baby_care_sock_skin_temperature").state == "34"
+    assert (
+        hass.states.get("sensor.owlet_baby_care_sock_sleep_state").state
+        == "light_sleep"
+    )
+
+
+async def test_sensors_awake(hass: HomeAssistant) -> None:
+    """Test sensor values."""
+    await async_init_integration(
+        hass, properties_fixture="update_properties_awake.json"
+    )
+
+    assert len(hass.states.async_all("sensor")) == 8
+
+    assert (
+        hass.states.get("sensor.owlet_baby_care_sock_battery_percentage").state
+        == "80.0"
+    )
+    assert (
+        hass.states.get("sensor.owlet_baby_care_sock_battery_remaining").state
+        == "600.0"
+    )
+    assert hass.states.get("sensor.owlet_baby_care_sock_heart_rate").state == "110.0"
+    assert hass.states.get("sensor.owlet_baby_care_sock_o2_saturation").state == "98.0"
+    assert (
+        hass.states.get(
+            "sensor.owlet_baby_care_sock_o2_saturation_10_minute_average"
+        ).state
+        == "98.0"
+    )
+    assert (
+        hass.states.get("sensor.owlet_baby_care_sock_signal_strength").state == "34.0"
+    )
+    assert hass.states.get("sensor.owlet_baby_care_sock_skin_temperature").state == "35"
+    assert hass.states.get("sensor.owlet_baby_care_sock_sleep_state").state == "awake"
+
+
+async def test_sensors_charging(hass: HomeAssistant) -> None:
+    """Test sensor values."""
+    await async_init_integration(
+        hass, properties_fixture="update_properties_charging.json"
+    )
+
+    assert len(hass.states.async_all("sensor")) == 8
+
+    assert (
+        hass.states.get("sensor.owlet_baby_care_sock_battery_percentage").state
+        == "100.0"
+    )
+    assert (
+        hass.states.get("sensor.owlet_baby_care_sock_battery_remaining").state
+        == "unknown"
+    )
+    assert hass.states.get("sensor.owlet_baby_care_sock_heart_rate").state == "unknown"
+    assert (
+        hass.states.get("sensor.owlet_baby_care_sock_o2_saturation").state == "unknown"
+    )
+    assert (
+        hass.states.get(
+            "sensor.owlet_baby_care_sock_o2_saturation_10_minute_average"
+        ).state
+        == "unknown"
+    )
+
+    assert (
+        hass.states.get("sensor.owlet_baby_care_sock_signal_strength").state == "34.0"
+    )
+    assert (
+        hass.states.get("sensor.owlet_baby_care_sock_skin_temperature").state
+        == "unknown"
+    )
+    assert hass.states.get("sensor.owlet_baby_care_sock_sleep_state").state == "unknown"