148 lines
4.1 KiB
Python
148 lines
4.1 KiB
Python
import pytest
|
|
from fastapi.testclient import TestClient
|
|
|
|
from src.main import app, get_db
|
|
from tests.base import avatar_index, client, password, username, register_user
|
|
from tests.config.database import clear_db, override_get_db
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_register():
|
|
"""Test the register endpoint"""
|
|
clear_db()
|
|
|
|
response = client.post(
|
|
"/register",
|
|
headers={"Content-Type": "application/json"},
|
|
json={"username": username, "password": password, "avatar_index": avatar_index},
|
|
)
|
|
|
|
assert response.status_code == 200
|
|
assert len(response.json()["access_token"]) > 0
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_register_duplicate_name_should_fail():
|
|
"""Test whether registering a user with an existing username fails"""
|
|
clear_db()
|
|
await register_user()
|
|
|
|
response = client.post(
|
|
"/register",
|
|
headers={"Content-Type": "application/json"},
|
|
json={"username": username, "password": password, "avatar_index": avatar_index},
|
|
)
|
|
|
|
assert response.status_code == 400
|
|
assert "access_token" not in response.json()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_register_without_username_should_fail():
|
|
"""Test whether registering a user without passing a username fails"""
|
|
clear_db()
|
|
|
|
response = client.post(
|
|
"/register",
|
|
headers={"Content-Type": "application/json"},
|
|
json={"password": password, "avatar_index": avatar_index},
|
|
)
|
|
|
|
assert response.status_code == 422
|
|
assert "access_token" not in response.json()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_register_without_password_should_fail():
|
|
"""Test whether registering a user without passing a password fails"""
|
|
clear_db()
|
|
|
|
response = client.post(
|
|
"/register",
|
|
headers={"Content-Type": "application/json"},
|
|
json={"username": username, "avatar_index": avatar_index},
|
|
)
|
|
|
|
assert response.status_code == 422
|
|
assert "access_token" not in response.json()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_register_without_avatar_should_fail():
|
|
"""Test whether registering a user without passing an avatar fails"""
|
|
clear_db()
|
|
|
|
response = client.post(
|
|
"/register",
|
|
headers={"Content-Type": "application/json"},
|
|
json={"username": username, "password": password},
|
|
)
|
|
|
|
# Not ideal that this is 400 instead of 422, but had no other choice than to give this field a default value
|
|
assert response.status_code == 400
|
|
assert "access_token" not in response.json()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_login():
|
|
"""Test the login endpoint"""
|
|
clear_db()
|
|
await register_user()
|
|
|
|
response = client.post(
|
|
"/login",
|
|
headers={"Content-Type": "application/json"},
|
|
json={"username": username, "password": password},
|
|
)
|
|
|
|
assert response.status_code == 200
|
|
assert len(response.json()["access_token"]) > 0
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_login_wrong_password_should_fail():
|
|
clear_db()
|
|
await register_user()
|
|
|
|
wrong_password = password + "extra characters"
|
|
response = client.post(
|
|
"/login",
|
|
headers={"Content-Type": "application/json"},
|
|
json={"username": username, "password": wrong_password},
|
|
)
|
|
|
|
assert response.status_code == 401
|
|
assert "access_token" not in response.json()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_login_without_username_should_fail():
|
|
"""Test whether logging in without passing a username fails"""
|
|
clear_db()
|
|
await register_user()
|
|
|
|
response = client.post(
|
|
"/login",
|
|
headers={"Content-Type": "application/json"},
|
|
json={"password": password},
|
|
)
|
|
|
|
assert response.status_code == 422
|
|
assert "access_token" not in response.json()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_login_without_password_should_fail():
|
|
"""Test whether logging in without passing a password fails"""
|
|
clear_db()
|
|
await register_user()
|
|
|
|
response = client.post(
|
|
"/login",
|
|
headers={"Content-Type": "application/json"},
|
|
json={"username": username},
|
|
)
|
|
|
|
assert response.status_code == 422
|
|
assert "access_token" not in response.json()
|