RyanClark123
26f81c14bf
###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
75 lines
2.6 KiB
Python
75 lines
2.6 KiB
Python
"""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
|