tests: assign the WebsocketError to an static variable

Signed-off-by: Eric Callahan <arksine.code@gmail.com>
2022-02-19 15:16:15 -05:00 · 2022-02-19 15:16:15 -05:00 · 2b7a5ff035
parent d630b79e8b
commit 2b7a5ff035
1 changed files with 1 additions and 0 deletions
--- a/tests/fixtures/websocket_client.py
+++ b/tests/fixtures/websocket_client.py
@ -24,6 +24,7 @@ class WebsocketError(Exception):
        self.code = code

 class WebsocketClient:
+    error = WebsocketError
    def __init__(self,
                 type: str = "ws",
                 port: int = 7010