From 3e5d0eb632313523ce4cfc637eb542294516d80d Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Wed, 14 Aug 2024 15:08:10 +0200 Subject: [PATCH] Improve type hints in owntracks tests (#123905) --- tests/components/owntracks/test_device_tracker.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/components/owntracks/test_device_tracker.py b/tests/components/owntracks/test_device_tracker.py index b4b5c00880c..bc2ae7ce4d8 100644 --- a/tests/components/owntracks/test_device_tracker.py +++ b/tests/components/owntracks/test_device_tracker.py @@ -1,7 +1,7 @@ """The tests for the Owntracks device tracker.""" import base64 -from collections.abc import Callable +from collections.abc import Callable, Generator import json import pickle from unittest.mock import patch @@ -294,7 +294,7 @@ def setup_comp( hass: HomeAssistant, mock_device_tracker_conf: list[Device], mqtt_mock: MqttMockHAClient, -): +) -> None: """Initialize components.""" hass.loop.run_until_complete(async_setup_component(hass, "device_tracker", {})) @@ -1426,7 +1426,7 @@ def mock_cipher(): @pytest.fixture -def config_context(hass, setup_comp): +def config_context(setup_comp: None) -> Generator[None]: """Set up the mocked context.""" patch_load = patch( "homeassistant.components.device_tracker.async_load_config",