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

tests: clean up code reuse in invites

This commit is contained in:
NotNite 2021-09-20 19:58:29 -04:00
parent d2c0b9ac61
commit 4cae4e240d
No known key found for this signature in database
GPG Key ID: BD91A5402CCEB08A
1 changed files with 23 additions and 35 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

58
tests/test_invites.py
View File

@ -23,42 +23,42 @@ from litecord.enums import MessageType
pytestmark = pytest.mark.asyncio pytestmark = pytest.mark.asyncio
async def test_invite_create(test_cli_user): async def _create_invite(test_cli_user, guild, channel):
""" "Test the creation of an invite."""
guild = await test_cli_user.create_guild()
channel = guild.channels[0]
resp = await test_cli_user.post( resp = await test_cli_user.post(
f'/api/v9/channels/{channel["id"]}/invites', json={} f'/api/v9/channels/{channel["id"]}/invites', json={}
) )
assert resp.status_code == 200 assert resp.status_code == 200
rjson = await resp.json rjson = await resp.json
assert rjson["channel"]["id"] == channel["id"] assert rjson["channel"]["id"] == channel["id"]
assert rjson["guild"]["id"] == str(guild.id) assert rjson["guild"]["id"] == str(guild.id)
return rjson
async def _join_invite(test_cli_user, invite, user):
resp = await test_cli_user.post(
f'/api/v9/invites/{invite["code"]}', headers={"Authorization": user.token}
)
assert resp.status_code == 200
async def test_invite_create(test_cli_user):
"""Test the creation of an invite."""
guild = await test_cli_user.create_guild()
channel = guild.channels[0]
await _create_invite(test_cli_user, guild, channel)
async def test_invite_join(test_cli_user): async def test_invite_join(test_cli_user):
""" "Test the ability to create & join an invite.""" """Test the ability to create & join an invite."""
guild = await test_cli_user.create_guild() guild = await test_cli_user.create_guild()
channel = guild.channels[0] channel = guild.channels[0]
resp = await test_cli_user.post( invite = await _create_invite(test_cli_user, guild, channel)
f'/api/v9/channels/{channel["id"]}/invites', json={}
)
assert resp.status_code == 200
rjson = await resp.json
invite_code = rjson["code"]
assert rjson["channel"]["id"] == channel["id"]
assert rjson["guild"]["id"] == str(guild.id)
user = await test_cli_user.create_user() user = await test_cli_user.create_user()
resp = await test_cli_user.post( await _join_invite(test_cli_user, invite, user)
f"/api/v9/invites/{invite_code}", headers={"Authorization": user.token}
)
assert resp.status_code == 200
async def test_invite_system_message(test_cli_user): async def test_invite_system_message(test_cli_user):
@ -79,22 +79,10 @@ async def test_invite_system_message(test_cli_user):
assert resp.status_code == 200 assert resp.status_code == 200
rjson = await resp.json rjson = await resp.json
assert rjson["system_channel_id"] == channel_id invite = await _create_invite(test_cli_user, guild, channel)
resp = await test_cli_user.post(f"/api/v9/channels/{channel_id}/invites", json={})
assert resp.status_code == 200
rjson = await resp.json
invite_code = rjson["code"]
assert rjson["channel"]["id"] == channel_id
assert rjson["guild"]["id"] == str(guild.id)
user = await test_cli_user.create_user() user = await test_cli_user.create_user()
resp = await test_cli_user.post( await _join_invite(test_cli_user, invite, user)
f"/api/v9/invites/{invite_code}", headers={"Authorization": user.token}
)
assert resp.status_code == 200
resp = await test_cli_user.get(f"/api/v9/channels/{channel_id}/messages") resp = await test_cli_user.get(f"/api/v9/channels/{channel_id}/messages")
assert resp.status_code == 200 assert resp.status_code == 200