Add type hints for requests_mock.Mocker in test fixtures (#118678)

This commit is contained in:
epenet 2024-06-03 10:30:08 +02:00 committed by GitHub
parent 9be972b13e
commit 6cf7889c38
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 6 additions and 3 deletions

View file

@ -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"))

View file

@ -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",

View file

@ -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",