From 5381b482ef2f52e1f055ff059f34d770ad14eddc Mon Sep 17 00:00:00 2001 From: Andrew-Chen-Wang Date: Fri, 1 May 2020 10:43:17 -0400 Subject: [PATCH] Added ping test for websocket --- .../users/tests/test_socket.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/test_socket.py b/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/test_socket.py index 56132c571..242402029 100644 --- a/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/test_socket.py +++ b/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/test_socket.py @@ -22,3 +22,16 @@ def test_accept_connection(): is_open = loop.run_until_complete(open_connection(url)) assert is_open loop.close() + + +def test_ping(): + async def ping(url): + async with connect(url) as websocket: + await websocket.send("ping") + return await websocket.recv() + + with run_server() as url: + loop = new_event_loop() + received_message = loop.run_until_complete(ping(url)) + assert received_message == "pong" + loop.close()