Refactoring, added tests

###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-05-30 13:30:59 +01:00
parent c6f37493ca
commit 26f81c14bf
21 changed files with 4202 additions and 128 deletions
@@ -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