mirror of
https://github.com/home-assistant/core.git
synced 2026-06-30 19:26:31 +01:00
Keilin Bickar
2a6b686254
Co-authored-by: Inca <inca@popre.net> Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>
124 lines
3.5 KiB
Python
124 lines
3.5 KiB
Python
"""Tests for the Sense integration setup."""
|
|
|
|
import socket
|
|
from unittest.mock import MagicMock
|
|
|
|
import pytest
|
|
from sense_energy import (
|
|
SenseAPIException,
|
|
SenseAPITimeoutException,
|
|
SenseAuthenticationException,
|
|
SenseMFARequiredException,
|
|
SenseWebsocketException,
|
|
)
|
|
|
|
from homeassistant.config_entries import ConfigEntryState
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"exception",
|
|
[
|
|
SenseAPITimeoutException(),
|
|
SenseWebsocketException(),
|
|
],
|
|
)
|
|
async def test_setup_entry_exceptions(
|
|
hass: HomeAssistant,
|
|
mock_sense: MagicMock,
|
|
config_entry: MockConfigEntry,
|
|
exception: Exception,
|
|
) -> None:
|
|
"""Test we handle exceptions during async_setup_entry and can recover."""
|
|
mock_sense.update_realtime.side_effect = exception
|
|
config_entry.add_to_hass(hass)
|
|
|
|
assert not await hass.config_entries.async_setup(config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
assert config_entry.state is ConfigEntryState.SETUP_RETRY
|
|
|
|
# Verify recovery: clear the error and reload the entry
|
|
mock_sense.update_realtime.side_effect = None
|
|
assert await hass.config_entries.async_reload(config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
assert config_entry.state is ConfigEntryState.LOADED
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"exception",
|
|
[
|
|
SenseAuthenticationException(),
|
|
SenseMFARequiredException(),
|
|
],
|
|
)
|
|
async def test_setup_get_monitor_data_auth_exceptions(
|
|
hass: HomeAssistant,
|
|
mock_sense: MagicMock,
|
|
config_entry: MockConfigEntry,
|
|
exception: Exception,
|
|
) -> None:
|
|
"""Test auth exceptions from get_monitor_data result in a failed entry."""
|
|
mock_sense.get_monitor_data.side_effect = exception
|
|
config_entry.add_to_hass(hass)
|
|
|
|
assert not await hass.config_entries.async_setup(config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
assert config_entry.state is ConfigEntryState.SETUP_ERROR
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"exception",
|
|
[
|
|
SenseAPITimeoutException(),
|
|
TimeoutError(),
|
|
SenseAPIException("connect error"),
|
|
socket.gaierror(),
|
|
],
|
|
)
|
|
async def test_setup_get_monitor_data_retry_exceptions(
|
|
hass: HomeAssistant,
|
|
mock_sense: MagicMock,
|
|
config_entry: MockConfigEntry,
|
|
exception: Exception,
|
|
) -> None:
|
|
"""Test timeout and connect exceptions from get_monitor_data result in a retryable entry."""
|
|
mock_sense.get_monitor_data.side_effect = exception
|
|
config_entry.add_to_hass(hass)
|
|
|
|
assert not await hass.config_entries.async_setup(config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
assert config_entry.state is ConfigEntryState.SETUP_RETRY
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"exception",
|
|
[
|
|
SenseAPITimeoutException(),
|
|
TimeoutError(),
|
|
SenseAPIException("connect error"),
|
|
socket.gaierror(),
|
|
SenseWebsocketException("ws error"),
|
|
SenseAPIException(),
|
|
],
|
|
)
|
|
async def test_setup_get_realtime_retry_exceptions(
|
|
hass: HomeAssistant,
|
|
mock_sense: MagicMock,
|
|
config_entry: MockConfigEntry,
|
|
exception: Exception,
|
|
) -> None:
|
|
"""Test timeout and connect exceptions from update_realtime result in a retryable entry."""
|
|
mock_sense.update_realtime.side_effect = exception
|
|
config_entry.add_to_hass(hass)
|
|
|
|
assert not await hass.config_entries.async_setup(config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
assert config_entry.state is ConfigEntryState.SETUP_RETRY
|