fix for freeze issue

Author: willmcgugan

src/textual/app.py

@@ -2011,15 +2011,15 @@ async def _prune_node(self, root: Widget) -> None:
         for children in reversed(node_children):
             # Closing children can be done asynchronously.
             close_messages = [
-                child._close_messages() for child in children if child._running
+                child._close_messages(wait=True) for child in children if child._running
             ]
             # TODO: What if a message pump refuses to exit?
             if close_messages:
                 await asyncio.gather(*close_messages)
                 for child in children:
                     self._unregister(child)
 
-        await root._close_messages()
+        await root._close_messages(wait=False)
         self._unregister(root)
 
     async def action_check_bindings(self, key: str) -> None:

src/textual/message_pump.py

@@ -284,7 +284,7 @@ def _close_messages_no_wait(self) -> None:
     async def _on_close_messages(self, message: messages.CloseMessages) -> None:
         await self._close_messages()
 
-    async def _close_messages(self) -> None:
+    async def _close_messages(self, wait: bool = True) -> None:
         """Close message queue, and optionally wait for queue to finish processing."""
         if self._closed or self._closing:
             return
@@ -296,7 +296,7 @@ async def _close_messages(self) -> None:
         await self._message_queue.put(events.Unmount(sender=self))
         Reactive._reset_object(self)
         await self._message_queue.put(None)
-        if self._task is not None and asyncio.current_task() != self._task:
+        if wait and self._task is not None and asyncio.current_task() != self._task:
             # Ensure everything is closed before returning
             await self._task