evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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