home-assistant/core
1
0
Fork 1
mirror of https://github.com/home-assistant/core.git synced 2025-09-10 23:31:37 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add more exception types for cannot_connect test in APCUPSD (#150830)

Browse Source
This commit is contained in:
Yuxin Wang
2025-08-17 22:19:47 -07:00
committed by GitHub
parent 2b7bd923d6
commit f44578f45f
1 changed files with 11 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
tests/components/apcupsd/test_config_flow.py
View File

@@ -2,6 +2,7 @@
from __future__ import annotations
import asyncio
from unittest.mock import AsyncMock, patch
import pytest
@@ -17,12 +18,18 @@ from . import CONF_DATA, MOCK_MINIMAL_STATUS, MOCK_STATUS
from tests.common import MockConfigEntry
async def test_config_flow_cannot_connect(hass: HomeAssistant) -> None:
"""Test config flow setup with connection error."""
@pytest.mark.parametrize(
"exception",
[OSError(), asyncio.IncompleteReadError(partial=b"", expected=100), TimeoutError()],
)
async def test_config_flow_cannot_connect(
hass: HomeAssistant, exception: Exception
) -> None:
"""Test config flow setup with a connection error."""
with patch(
"homeassistant.components.apcupsd.coordinator.aioapcaccess.request_status"
) as mock_get:
mock_get.side_effect = OSError()
) as mock_request_status:
mock_request_status.side_effect = exception
result = await hass.config_entries.flow.async_init(
DOMAIN,