Fix echo server

lexiforest · Dec 27, 2023 · f361e91 · f361e91
1 parent 905cf94
commit f361e91
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/tests/unittest/conftest.py b/tests/unittest/conftest.py
@@ -560,11 +560,12 @@ async def watch_restarts(self):  # pragma: nocover
 
 
 async def echo(websocket):
-    name = (await websocket.recv()).decode()
-    # print(f"<<< {name}")
+    while True:
+        name = (await websocket.recv()).decode()
+        # print(f"<<< {name}")
 
-    await websocket.send(name)
-    # print(f">>> {name}")
+        await websocket.send(name)
+        # print(f">>> {name}")
 
 
 class TestWebsocketServer:

diff --git a/tests/unittest/test_websockets.py b/tests/unittest/test_websockets.py
@@ -16,8 +16,7 @@ def test_hello(ws_server):
 
 def test_hello_twice(ws_server):
     with Session() as s:
-        # w = s.ws_connect(ws_server.url)
-        w = s.ws_connect("ws://echo.websocket.events")
+        w = s.ws_connect(ws_server.url)
 
         w.send(b"Bar")
         reply, _ = w.recv()