From 660f41772cbd61cadd2ff7c1426f33e2c5d9a7f4 Mon Sep 17 00:00:00 2001 From: Angus Gratton Date: Thu, 17 Dec 2020 14:16:48 +1100 Subject: [PATCH] ci: Fix websocket example test error handling If connection fails for websocket test, a second exception was generated in __exit__ --- examples/protocols/websocket/example_test.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/examples/protocols/websocket/example_test.py b/examples/protocols/websocket/example_test.py index 6dfdccf5d8..56cbc6078a 100644 --- a/examples/protocols/websocket/example_test.py +++ b/examples/protocols/websocket/example_test.py @@ -37,6 +37,7 @@ class Websocket: self.socket.settimeout(10.0) self.send_q = queue.Queue() self.shutdown = Event() + self.conn = None def __enter__(self): try: @@ -55,7 +56,8 @@ class Websocket: self.shutdown.set() self.server_thread.join() self.socket.close() - self.conn.close() + if self.conn: + self.conn.close() def run_server(self): self.conn, address = self.socket.accept() # accept new connection