RyanClark123
50fe1a8765
### Feature * Support added for V2 sock * Added tests for binary sensors ### Fix * Bumping pyowletapi to 2023.11.4 to allow V2 support * Refactoring * Corrected spelling of sock disconnected sensor
137 lines
4.1 KiB
Python
137 lines
4.1 KiB
Python
"""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) == 18
|
|
|
|
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)
|