jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-08 13:57:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Catch exceptions on entry setup for Autarco integration (#132227)

Browse Source
This commit is contained in:
Klaas Schoute 2024-12-04 11:17:39 +01:00 committed by GitHub
parent db266d80ec
commit f0c07d68c5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 25 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
homeassistant/components/autarco/__init__.py
View File

@ -4,11 +4,12 @@ from __future__ import annotations
import asyncio import asyncio
from autarco import Autarco from autarco import Autarco, AutarcoConnectionError
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_EMAIL, CONF_PASSWORD, Platform from homeassistant.const import CONF_EMAIL, CONF_PASSWORD, Platform
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryNotReady
from homeassistant.helpers.aiohttp_client import async_get_clientsession from homeassistant.helpers.aiohttp_client import async_get_clientsession
from .coordinator import AutarcoDataUpdateCoordinator from .coordinator import AutarcoDataUpdateCoordinator
@ -25,7 +26,12 @@ async def async_setup_entry(hass: HomeAssistant, entry: AutarcoConfigEntry) -> b
password=entry.data[CONF_PASSWORD], password=entry.data[CONF_PASSWORD],
session=async_get_clientsession(hass), session=async_get_clientsession(hass),
) )
try:
account_sites = await client.get_account() account_sites = await client.get_account()
except AutarcoConnectionError as err:
await client.close()
raise ConfigEntryNotReady from err
coordinators: list[AutarcoDataUpdateCoordinator] = [ coordinators: list[AutarcoDataUpdateCoordinator] = [
AutarcoDataUpdateCoordinator(hass, client, site) for site in account_sites AutarcoDataUpdateCoordinator(hass, client, site) for site in account_sites

2
homeassistant/components/autarco/strings.json
View File

@ -28,7 +28,7 @@
"cannot_connect": "[%key:common::config_flow::error::cannot_connect%]" "cannot_connect": "[%key:common::config_flow::error::cannot_connect%]"
}, },
"abort": { "abort": {
"already_configured": "[%key:common::config_flow::abort::already_configured_device%]", "already_configured": "[%key:common::config_flow::abort::already_configured_account%]",
"reauth_successful": "[%key:common::config_flow::abort::reauth_successful%]" "reauth_successful": "[%key:common::config_flow::abort::reauth_successful%]"
} }
}, },

17
tests/components/autarco/test_init.py
View File

@ -4,7 +4,7 @@ from __future__ import annotations
from unittest.mock import AsyncMock from unittest.mock import AsyncMock
from autarco import AutarcoAuthenticationError from autarco import AutarcoAuthenticationError, AutarcoConnectionError
from homeassistant.config_entries import ConfigEntryState from homeassistant.config_entries import ConfigEntryState
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
@ -30,6 +30,21 @@ async def test_load_unload_entry(
assert mock_config_entry.state is ConfigEntryState.NOT_LOADED assert mock_config_entry.state is ConfigEntryState.NOT_LOADED
async def test_config_entry_not_ready(
hass: HomeAssistant,
mock_autarco_client: AsyncMock,
mock_config_entry: MockConfigEntry,
) -> None:
"""Test the Autarco configuration entry not ready."""
mock_autarco_client.get_account.side_effect = AutarcoConnectionError
mock_config_entry.add_to_hass(hass)
await hass.config_entries.async_setup(mock_config_entry.entry_id)
await hass.async_block_till_done()
assert mock_config_entry.state is ConfigEntryState.SETUP_RETRY
async def test_setup_entry_exception( async def test_setup_entry_exception(
hass: HomeAssistant, hass: HomeAssistant,
mock_autarco_client: AsyncMock, mock_autarco_client: AsyncMock,