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:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user