Fixed imports

Author: Joel Collins

tests/test_server_default_views_socket_handler.py

@@ -1,4 +1,7 @@
-from labthings.server.default_views.sockets import socket_handler
+from labthings.server.default_views.sockets import (
+    socket_handler,
+    process_socket_message,
+)
 
 
 def test_socket_handler(thing_ctx, fake_websocket):
@@ -8,3 +11,9 @@ def test_socket_handler(thing_ctx, fake_websocket):
         # Only responses should be announcing new subscribers
         for response in ws.responses:
             assert '"message": "Added subscriber' in response
+
+
+### Will need regular updating as new message handlers are added
+def test_process_socket_message():
+    assert process_socket_message("message") is None
+    assert process_socket_message(None) is None

tests/test_server_sockets.py

@@ -124,15 +124,3 @@ def ws_view_func(ws):
     with ws_client as c:
         c.environ_base["HTTP_COOKIE"] = {"key": "value"}
         assert c.connect("/", message="hello") == ["hello"]
-
-
-def test_socket_handler_loop(fake_websocket):
-    ws = fake_websocket("hello", recieve_once=True)
-
-    gsocket.socket_handler_loop(ws)
-
-
-### Will need regular updating as new message handlers are added
-def test_process_socket_message():
-    assert gsocket.process_socket_message("message") is None
-    assert gsocket.process_socket_message(None) is None