Skip to content

Commit 233af4e

Browse files
committed
fix tests
1 parent d6d1b03 commit 233af4e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/textual_dev/service.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -276,11 +276,11 @@ async def close(self) -> None:
276276

277277
# Stop any writes to the websocket first
278278
await self.outgoing_queue.put(None)
279-
await self.outgoing_messages_task
279+
await asyncio.shield(self.outgoing_messages_task)
280280

281281
# Now we can shut the socket down
282282
await self.websocket.close()
283283

284284
# This task is independent of the websocket
285285
await self.incoming_queue.put(None)
286-
await self.incoming_messages_task
286+
await asyncio.shield(self.incoming_messages_task)

0 commit comments

Comments
 (0)