mirror of
https://github.com/home-assistant/core.git
synced 2025-07-22 20:57:21 +00:00
Add missing hass type in tests/test_util (#124043)
This commit is contained in:
parent
ea52acd7bd
commit
0a846cfca8
@ -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):
|
||||||
|
Loading…
x
Reference in New Issue
Block a user