evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add better connection management for Idasen Desk (#102135)

Browse source
This commit is contained in:
Abílio Costa 2023-10-18 23:58:31 +01:00 • committed by GitHub
parent 2531b0bc09
commit 606b76c681
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
9 changed files with 111 additions and 46 deletions
Download patch file Download diff file Expand all files Collapse all files

17
tests/components/idasen_desk/test_init.py
View file

@ -1,7 +1,9 @@
"""Test the IKEA Idasen Desk init."""
from unittest import mock
from unittest.mock import AsyncMock, MagicMock
from bleak import BleakError
from bleak.exc import BleakError
from idasen_ha.errors import AuthFailedError
import pytest
from homeassistant.components.idasen_desk.const import DOMAIN
@ -28,7 +30,7 @@ async def test_setup_and_shutdown(
mock_desk_api.disconnect.assert_called_once()
@pytest.mark.parametrize("exception", [TimeoutError(), BleakError()])
@pytest.mark.parametrize("exception", [AuthFailedError(), TimeoutError(), BleakError()])
async def test_setup_connect_exception(
hass: HomeAssistant, mock_desk_api: MagicMock, exception: Exception
) -> None:
@ -39,6 +41,17 @@ async def test_setup_connect_exception(
assert entry.state is ConfigEntryState.SETUP_RETRY
async def test_no_ble_device(hass: HomeAssistant, mock_desk_api: MagicMock) -> None:
"""Test setup with no BLEDevice from address."""
with mock.patch(
"homeassistant.components.idasen_desk.bluetooth.async_ble_device_from_address",
return_value=None,
):
entry = await init_integration(hass)
assert len(hass.config_entries.async_entries(DOMAIN)) == 1
assert entry.state is ConfigEntryState.SETUP_RETRY
async def test_unload_entry(hass: HomeAssistant, mock_desk_api: MagicMock) -> None:
"""Test successful unload of entry."""
entry = await init_integration(hass)