2
2
3
3
from __future__ import annotations
4
4
5
- from zha .application .gateway import WebSocketGateway
5
+ from zha .application .gateway import WebSocketServerGateway
6
6
from zha .application .helpers import ZHAData
7
7
from zha .websocket .client .client import Client
8
8
from zha .websocket .client .controller import Controller
@@ -14,7 +14,7 @@ async def test_server_client_connect_disconnect(
14
14
) -> None :
15
15
"""Tests basic connect/disconnect logic."""
16
16
17
- async with WebSocketGateway (zha_data ) as gateway :
17
+ async with WebSocketServerGateway (zha_data ) as gateway :
18
18
assert gateway .is_serving
19
19
assert gateway ._ws_server is not None
20
20
@@ -37,17 +37,17 @@ async def test_server_client_connect_disconnect(
37
37
38
38
39
39
async def test_client_message_id_uniqueness (
40
- connected_client_and_server : tuple [Controller , WebSocketGateway ],
40
+ connected_client_and_server : tuple [Controller , WebSocketServerGateway ],
41
41
) -> None :
42
42
"""Tests that client message IDs are unique."""
43
- controller , gateway = connected_client_and_server
43
+ controller , _ = connected_client_and_server
44
44
45
45
ids = [controller .client .new_message_id () for _ in range (1000 )]
46
46
assert len (ids ) == len (set (ids ))
47
47
48
48
49
49
async def test_client_stop_server (
50
- connected_client_and_server : tuple [Controller , WebSocketGateway ],
50
+ connected_client_and_server : tuple [Controller , WebSocketServerGateway ],
51
51
) -> None :
52
52
"""Tests that the client can stop the server."""
53
53
controller , gateway = connected_client_and_server
0 commit comments