TST: unit tests for sockets with authentication

Author: dmgav

bluesky_httpserver/tests/test_console_output.py

@@ -353,7 +353,8 @@ def __init__(self, api_key=API_KEY_FOR_TESTS, **kwargs):
 
     def run(self):
         websocket_uri = f"ws://{SERVER_ADDRESS}:{SERVER_PORT}/api/console_output/ws"
-        with connect(websocket_uri) as websocket:
+        additional_headers = {"Authorization": f"ApiKey {self._api_key}"}
+        with connect(websocket_uri, additional_headers=additional_headers) as websocket:
             while not self._exit:
                 try:
                     msg_json = websocket.recv(timeout=0.1, decode=False)

bluesky_httpserver/tests/test_system_info_socket.py

@@ -35,7 +35,8 @@ def __init__(self, *, endpoint, api_key=API_KEY_FOR_TESTS, **kwargs):
 
     def run(self):
         websocket_uri = f"ws://{SERVER_ADDRESS}:{SERVER_PORT}/api{self._endpoint}"
-        with connect(websocket_uri) as websocket:
+        additional_headers = {"Authorization": f"ApiKey {self._api_key}"}
+        with connect(websocket_uri, additional_headers=additional_headers) as websocket:
             while not self._exit:
                 try:
                     msg_json = websocket.recv(timeout=0.1, decode=False)