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

Improve hass_ws_client type hint in tests (#89703)

Browse source
This commit is contained in:
epenet 2023-03-14 16:31:40 +01:00 • committed by GitHub
parent 1bc4802c04
commit 71dc98a39c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
17 changed files with 52 additions and 71 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/components/mysensors/test_init.py
View file

@ -1,9 +1,6 @@
"""Test function in __init__.py."""
from __future__ import annotations
from collections.abc import Awaitable, Callable
from aiohttp import ClientWebSocketResponse
from mysensors import BaseSyncGateway
from mysensors.sensor import Sensor
@ -13,6 +10,7 @@ from homeassistant.helpers import device_registry as dr, entity_registry as er
from homeassistant.setup import async_setup_component
from tests.common import MockConfigEntry
from tests.typing import WebSocketGenerator
async def test_remove_config_entry_device(
@ -20,7 +18,7 @@ async def test_remove_config_entry_device(
gps_sensor: Sensor,
integration: MockConfigEntry,
gateway: BaseSyncGateway,
hass_ws_client: Callable[[HomeAssistant], Awaitable[ClientWebSocketResponse]],
hass_ws_client: WebSocketGenerator,
) -> None:
"""Test that a device can be removed ok."""
entity_id = "sensor.gps_sensor_1_1"