velvox
/
litecord
mirror of https://gitlab.com/litecord/litecord.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

test_websocket: add a _close helper

This commit is contained in:
Luna 2019-03-26 23:47:52 -03:00
parent 8d9ac79dbc
commit 07c66d121a
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/test_websocket.py
View File

@ -42,6 +42,10 @@ async def _json_send_op(conn, opcode, data=None):
}) })
async def _close(conn):
await conn.close(1000, 'test end')
async def get_gw(test_cli) -> str: async def get_gw(test_cli) -> str:
"""Get the Gateway URL.""" """Get the Gateway URL."""
gw_resp = await test_cli.get('/api/v6/gateway') gw_resp = await test_cli.get('/api/v6/gateway')
@ -128,7 +132,7 @@ async def test_ready_fields(test_cli):
assert isinstance(data['guilds'], list) assert isinstance(data['guilds'], list)
assert isinstance(data['session_id'], str) assert isinstance(data['session_id'], str)
await conn.close(1000, 'test end') await _close(conn)
@pytest.mark.asyncio @pytest.mark.asyncio
@ -157,3 +161,5 @@ async def test_heartbeat(test_cli):
recv = await _json(conn) recv = await _json(conn)
assert isinstance(recv, dict) assert isinstance(recv, dict)
assert recv['op'] == OP.HEARTBEAT_ACK assert recv['op'] == OP.HEARTBEAT_ACK
await _close(conn)