Add missing hass type in tests/test_util (#124043)

This commit is contained in:
epenet 2024-08-16 13:43:43 +02:00 committed by GitHub
parent ea52acd7bd
commit 0a846cfca8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -5,6 +5,7 @@ from collections.abc import Iterator
from contextlib import contextmanager
from http import HTTPStatus
import re
from typing import Any
from unittest import mock
from urllib.parse import parse_qs
@ -19,6 +20,7 @@ from multidict import CIMultiDict
from yarl import URL
from homeassistant.const import EVENT_HOMEASSISTANT_CLOSE
from homeassistant.core import HomeAssistant
from homeassistant.helpers.json import json_dumps
from homeassistant.util.json import json_loads
@ -301,7 +303,7 @@ def mock_aiohttp_client() -> Iterator[AiohttpClientMocker]:
"""Context manager to mock aiohttp client."""
mocker = AiohttpClientMocker()
def create_session(hass, *args, **kwargs):
def create_session(hass: HomeAssistant, *args: Any, **kwargs: Any) -> ClientSession:
session = mocker.create_session(hass.loop)
async def close_session(event):