1
0
mirror of https://github.com/home-assistant/core.git synced 2025-12-24 21:06:19 +00:00

Add type hints to integration tests (b-c) (#87698)

This commit is contained in:
epenet
2023-02-08 18:08:43 +01:00
committed by GitHub
parent 1a414f1433
commit 807c69f621
66 changed files with 548 additions and 350 deletions

View File

@@ -11,7 +11,7 @@ from homeassistant.core import HomeAssistant
from tests.common import MockConfigEntry
async def test_form(hass):
async def test_form(hass: HomeAssistant) -> None:
"""Test we get the form."""
result = await hass.config_entries.flow.async_init(
@@ -49,7 +49,7 @@ async def test_form(hass):
assert len(mock_setup_entry.mock_calls) == 1
async def test_form_2fa(hass):
async def test_form_2fa(hass: HomeAssistant) -> None:
"""Test we get the 2fa form."""
result = await hass.config_entries.flow.async_init(
@@ -91,7 +91,7 @@ async def test_form_2fa(hass):
assert len(mock_setup_entry.mock_calls) == 1
async def test_form_2fa_connect_error(hass):
async def test_form_2fa_connect_error(hass: HomeAssistant) -> None:
"""Test we report a connect error during 2fa setup."""
result = await hass.config_entries.flow.async_init(
@@ -130,7 +130,7 @@ async def test_form_2fa_connect_error(hass):
assert result3["errors"] == {"base": "cannot_connect"}
async def test_form_2fa_invalid_key(hass):
async def test_form_2fa_invalid_key(hass: HomeAssistant) -> None:
"""Test we report an error if key is invalid."""
result = await hass.config_entries.flow.async_init(
@@ -171,7 +171,7 @@ async def test_form_2fa_invalid_key(hass):
assert result3["errors"] == {"base": "invalid_access_token"}
async def test_form_2fa_unknown_error(hass):
async def test_form_2fa_unknown_error(hass: HomeAssistant) -> None:
"""Test we report an unknown error during 2fa setup."""
result = await hass.config_entries.flow.async_init(
@@ -210,7 +210,7 @@ async def test_form_2fa_unknown_error(hass):
assert result3["errors"] == {"base": "unknown"}
async def test_form_invalid_auth(hass):
async def test_form_invalid_auth(hass: HomeAssistant) -> None:
"""Test we handle invalid auth."""
result = await hass.config_entries.flow.async_init(
DOMAIN, context={"source": config_entries.SOURCE_USER}
@@ -228,7 +228,7 @@ async def test_form_invalid_auth(hass):
assert result2["errors"] == {"base": "invalid_auth"}
async def test_form_unknown_error(hass):
async def test_form_unknown_error(hass: HomeAssistant) -> None:
"""Test we handle unknown error at startup."""
result = await hass.config_entries.flow.async_init(
DOMAIN, context={"source": config_entries.SOURCE_USER}
@@ -246,7 +246,7 @@ async def test_form_unknown_error(hass):
assert result2["errors"] == {"base": "unknown"}
async def test_reauth_shows_user_step(hass):
async def test_reauth_shows_user_step(hass: HomeAssistant) -> None:
"""Test reauth shows the user form."""
result = await hass.config_entries.flow.async_init(
DOMAIN,