jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-21 12:17:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
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
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_util/aiohttp.py
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):