Add missing hass type in tests/test_util (#124043)
This commit is contained in:
parent
ea52acd7bd
commit
0a846cfca8
1 changed files with 3 additions and 1 deletions
|
@ -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):
|
||||
|
|
Loading…
Add table
Reference in a new issue