Write most backend tests
This commit is contained in:
139
tests/test_authentication.py
Normal file
139
tests/test_authentication.py
Normal file
@@ -0,0 +1,139 @@
|
||||
import sys
|
||||
|
||||
import pytest
|
||||
from fastapi.testclient import TestClient
|
||||
|
||||
sys.path.append("..")
|
||||
|
||||
from src.main import app, get_db
|
||||
from tests.config.database import clear_db, override_get_db
|
||||
|
||||
app.dependency_overrides[get_db] = override_get_db
|
||||
|
||||
client = TestClient(app)
|
||||
|
||||
username1 = "user1"
|
||||
username2 = "user2"
|
||||
password = "password"
|
||||
avatar = "lion"
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_register():
|
||||
"""LEAVE THIS TEST AT THE TOP OF THE FILE!"""
|
||||
"""Test the register endpoint"""
|
||||
clear_db()
|
||||
|
||||
response = client.post(
|
||||
"/register",
|
||||
headers={"Content-Type": "application/json"},
|
||||
json={"username": username1, "password": password, "avatar": avatar},
|
||||
)
|
||||
|
||||
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"""
|
||||
response = client.post(
|
||||
"/register",
|
||||
headers={"Content-Type": "application/json"},
|
||||
json={"username": username1, "password": password, "avatar": avatar},
|
||||
)
|
||||
|
||||
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"""
|
||||
response = client.post(
|
||||
"/register",
|
||||
headers={"Content-Type": "application/json"},
|
||||
json={"password": password, "avatar": avatar},
|
||||
)
|
||||
|
||||
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"""
|
||||
response = client.post(
|
||||
"/register",
|
||||
headers={"Content-Type": "application/json"},
|
||||
json={"username": username2, "avatar": avatar},
|
||||
)
|
||||
|
||||
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"""
|
||||
response = client.post(
|
||||
"/register",
|
||||
headers={"Content-Type": "application/json"},
|
||||
json={"username": username2, "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"""
|
||||
response = client.post(
|
||||
"/login",
|
||||
headers={"Content-Type": "application/json"},
|
||||
json={"username": username1, "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():
|
||||
wrong_password = password + "extra characters"
|
||||
response = client.post(
|
||||
"/login",
|
||||
headers={"Content-Type": "application/json"},
|
||||
json={"username": username1, "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"""
|
||||
response = client.post(
|
||||
"/login",
|
||||
headers={"Content-Type": "application/json"},
|
||||
json={"username": username1},
|
||||
)
|
||||
|
||||
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"""
|
||||
response = client.post(
|
||||
"/login",
|
||||
headers={"Content-Type": "application/json"},
|
||||
json={"username": username1},
|
||||
)
|
||||
|
||||
assert response.status_code == 422
|
||||
assert "access_token" not in response.json()
|
||||
Reference in New Issue
Block a user