1
0
mirror of https://github.com/home-assistant/core.git synced 2025-12-24 12:59:34 +00:00

Fix tests typing helper (#86956)

This commit is contained in:
epenet
2023-01-31 08:48:35 +01:00
committed by GitHub
parent 33ede351f0
commit 7db166b515
4 changed files with 14 additions and 16 deletions

View File

@@ -7,7 +7,6 @@ import logging
from typing import Any
from unittest.mock import AsyncMock, Mock, patch
from aiohttp import ClientWebSocketResponse
import pytest
from homeassistant import config_entries, data_entry_flow
@@ -31,6 +30,7 @@ from homeassistant.helpers import config_entry_oauth2_flow
from homeassistant.setup import async_setup_component
from tests.common import MockConfigEntry, mock_platform
from tests.typing import WebSocketGenerator
CLIENT_ID = "some-client-id"
CLIENT_SECRET = "some-client-secret"
@@ -211,9 +211,7 @@ ClientFixture = Callable[[], Client]
@pytest.fixture
async def ws_client(
hass_ws_client: Callable[[...], ClientWebSocketResponse]
) -> ClientFixture:
async def ws_client(hass_ws_client: WebSocketGenerator) -> ClientFixture:
"""Fixture for creating the test websocket client."""
async def create_client() -> Client: