coverage

Author: dmulcahey

tests/websocket/test_websocket_server_client.py

@@ -5,7 +5,7 @@
 import pytest
 
 from tests.conftest import CombinedWebsocketGateways
-from zha.application.gateway import WebSocketServerGateway
+from zha.application.gateway import WebSocketClientGateway, WebSocketServerGateway
 from zha.application.helpers import ZHAData
 from zha.websocket.client.client import Client
 
@@ -33,6 +33,13 @@ async def test_server_client_connect_disconnect(
         assert "not connected" in repr(client)
         assert not client.connected
 
+        async with WebSocketClientGateway(zha_data) as client_gateway:
+            assert client_gateway.client.connected
+            assert client_gateway.client._listen_task is not None
+
+        assert not client_gateway.client.connected
+        assert client_gateway.client._listen_task is None
+
     assert not gateway.is_serving
     assert gateway._ws_server is None