mirror of
https://github.com/home-assistant/core.git
synced 2026-09-06 21:41:28 +01:00
Paulus SchoutsenandClaude Opus 4.8
f65fb38168
Co-authored-by: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
80 lines
2.5 KiB
Python
80 lines
2.5 KiB
Python
"""Tests for the Samsung ExLink integration init."""
|
|
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
import pytest
|
|
from samsung_exlink import SamsungTVError
|
|
|
|
from homeassistant.config_entries import ConfigEntryState
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from .conftest import MockSamsungTV
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("failing_method", "exception"),
|
|
[
|
|
pytest.param("connect", TimeoutError, id="connect-timeout"),
|
|
pytest.param(
|
|
"refresh", SamsungTVError("unexpected response"), id="refresh-tv-error"
|
|
),
|
|
],
|
|
)
|
|
async def test_setup_connection_error(
|
|
hass: HomeAssistant,
|
|
mock_samsung_tv: MockSamsungTV,
|
|
mock_config_entry: MockConfigEntry,
|
|
failing_method: str,
|
|
exception: Exception,
|
|
) -> None:
|
|
"""Test a connection failure results in a retry."""
|
|
getattr(mock_samsung_tv, failing_method).side_effect = exception
|
|
|
|
with patch(
|
|
"homeassistant.components.samsung_exlink.SamsungTV",
|
|
return_value=mock_samsung_tv,
|
|
):
|
|
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_unload_entry(
|
|
hass: HomeAssistant,
|
|
mock_samsung_tv: MockSamsungTV,
|
|
init_components: None,
|
|
mock_config_entry: MockConfigEntry,
|
|
) -> None:
|
|
"""Test unloading a config entry disconnects from the TV."""
|
|
assert mock_config_entry.state is ConfigEntryState.LOADED
|
|
|
|
await hass.config_entries.async_unload(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
assert mock_config_entry.state is ConfigEntryState.NOT_LOADED
|
|
mock_samsung_tv.disconnect.assert_awaited_once()
|
|
|
|
|
|
async def test_remove_entry_while_loaded(
|
|
hass: HomeAssistant,
|
|
mock_samsung_tv: MockSamsungTV,
|
|
init_components: AsyncMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
) -> None:
|
|
"""Test removing a config entry does not schedule a reload.
|
|
|
|
When removing a loaded entry, disconnect() fires the subscriber callback
|
|
with state=None. The callback must not schedule a reload because the entry
|
|
is already being removed (state is no longer LOADED).
|
|
"""
|
|
assert mock_config_entry.state is ConfigEntryState.LOADED
|
|
|
|
await hass.config_entries.async_remove(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
assert mock_config_entry.state is ConfigEntryState.NOT_LOADED
|
|
mock_samsung_tv.disconnect.assert_awaited_once()
|