jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-08-06 12:08:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Adjust nest tests

Browse Source
This commit is contained in:
Erik 2025-05-07 17:13:05 +02:00
parent fdcaf2897a
commit 8aac4777b1
1 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
tests/components/nest/test_init.py
View File

@ -8,7 +8,7 @@ mode (e.g. yaml, ConfigEntry, etc) however some tests override and just run in
relevant modes. relevant modes.
""" """
from collections.abc import Generator from collections.abc import Callable, Generator
import datetime import datetime
from http import HTTPStatus from http import HTTPStatus
import logging import logging
@ -146,23 +146,31 @@ async def test_setup_device_manager_failure(
@pytest.mark.parametrize("token_expiration_time", [EXPIRED_TOKEN_TIMESTAMP]) @pytest.mark.parametrize("token_expiration_time", [EXPIRED_TOKEN_TIMESTAMP])
@pytest.mark.parametrize( @pytest.mark.parametrize(
("token_response_args", "expected_state", "expected_steps"), ("token_response_args", "token_response_exc", "expected_state", "expected_steps"),
[ [
# Cases that retry integration setup # Cases that retry integration setup
( (
{"status": HTTPStatus.INTERNAL_SERVER_ERROR}, {"status": HTTPStatus.INTERNAL_SERVER_ERROR},
lambda: None,
ConfigEntryState.SETUP_RETRY,
[],
),
(
{},
lambda: aiohttp.ClientError("No internet"),
ConfigEntryState.SETUP_RETRY, ConfigEntryState.SETUP_RETRY,
[], [],
), ),
({"exc": aiohttp.ClientError("No internet")}, ConfigEntryState.SETUP_RETRY, []),
# Cases that require the user to reauthenticate in a config flow # Cases that require the user to reauthenticate in a config flow
( (
{"status": HTTPStatus.BAD_REQUEST}, {"status": HTTPStatus.BAD_REQUEST},
lambda: None,
ConfigEntryState.SETUP_ERROR, ConfigEntryState.SETUP_ERROR,
["reauth_confirm"], ["reauth_confirm"],
), ),
( (
{"status": HTTPStatus.FORBIDDEN}, {"status": HTTPStatus.FORBIDDEN},
lambda: None,
ConfigEntryState.SETUP_ERROR, ConfigEntryState.SETUP_ERROR,
["reauth_confirm"], ["reauth_confirm"],
), ),
@ -173,6 +181,7 @@ async def test_expired_token_refresh_error(
setup_base_platform: PlatformSetup, setup_base_platform: PlatformSetup,
aioclient_mock: AiohttpClientMocker, aioclient_mock: AiohttpClientMocker,
token_response_args: dict, token_response_args: dict,
token_response_exc: Callable[[], Exception | None],
expected_state: ConfigEntryState, expected_state: ConfigEntryState,
expected_steps: list[str], expected_steps: list[str],
) -> None: ) -> None:
@ -180,6 +189,7 @@ async def test_expired_token_refresh_error(
aioclient_mock.post( aioclient_mock.post(
OAUTH2_TOKEN, OAUTH2_TOKEN,
exc=token_response_exc(),
**token_response_args, **token_response_args,
) )