fix: update disconnect method to close transport directly

Author: JakubAndrysek

src/wokwi_client/client.py

@@ -49,7 +49,8 @@ def __init__(self, token: str, server: Optional[str] = None):
         self._transport = Transport(token, server or DEFAULT_WS_URL)
         self.last_pause_nanos = 0
         self._transport.add_event_listener("sim:pause", self._on_pause)
-        self._pause_queue = EventQueue(self._transport, "sim:pause")
+        # Lazily create in an active event loop (important for py3.9 and sync client)
+        self._pause_queue: Optional[EventQueue] = None
         self._serial_monitor_tasks: set[asyncio.Task[None]] = set()
 
     async def connect(self) -> dict[str, Any]:
@@ -181,6 +182,8 @@ async def wait_until_simulation_time(self, seconds: float) -> None:
         await pause(self._transport)
         remaining_nanos = seconds * 1e9 - self.last_pause_nanos
         if remaining_nanos > 0:
+            if self._pause_queue is None:
+                self._pause_queue = EventQueue(self._transport, "sim:pause")
             self._pause_queue.flush()
             await resume(self._transport, int(remaining_nanos))
             await self._pause_queue.get()

src/wokwi_client/client_sync.py

@@ -119,7 +119,7 @@ def disconnect(self) -> None:
 
         # (2) Disconnect transport
         with contextlib.suppress(Exception):
-            self._call(self._async_client.disconnect())
+            self._call(self._async_client._transport.close())
 
         # (3) Stop loop / join thread
         if self._loop.is_running():

src/wokwi_client/event_queue.py

@@ -23,12 +23,16 @@ class EventQueue:
     """A queue for events from a specific event type."""
 
     def __init__(self, transport: Transport, event_type: str) -> None:
+        # Bind the queue to the current running loop (important for py3.9)
+        self._loop = asyncio.get_running_loop()
         self._queue: asyncio.Queue[EventMessage] = asyncio.Queue()
         self._transport = transport
         self._event_type = event_type
 
         def listener(event: EventMessage) -> None:
-            self._queue.put_nowait(event)
+            # Ensure we enqueue on the loop that owns the queue, even if
+            # the listener is invoked from a different loop/thread.
+            self._loop.call_soon_threadsafe(self._queue.put_nowait, event)
 
         self._listener = listener
         self._transport.add_event_listener(self._event_type, self._listener)