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

Improve type hints in lcn tests (#124415)

Browse Source
This commit is contained in:
epenet 2024-08-22 14:50:45 +02:00 committed by GitHub
parent 4e3bd55c73
commit 403db6277f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 15 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
tests/components/lcn/test_config_flow.py
View File

@ -129,7 +129,7 @@ async def test_show_form(hass: HomeAssistant) -> None:
assert result["step_id"] == "user" assert result["step_id"] == "user"
async def test_step_user(hass): async def test_step_user(hass: HomeAssistant) -> None:
"""Test for user step.""" """Test for user step."""
with ( with (
patch("pypck.connection.PchkConnectionManager.async_connect"), patch("pypck.connection.PchkConnectionManager.async_connect"),
@ -150,7 +150,9 @@ async def test_step_user(hass):
} }
async def test_step_user_existing_host(hass, entry): async def test_step_user_existing_host(
hass: HomeAssistant, entry: MockConfigEntry
) -> None:
"""Test for user defined host already exists.""" """Test for user defined host already exists."""
entry.add_to_hass(hass) entry.add_to_hass(hass)
@ -172,7 +174,9 @@ async def test_step_user_existing_host(hass, entry):
(TimeoutError, {CONF_BASE: "connection_refused"}), (TimeoutError, {CONF_BASE: "connection_refused"}),
], ],
) )
async def test_step_user_error(hass, error, errors): async def test_step_user_error(
hass: HomeAssistant, error: type[Exception], errors: dict[str, str]
) -> None:
"""Test for error in user step is handled correctly.""" """Test for error in user step is handled correctly."""
with patch( with patch(
"pypck.connection.PchkConnectionManager.async_connect", side_effect=error "pypck.connection.PchkConnectionManager.async_connect", side_effect=error
@ -187,7 +191,7 @@ async def test_step_user_error(hass, error, errors):
assert result["errors"] == errors assert result["errors"] == errors
async def test_step_reconfigure(hass, entry): async def test_step_reconfigure(hass: HomeAssistant, entry: MockConfigEntry) -> None:
"""Test for reconfigure step.""" """Test for reconfigure step."""
entry.add_to_hass(hass) entry.add_to_hass(hass)
old_entry_data = entry.data.copy() old_entry_data = entry.data.copy()
@ -222,7 +226,12 @@ async def test_step_reconfigure(hass, entry):
(TimeoutError, {CONF_BASE: "connection_refused"}), (TimeoutError, {CONF_BASE: "connection_refused"}),
], ],
) )
async def test_step_reconfigure_error(hass, entry, error, errors): async def test_step_reconfigure_error(
hass: HomeAssistant,
entry: MockConfigEntry,
error: type[Exception],
errors: dict[str, str],
) -> None:
"""Test for error in reconfigure step is handled correctly.""" """Test for error in reconfigure step is handled correctly."""
entry.add_to_hass(hass) entry.add_to_hass(hass)
with patch( with patch(
@ -242,7 +251,7 @@ async def test_step_reconfigure_error(hass, entry, error, errors):
assert result["errors"] == errors assert result["errors"] == errors
async def test_validate_connection(): async def test_validate_connection() -> None:
"""Test the connection validation.""" """Test the connection validation."""
data = CONNECTION_DATA.copy() data = CONNECTION_DATA.copy()