Skip to content

Commit 7bdf0ef

Browse files
committed
rename websocketgateway to websocketservergateway
1 parent 0b11a9b commit 7bdf0ef

File tree

27 files changed

+69
-65
lines changed

27 files changed

+69
-65
lines changed

tests/conftest.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@
2424
import zigpy.zdo.types as zdo_t
2525

2626
from zha.application import Platform
27-
from zha.application.gateway import Gateway, WebSocketGateway
27+
from zha.application.gateway import Gateway, WebSocketServerGateway
2828
from zha.application.helpers import (
2929
AlarmControlPanelOptions,
3030
CoordinatorConfiguration,
@@ -326,7 +326,7 @@ async def connected_client_and_server(
326326
zha_data: ZHAData,
327327
zigpy_app_controller: ControllerApplication,
328328
caplog: pytest.LogCaptureFixture, # pylint: disable=unused-argument
329-
) -> AsyncGenerator[tuple[Controller, WebSocketGateway], None]:
329+
) -> AsyncGenerator[tuple[Controller, WebSocketServerGateway], None]:
330330
"""Return the connected client and server fixture."""
331331

332332
with (
@@ -339,7 +339,7 @@ async def connected_client_and_server(
339339
return_value=zigpy_app_controller,
340340
),
341341
):
342-
ws_gateway = await WebSocketGateway.async_from_config(zha_data)
342+
ws_gateway = await WebSocketServerGateway.async_from_config(zha_data)
343343
await ws_gateway.async_initialize()
344344
await ws_gateway.async_block_till_done()
345345
await ws_gateway.async_initialize_devices_and_entities()

tests/websocket/test_alarm_control_panel.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
import zigpy.zcl.foundation as zcl_f
1010

1111
from zha.application import Platform
12-
from zha.application.gateway import WebSocketGateway as Server
12+
from zha.application.gateway import WebSocketServerGateway as Server
1313
from zha.application.platforms.model import AlarmControlPanelEntity
1414
from zha.websocket.client.controller import Controller
1515
from zha.websocket.client.proxy import DeviceProxy

tests/websocket/test_binary_sensor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
from zigpy.zcl.clusters import general, measurement, security
99

1010
from zha.application.discovery import Platform
11-
from zha.application.gateway import WebSocketGateway as Server
11+
from zha.application.gateway import WebSocketServerGateway as Server
1212
from zha.application.platforms.model import BasePlatformEntity, BinarySensorEntity
1313
from zha.websocket.client.controller import Controller
1414
from zha.websocket.client.proxy import DeviceProxy

tests/websocket/test_button.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
import zigpy.zcl.foundation as zcl_f
1010

1111
from zha.application.discovery import Platform
12-
from zha.application.gateway import WebSocketGateway as Server
12+
from zha.application.gateway import WebSocketServerGateway as Server
1313
from zha.application.platforms.model import BasePlatformEntity, ButtonEntity
1414
from zha.websocket.client.controller import Controller
1515
from zha.websocket.client.proxy import DeviceProxy

tests/websocket/test_client_controller.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
DevicePairingStatus,
1717
RawDeviceInitializedDeviceInfo,
1818
RawDeviceInitializedEvent,
19-
WebSocketGateway as Server,
19+
WebSocketServerGateway as Server,
2020
)
2121
from zha.application.model import DeviceJoinedEvent, DeviceLeftEvent
2222
from zha.application.platforms.model import (
@@ -117,7 +117,7 @@ async def device_switch_2(
117117
) -> Device:
118118
"""Test zha switch platform."""
119119

120-
controller, server = connected_client_and_server
120+
_, server = connected_client_and_server
121121
zigpy_device = create_mock_zigpy_device(
122122
server,
123123
{

tests/websocket/test_number.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
from zigpy.zcl.clusters import general
99

1010
from zha.application.discovery import Platform
11-
from zha.application.gateway import WebSocketGateway as Server
11+
from zha.application.gateway import WebSocketServerGateway as Server
1212
from zha.application.platforms.model import BasePlatformEntity, NumberEntity
1313
from zha.websocket.client.controller import Controller
1414
from zha.websocket.client.proxy import DeviceProxy

tests/websocket/test_siren.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
import zigpy.zcl.foundation as zcl_f
1212

1313
from zha.application.discovery import Platform
14-
from zha.application.gateway import WebSocketGateway as Server
14+
from zha.application.gateway import WebSocketServerGateway as Server
1515
from zha.application.platforms.model import BasePlatformEntity
1616
from zha.websocket.client.controller import Controller
1717
from zha.websocket.client.proxy import DeviceProxy

tests/websocket/test_switch.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414

1515
from tests.common import mock_coro
1616
from zha.application.discovery import Platform
17-
from zha.application.gateway import WebSocketGateway as Server
17+
from zha.application.gateway import WebSocketServerGateway as Server
1818
from zha.application.platforms.model import (
1919
BasePlatformEntity,
2020
SwitchEntity,
@@ -65,7 +65,7 @@ def get_group_entity(
6565
@pytest.fixture
6666
def zigpy_device(connected_client_and_server: tuple[Controller, Server]) -> ZigpyDevice:
6767
"""Device tracker zigpy device."""
68-
controller, server = connected_client_and_server
68+
_, server = connected_client_and_server
6969
zigpy_device = create_mock_zigpy_device(
7070
server,
7171
{

tests/websocket/test_websocket_server_client.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
from __future__ import annotations
44

5-
from zha.application.gateway import WebSocketGateway
5+
from zha.application.gateway import WebSocketServerGateway
66
from zha.application.helpers import ZHAData
77
from zha.websocket.client.client import Client
88
from zha.websocket.client.controller import Controller
@@ -14,7 +14,7 @@ async def test_server_client_connect_disconnect(
1414
) -> None:
1515
"""Tests basic connect/disconnect logic."""
1616

17-
async with WebSocketGateway(zha_data) as gateway:
17+
async with WebSocketServerGateway(zha_data) as gateway:
1818
assert gateway.is_serving
1919
assert gateway._ws_server is not None
2020

@@ -37,17 +37,17 @@ async def test_server_client_connect_disconnect(
3737

3838

3939
async def test_client_message_id_uniqueness(
40-
connected_client_and_server: tuple[Controller, WebSocketGateway],
40+
connected_client_and_server: tuple[Controller, WebSocketServerGateway],
4141
) -> None:
4242
"""Tests that client message IDs are unique."""
43-
controller, gateway = connected_client_and_server
43+
controller, _ = connected_client_and_server
4444

4545
ids = [controller.client.new_message_id() for _ in range(1000)]
4646
assert len(ids) == len(set(ids))
4747

4848

4949
async def test_client_stop_server(
50-
connected_client_and_server: tuple[Controller, WebSocketGateway],
50+
connected_client_and_server: tuple[Controller, WebSocketServerGateway],
5151
) -> None:
5252
"""Tests that the client can stop the server."""
5353
controller, gateway = connected_client_and_server

zha/application/gateway.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -718,7 +718,7 @@ def handle_message( # pylint: disable=unused-argument
718718
self.async_update_device(sender, available=True)
719719

720720

721-
class WebSocketGateway(Gateway):
721+
class WebSocketServerGateway(Gateway):
722722
"""ZHAWSS server implementation."""
723723

724724
def __init__(self, config: ZHAData) -> None:
@@ -835,7 +835,7 @@ async def async_block_till_done(self, wait_background_tasks=False):
835835
await asyncio.sleep(0.001)
836836
await super().async_block_till_done(wait_background_tasks=wait_background_tasks)
837837

838-
async def __aenter__(self) -> WebSocketGateway:
838+
async def __aenter__(self) -> WebSocketServerGateway:
839839
"""Enter the context manager."""
840840
await self.start_server()
841841
return self

0 commit comments

Comments
 (0)