mirror of
https://github.com/home-assistant/core.git
synced 2025-07-21 12:17:07 +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 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…
x
Reference in New Issue
Block a user