fix: remove listen

realtime/_async/client.py

@@ -78,6 +78,8 @@ def __init__(
         self.max_retries = max_retries
         self.initial_backoff = initial_backoff
         self.timeout = DEFAULT_TIMEOUT
+        self._listen_task: Optional[asyncio.Task] = None
+        self._heartbeat_task: Optional[asyncio.Task] = None
 
     async def _listen(self) -> None:
         """
@@ -154,11 +156,15 @@ async def connect(self) -> None:
             f"Failed to establish WebSocket connection after {self.max_retries} attempts"
         )
 
-    async def listen(self):
-        await asyncio.gather(self._listen(), self._heartbeat())
+    # async def listen(self):
+    #     await asyncio.gather(self._listen(), self._heartbeat())
 
     async def _on_connect(self):
         self.is_connected = True
+
+        self._listen_task = asyncio.create_task(self._listen())
+        self._heartbeat_task = asyncio.create_task(self._heartbeat())
+
         await self._flush_send_buffer()
 
     async def _flush_send_buffer(self):
@@ -180,6 +186,8 @@ async def close(self) -> None:
 
         await self.ws_connection.close()
         self.is_connected = False
+        self._listen_task.cancel()
+        self._heartbeat_task.cancel()
 
     async def _heartbeat(self) -> None:
         while self.is_connected:

tests/test_connection.py

@@ -58,7 +58,6 @@ async def test_set_auth(socket: AsyncRealtimeClient):
 @pytest.mark.asyncio
 async def test_broadcast_events(socket: AsyncRealtimeClient):
     await socket.connect()
-    listen_task = asyncio.create_task(socket.listen())
 
     channel = socket.channel(
         "test-broadcast", params={"config": {"broadcast": {"self": True}}}
@@ -94,15 +93,13 @@ def broadcast_callback(payload):
     assert received_events[2]["payload"]["message"] == "Event 3"
 
     await socket.close()
-    listen_task.cancel()
 
 
 @pytest.mark.asyncio
 async def test_postgrest_changes(socket: AsyncRealtimeClient):
     token = await access_token()
 
     await socket.connect()
-    listen_task = asyncio.create_task(socket.listen())
 
     await socket.set_auth(token)
 
@@ -192,7 +189,6 @@ def delete_callback(payload):
     assert received_events["delete"] == [delete]
 
     await socket.close()
-    listen_task.cancel()
 
 
 async def create_todo(access_token: str, todo: dict) -> str:

tests/test_presence.py

@@ -27,8 +27,6 @@ def socket() -> AsyncRealtimeClient:
 async def test_presence(socket: AsyncRealtimeClient):
     await socket.connect()
 
-    listen_task = asyncio.create_task(socket.listen())
-
     channel: AsyncRealtimeChannel = socket.channel("room")
 
     join_events: List[Tuple[str, List[Dict], List[Dict]]] = []
@@ -106,7 +104,6 @@ def on_leave(key: str, current_presences: List[Dict], left_presences: List[Dict]
     assert leave_events[0] != leave_events[1]
 
     await socket.close()
-    listen_task.cancel()
 
 
 def test_transform_state_raw_presence_state():