cookiecutter/cookiecutter-django
1
1
Fork 0
mirror of https://github.com/cookiecutter/cookiecutter-django.git synced 2025-08-15 01:14:53 +03:00
Code Issues Packages Projects Releases Wiki Activity

Added ping test for websocket

Browse Source
This commit is contained in:
Andrew-Chen-Wang 2020-05-01 10:43:17 -04:00
parent f9122a55a8
commit 5381b482ef
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/test_socket.py
View File

@ -22,3 +22,16 @@ def test_accept_connection():
is_open = loop.run_until_complete(open_connection(url)) is_open = loop.run_until_complete(open_connection(url))
assert is_open assert is_open
loop.close() 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()