From 6cf7889c38898f5fa791f5884a655101f44db7d7 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Mon, 3 Jun 2024 10:30:08 +0200 Subject: [PATCH] Add type hints for requests_mock.Mocker in test fixtures (#118678) --- tests/components/abode/conftest.py | 3 ++- tests/components/ecobee/conftest.py | 3 ++- tests/components/vultr/conftest.py | 3 ++- 3 files changed, 6 insertions(+), 3 deletions(-) diff --git a/tests/components/abode/conftest.py b/tests/components/abode/conftest.py index 0e5e24b24f4..8e42dba4d87 100644 --- a/tests/components/abode/conftest.py +++ b/tests/components/abode/conftest.py @@ -5,6 +5,7 @@ from unittest.mock import AsyncMock, patch from jaraco.abode.helpers import urls as URL import pytest +from requests_mock import Mocker from tests.common import load_fixture from tests.components.light.conftest import mock_light_profiles # noqa: F401 @@ -20,7 +21,7 @@ def mock_setup_entry() -> Generator[AsyncMock, None, None]: @pytest.fixture(autouse=True) -def requests_mock_fixture(requests_mock) -> None: +def requests_mock_fixture(requests_mock: Mocker) -> None: """Fixture to provide a requests mocker.""" # Mocks the login response for abodepy. requests_mock.post(URL.LOGIN, text=load_fixture("login.json", "abode")) diff --git a/tests/components/ecobee/conftest.py b/tests/components/ecobee/conftest.py index 27d5a949c58..68a17dbfe00 100644 --- a/tests/components/ecobee/conftest.py +++ b/tests/components/ecobee/conftest.py @@ -4,6 +4,7 @@ from collections.abc import Generator from unittest.mock import MagicMock, patch import pytest +from requests_mock import Mocker from homeassistant.components.ecobee import ECOBEE_API_KEY, ECOBEE_REFRESH_TOKEN @@ -11,7 +12,7 @@ from tests.common import load_fixture, load_json_object_fixture @pytest.fixture(autouse=True) -def requests_mock_fixture(requests_mock): +def requests_mock_fixture(requests_mock: Mocker) -> None: """Fixture to provide a requests mocker.""" requests_mock.get( "https://api.ecobee.com/1/thermostat", diff --git a/tests/components/vultr/conftest.py b/tests/components/vultr/conftest.py index f8ecd1cf321..ae0ce9d6886 100644 --- a/tests/components/vultr/conftest.py +++ b/tests/components/vultr/conftest.py @@ -4,6 +4,7 @@ import json from unittest.mock import patch import pytest +from requests_mock import Mocker from homeassistant.components import vultr from homeassistant.core import HomeAssistant @@ -14,7 +15,7 @@ from tests.common import load_fixture @pytest.fixture(name="valid_config") -def valid_config(hass: HomeAssistant, requests_mock): +def valid_config(hass: HomeAssistant, requests_mock: Mocker) -> None: """Load a valid config.""" requests_mock.get( "https://api.vultr.com/v1/account/info?api_key=ABCDEFG1234567",