mirror of
https://github.com/home-assistant/core.git
synced 2025-06-26 16:07:08 +00:00
9a0837593a
Improve calendar loading by executing in a separate thread and add test for CalendarParseError
87 lines
2.4 KiB
Python
87 lines
2.4 KiB
Python
"""Tests for init platform of Remote Calendar."""
|
|
|
|
from httpx import ConnectError, Response, UnsupportedProtocol
|
|
import pytest
|
|
import respx
|
|
|
|
from homeassistant.config_entries import ConfigEntryState
|
|
from homeassistant.const import STATE_OFF
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from . import setup_integration
|
|
from .conftest import CALENDER_URL, TEST_ENTITY
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
@respx.mock
|
|
async def test_load_unload(
|
|
hass: HomeAssistant, config_entry: MockConfigEntry, ics_content: str
|
|
) -> None:
|
|
"""Test loading and unloading a config entry."""
|
|
respx.get(CALENDER_URL).mock(
|
|
return_value=Response(
|
|
status_code=200,
|
|
text=ics_content,
|
|
)
|
|
)
|
|
await setup_integration(hass, config_entry)
|
|
assert config_entry.state is ConfigEntryState.LOADED
|
|
|
|
state = hass.states.get(TEST_ENTITY)
|
|
assert state
|
|
assert state.state == STATE_OFF
|
|
|
|
await hass.config_entries.async_unload(config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
assert config_entry.state is ConfigEntryState.NOT_LOADED
|
|
|
|
|
|
@respx.mock
|
|
async def test_raise_for_status(
|
|
hass: HomeAssistant,
|
|
config_entry: MockConfigEntry,
|
|
) -> None:
|
|
"""Test update failed using respx to simulate HTTP exceptions."""
|
|
respx.get(CALENDER_URL).mock(
|
|
return_value=Response(
|
|
status_code=403,
|
|
)
|
|
)
|
|
await setup_integration(hass, config_entry)
|
|
assert config_entry.state is ConfigEntryState.SETUP_RETRY
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"side_effect",
|
|
[
|
|
ConnectError("Connection failed"),
|
|
UnsupportedProtocol("Unsupported protocol"),
|
|
ValueError("Invalid response"),
|
|
],
|
|
)
|
|
@respx.mock
|
|
async def test_update_failed(
|
|
hass: HomeAssistant,
|
|
config_entry: MockConfigEntry,
|
|
side_effect: Exception,
|
|
) -> None:
|
|
"""Test update failed using respx to simulate different exceptions."""
|
|
respx.get(CALENDER_URL).mock(side_effect=side_effect)
|
|
await setup_integration(hass, config_entry)
|
|
assert config_entry.state is ConfigEntryState.SETUP_RETRY
|
|
|
|
|
|
@respx.mock
|
|
async def test_calendar_parse_error(
|
|
hass: HomeAssistant,
|
|
config_entry: MockConfigEntry,
|
|
) -> None:
|
|
"""Test CalendarParseError using respx."""
|
|
respx.get(CALENDER_URL).mock(
|
|
return_value=Response(status_code=200, text="not a calendar")
|
|
)
|
|
await setup_integration(hass, config_entry)
|
|
assert config_entry.state is ConfigEntryState.SETUP_RETRY
|