test_invites: loop instead of code reuse

This commit is contained in:
NotNite 2021-09-21 12:41:37 -04:00
parent 639368bed3
commit 41c9531d13
No known key found for this signature in database
GPG Key ID: BD91A5402CCEB08A
1 changed files with 19 additions and 45 deletions

View File

@ -101,50 +101,24 @@ async def test_leave_join_invite_cycle(test_cli_user):
invite = await _create_invite(test_cli_user, guild, channel) invite = await _create_invite(test_cli_user, guild, channel)
user = await test_cli_user.create_user() user = await test_cli_user.create_user()
# pass 1 for x in range(1, 4):
await _join_invite(test_cli_user, invite, user) print(f"pass {x}")
resp = await test_cli_user.get("/api/v6/users/@me/guilds", as_user=user)
assert resp.status_code == 200
rjson = await resp.json
assert any(incoming_guild["id"] == str(guild.id) for incoming_guild in rjson)
resp = await test_cli_user.delete(
f"/api/v6/users/@me/guilds/{guild.id}", as_user=user
)
assert resp.status_code == 204
resp = await test_cli_user.get("/api/v6/users/@me/guilds", as_user=user)
assert resp.status_code == 200
rjson = await resp.json
for incoming_guild in rjson:
assert incoming_guild["id"] != str(guild.id)
# pass 2 await _join_invite(test_cli_user, invite, user)
await _join_invite(test_cli_user, invite, user) resp = await test_cli_user.get("/api/v6/users/@me/guilds", as_user=user)
resp = await test_cli_user.get("/api/v6/users/@me/guilds", as_user=user) assert resp.status_code == 200
assert resp.status_code == 200 rjson = await resp.json
rjson = await resp.json
assert any(incoming_guild["id"] == str(guild.id) for incoming_guild in rjson)
resp = await test_cli_user.delete(
f"/api/v6/users/@me/guilds/{guild.id}", as_user=user
)
assert resp.status_code == 204
resp = await test_cli_user.get("/api/v6/users/@me/guilds", as_user=user)
assert resp.status_code == 200
rjson = await resp.json
for incoming_guild in rjson:
assert incoming_guild["id"] != str(guild.id)
# pass 3 assert any(incoming_guild["id"] == str(guild.id) for incoming_guild in rjson)
await _join_invite(test_cli_user, invite, user)
resp = await test_cli_user.get("/api/v6/users/@me/guilds", as_user=user) resp = await test_cli_user.delete(
assert resp.status_code == 200 f"/api/v6/users/@me/guilds/{guild.id}", as_user=user
rjson = await resp.json )
assert any(incoming_guild["id"] == str(guild.id) for incoming_guild in rjson) assert resp.status_code == 204
resp = await test_cli_user.delete(
f"/api/v6/users/@me/guilds/{guild.id}", as_user=user resp = await test_cli_user.get("/api/v6/users/@me/guilds", as_user=user)
) assert resp.status_code == 200
assert resp.status_code == 204 rjson = await resp.json
resp = await test_cli_user.get("/api/v6/users/@me/guilds", as_user=user)
assert resp.status_code == 200 for incoming_guild in rjson:
rjson = await resp.json assert incoming_guild["id"] != str(guild.id)
for incoming_guild in rjson:
assert incoming_guild["id"] != str(guild.id)