From 5fc99c89d56eaa71d2f6216e9f6f096d54612c94 Mon Sep 17 00:00:00 2001 From: Martin Pitt Date: Mon, 5 Aug 2024 19:56:17 +0200 Subject: [PATCH] FIXUP future cleanup --- test/common/bidi.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/test/common/bidi.py b/test/common/bidi.py index dd78942cfee2..c38b2b90dd1a 100644 --- a/test/common/bidi.py +++ b/test/common/bidi.py @@ -129,7 +129,8 @@ async def close(self) -> None: def ws_done_callback(self, future): for fut in self.pending_commands.values(): - fut.set_exception(WebdriverError("websocket closed")) + if not fut.done(): + fut.set_exception(WebdriverError("websocket closed")) if not future.cancelled(): log_proto.error("ws_reader crashed: %r", future.result())