Process control messages before shell messages if no separate control

thread

Author: ianthomas23

ipykernel/kernelbase.py

@@ -492,6 +492,19 @@ async def process_one(self, wait=True):
                 t, dispatch, args = self.msg_queue.get_nowait()
             except (asyncio.QueueEmpty, QueueEmpty):
                 return
+
+        if self.control_thread is None:
+            # If there isn't a separate control thread then this main thread handles both shell
+            # and control messages. Before processing a shell message need to flush all control
+            # messages and allow them all to be processed.
+            await asyncio.sleep(0)
+            self.control_stream.flush()
+
+            socket = self.control_stream.socket
+            while socket.poll(1):
+                await asyncio.sleep(0)
+                self.control_stream.flush()
+
         await dispatch(*args)
 
     async def dispatch_queue(self):