mirror of https://gitlab.com/litecord/litecord.git
123 lines
3.5 KiB
Python
123 lines
3.5 KiB
Python
"""
|
|
|
|
Litecord
|
|
Copyright (C) 2018-2021 Luna Mendes and Litecord Contributors
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, version 3 of the License.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
"""
|
|
|
|
import pytest
|
|
import secrets
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_me(test_cli_user):
|
|
resp = await test_cli_user.get("/api/v6/users/@me")
|
|
|
|
assert resp.status_code == 200
|
|
rjson = await resp.json
|
|
assert isinstance(rjson, dict)
|
|
|
|
# incomplete user assertions, but should be enough
|
|
assert isinstance(rjson["id"], str)
|
|
assert isinstance(rjson["username"], str)
|
|
assert isinstance(rjson["discriminator"], str)
|
|
assert rjson["avatar"] is None or isinstance(rjson["avatar"], str)
|
|
assert isinstance(rjson["flags"], int)
|
|
assert isinstance(rjson["bot"], bool)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_me_guilds(test_cli_user):
|
|
resp = await test_cli_user.get("/api/v6/users/@me/guilds")
|
|
|
|
assert resp.status_code == 200
|
|
rjson = await resp.json
|
|
assert isinstance(rjson, list)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_profile_self(test_cli_user):
|
|
user_id = test_cli_user.user["id"]
|
|
resp = await test_cli_user.get(f"/api/v6/users/{user_id}/profile")
|
|
|
|
assert resp.status_code == 200
|
|
rjson = await resp.json
|
|
assert isinstance(rjson, dict)
|
|
assert isinstance(rjson["user"], dict)
|
|
assert isinstance(rjson["connected_accounts"], list)
|
|
assert rjson["premium_since"] is None or isinstance(rjson["premium_since"], str)
|
|
assert isinstance(rjson["mutual_guilds"], list)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_create_user(test_cli):
|
|
"""Test the creation and deletion of a user."""
|
|
username = secrets.token_hex(4)
|
|
_email = secrets.token_hex(5)
|
|
email = f"{_email}@{_email}.com"
|
|
password = secrets.token_hex(6)
|
|
|
|
resp = await test_cli.post(
|
|
"/api/v6/auth/register",
|
|
json={"username": username, "email": email, "password": password},
|
|
)
|
|
|
|
assert resp.status_code == 200
|
|
rjson = await resp.json
|
|
|
|
assert isinstance(rjson, dict)
|
|
token = rjson["token"]
|
|
assert isinstance(token, str)
|
|
|
|
resp = await test_cli.get("/api/v6/users/@me", headers={"Authorization": token})
|
|
|
|
assert resp.status_code == 200
|
|
rjson = await resp.json
|
|
assert rjson["username"] == username
|
|
assert rjson["email"] == email
|
|
|
|
resp = await test_cli.post(
|
|
"/api/v6/users/@me/delete",
|
|
headers={"Authorization": token},
|
|
json={"password": password},
|
|
)
|
|
|
|
assert resp.status_code == 204
|
|
|
|
await test_cli.app.db.execute(
|
|
"""
|
|
DELETE FROM users WHERE id = $1
|
|
""",
|
|
int(rjson["id"]),
|
|
)
|
|
|
|
|
|
WANTED_BIO = "hello world!"
|
|
WANTED_ACCENT_COLOR = 0x424242
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_patch_me_bio_accent_color(test_cli_user):
|
|
resp = await test_cli_user.patch(
|
|
"/api/v6/users/@me",
|
|
json={"bio": WANTED_BIO, "accent_color": WANTED_ACCENT_COLOR},
|
|
)
|
|
|
|
assert resp.status_code == 200
|
|
rjson = await resp.json
|
|
assert isinstance(rjson, dict)
|
|
assert rjson["bio"] == WANTED_BIO
|
|
assert rjson["accent_color"] == WANTED_ACCENT_COLOR
|