From f39dd40be1d9f69eb324315e23f3374c59772942 Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Mon, 3 Jun 2024 10:20:57 +0200
Subject: [PATCH] Add type hints for hass_storage in test fixtures (#118682)

---
 tests/components/smartthings/conftest.py | 8 ++++++--
 tests/components/tag/test_init.py        | 4 ++--
 2 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/tests/components/smartthings/conftest.py b/tests/components/smartthings/conftest.py
index d25cc8849e5..b638b9bbf4f 100644
--- a/tests/components/smartthings/conftest.py
+++ b/tests/components/smartthings/conftest.py
@@ -1,6 +1,7 @@
 """Test configuration and mocks for the SmartThings component."""
 
 import secrets
+from typing import Any
 from unittest.mock import Mock, patch
 from uuid import uuid4
 
@@ -45,6 +46,7 @@ from homeassistant.const import (
     CONF_CLIENT_SECRET,
     CONF_WEBHOOK_ID,
 )
+from homeassistant.core import HomeAssistant
 from homeassistant.setup import async_setup_component
 
 from tests.common import MockConfigEntry
@@ -74,7 +76,9 @@ async def setup_platform(hass, platform: str, *, devices=None, scenes=None):
 
 
 @pytest.fixture(autouse=True)
-async def setup_component(hass, config_file, hass_storage):
+async def setup_component(
+    hass: HomeAssistant, config_file: dict[str, str], hass_storage: dict[str, Any]
+) -> None:
     """Load the SmartThing component."""
     hass_storage[STORAGE_KEY] = {"data": config_file, "version": STORAGE_VERSION}
     await async_process_ha_core_config(
@@ -166,7 +170,7 @@ def installed_apps_fixture(installed_app, locations, app):
 
 
 @pytest.fixture(name="config_file")
-def config_file_fixture():
+def config_file_fixture() -> dict[str, str]:
     """Fixture representing the local config file contents."""
     return {CONF_INSTANCE_ID: str(uuid4()), CONF_WEBHOOK_ID: secrets.token_hex()}
 
diff --git a/tests/components/tag/test_init.py b/tests/components/tag/test_init.py
index db7e9d5dbc7..d2d2bf90a7c 100644
--- a/tests/components/tag/test_init.py
+++ b/tests/components/tag/test_init.py
@@ -55,7 +55,7 @@ def storage_setup(hass: HomeAssistant, hass_storage: dict[str, Any]):
 
 
 @pytest.fixture
-def storage_setup_1_1(hass: HomeAssistant, hass_storage):
+def storage_setup_1_1(hass: HomeAssistant, hass_storage: dict[str, Any]):
     """Storage version 1.1 setup."""
 
     async def _storage(items=None):
@@ -87,7 +87,7 @@ async def test_migration(
     hass_ws_client: WebSocketGenerator,
     storage_setup_1_1,
     freezer: FrozenDateTimeFactory,
-    hass_storage,
+    hass_storage: dict[str, Any],
     snapshot: SnapshotAssertion,
 ) -> None:
     """Test migrating tag store."""