Felipe Cardoso
c589b565f0
- Introduced `pyproject.toml` to centralize backend tool configurations (e.g., Ruff, mypy, coverage, pytest). - Replaced Black, isort, and Flake8 with Ruff for linting, formatting, and import sorting. - Updated `requirements.txt` to include Ruff and remove replaced tools. - Added `Makefile` to streamline development workflows with commands for linting, formatting, type-checking, testing, and cleanup.
309 lines
11 KiB
Python
Executable File
309 lines
11 KiB
Python
Executable File
# tests/api/test_auth_endpoints.py
|
|
"""
|
|
Tests for authentication endpoints.
|
|
"""
|
|
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
from fastapi import status
|
|
from sqlalchemy import select
|
|
|
|
from app.models.user import User
|
|
|
|
|
|
# Disable rate limiting for tests
|
|
@pytest.fixture(autouse=True)
|
|
def disable_rate_limit():
|
|
"""Disable rate limiting for all tests in this module."""
|
|
with patch("app.api.routes.auth.limiter.enabled", False):
|
|
yield
|
|
|
|
|
|
class TestRegisterEndpoint:
|
|
"""Tests for POST /auth/register endpoint."""
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_register_success(self, client):
|
|
"""Test successful user registration."""
|
|
response = await client.post(
|
|
"/api/v1/auth/register",
|
|
json={
|
|
"email": "newuser@example.com",
|
|
"password": "SecurePassword123!",
|
|
"first_name": "New",
|
|
"last_name": "User",
|
|
},
|
|
)
|
|
|
|
assert response.status_code == status.HTTP_201_CREATED
|
|
data = response.json()
|
|
assert data["email"] == "newuser@example.com"
|
|
assert data["first_name"] == "New"
|
|
assert "password" not in data
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_register_duplicate_email(self, client, async_test_user):
|
|
"""Test registering with existing email.
|
|
|
|
Note: Returns 400 with generic message to prevent user enumeration.
|
|
"""
|
|
response = await client.post(
|
|
"/api/v1/auth/register",
|
|
json={
|
|
"email": async_test_user.email,
|
|
"password": "SecurePassword123!",
|
|
"first_name": "Duplicate",
|
|
"last_name": "User",
|
|
},
|
|
)
|
|
|
|
# Security: Returns 400 with generic message to prevent email enumeration
|
|
assert response.status_code == status.HTTP_400_BAD_REQUEST
|
|
data = response.json()
|
|
assert data["success"] is False
|
|
assert "registration failed" in data["errors"][0]["message"].lower()
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_register_weak_password(self, client):
|
|
"""Test registration with weak password."""
|
|
response = await client.post(
|
|
"/api/v1/auth/register",
|
|
json={
|
|
"email": "weakpass@example.com",
|
|
"password": "weak",
|
|
"first_name": "Weak",
|
|
"last_name": "Pass",
|
|
},
|
|
)
|
|
|
|
assert response.status_code == status.HTTP_422_UNPROCESSABLE_ENTITY
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_register_unexpected_error(self, client):
|
|
"""Test registration with unexpected error."""
|
|
with patch("app.services.auth_service.AuthService.create_user") as mock_create:
|
|
mock_create.side_effect = Exception("Unexpected error")
|
|
|
|
response = await client.post(
|
|
"/api/v1/auth/register",
|
|
json={
|
|
"email": "error@example.com",
|
|
"password": "SecurePassword123!",
|
|
"first_name": "Error",
|
|
"last_name": "User",
|
|
},
|
|
)
|
|
|
|
assert response.status_code == status.HTTP_500_INTERNAL_SERVER_ERROR
|
|
|
|
|
|
class TestLoginEndpoint:
|
|
"""Tests for POST /auth/login endpoint."""
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_login_success(self, client, async_test_user):
|
|
"""Test successful login."""
|
|
response = await client.post(
|
|
"/api/v1/auth/login",
|
|
json={"email": async_test_user.email, "password": "TestPassword123!"},
|
|
)
|
|
|
|
assert response.status_code == status.HTTP_200_OK
|
|
data = response.json()
|
|
assert "access_token" in data
|
|
assert "refresh_token" in data
|
|
assert data["token_type"] == "bearer"
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_login_wrong_password(self, client, async_test_user):
|
|
"""Test login with wrong password."""
|
|
response = await client.post(
|
|
"/api/v1/auth/login",
|
|
json={"email": async_test_user.email, "password": "WrongPassword123"},
|
|
)
|
|
|
|
assert response.status_code == status.HTTP_401_UNAUTHORIZED
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_login_nonexistent_user(self, client):
|
|
"""Test login with non-existent email."""
|
|
response = await client.post(
|
|
"/api/v1/auth/login",
|
|
json={"email": "nonexistent@example.com", "password": "Password123!"},
|
|
)
|
|
|
|
assert response.status_code == status.HTTP_401_UNAUTHORIZED
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_login_inactive_user(self, client, async_test_user, async_test_db):
|
|
"""Test login with inactive user."""
|
|
_test_engine, AsyncTestingSessionLocal = async_test_db
|
|
async with AsyncTestingSessionLocal() as session:
|
|
# Get the user in this session and make it inactive
|
|
result = await session.execute(
|
|
select(User).where(User.id == async_test_user.id)
|
|
)
|
|
user_in_session = result.scalar_one_or_none()
|
|
user_in_session.is_active = False
|
|
await session.commit()
|
|
|
|
response = await client.post(
|
|
"/api/v1/auth/login",
|
|
json={"email": async_test_user.email, "password": "TestPassword123!"},
|
|
)
|
|
|
|
assert response.status_code == status.HTTP_401_UNAUTHORIZED
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_login_unexpected_error(self, client, async_test_user):
|
|
"""Test login with unexpected error."""
|
|
with patch(
|
|
"app.services.auth_service.AuthService.authenticate_user"
|
|
) as mock_auth:
|
|
mock_auth.side_effect = Exception("Database error")
|
|
|
|
response = await client.post(
|
|
"/api/v1/auth/login",
|
|
json={"email": async_test_user.email, "password": "TestPassword123!"},
|
|
)
|
|
|
|
assert response.status_code == status.HTTP_500_INTERNAL_SERVER_ERROR
|
|
|
|
|
|
class TestOAuthLoginEndpoint:
|
|
"""Tests for POST /auth/login/oauth endpoint."""
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_oauth_login_success(self, client, async_test_user):
|
|
"""Test successful OAuth login."""
|
|
response = await client.post(
|
|
"/api/v1/auth/login/oauth",
|
|
data={"username": async_test_user.email, "password": "TestPassword123!"},
|
|
)
|
|
|
|
assert response.status_code == status.HTTP_200_OK
|
|
data = response.json()
|
|
assert "access_token" in data
|
|
assert "refresh_token" in data
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_oauth_login_wrong_credentials(self, client, async_test_user):
|
|
"""Test OAuth login with wrong credentials."""
|
|
response = await client.post(
|
|
"/api/v1/auth/login/oauth",
|
|
data={"username": async_test_user.email, "password": "WrongPassword"},
|
|
)
|
|
|
|
assert response.status_code == status.HTTP_401_UNAUTHORIZED
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_oauth_login_inactive_user(
|
|
self, client, async_test_user, async_test_db
|
|
):
|
|
"""Test OAuth login with inactive user."""
|
|
_test_engine, AsyncTestingSessionLocal = async_test_db
|
|
async with AsyncTestingSessionLocal() as session:
|
|
# Get the user in this session and make it inactive
|
|
result = await session.execute(
|
|
select(User).where(User.id == async_test_user.id)
|
|
)
|
|
user_in_session = result.scalar_one_or_none()
|
|
user_in_session.is_active = False
|
|
await session.commit()
|
|
|
|
response = await client.post(
|
|
"/api/v1/auth/login/oauth",
|
|
data={"username": async_test_user.email, "password": "TestPassword123!"},
|
|
)
|
|
|
|
assert response.status_code == status.HTTP_401_UNAUTHORIZED
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_oauth_login_unexpected_error(self, client, async_test_user):
|
|
"""Test OAuth login with unexpected error."""
|
|
with patch(
|
|
"app.services.auth_service.AuthService.authenticate_user"
|
|
) as mock_auth:
|
|
mock_auth.side_effect = Exception("Unexpected error")
|
|
|
|
response = await client.post(
|
|
"/api/v1/auth/login/oauth",
|
|
data={
|
|
"username": async_test_user.email,
|
|
"password": "TestPassword123!",
|
|
},
|
|
)
|
|
|
|
assert response.status_code == status.HTTP_500_INTERNAL_SERVER_ERROR
|
|
|
|
|
|
class TestRefreshTokenEndpoint:
|
|
"""Tests for POST /auth/refresh endpoint."""
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_refresh_token_success(self, client, async_test_user):
|
|
"""Test successful token refresh."""
|
|
# First, login to get a refresh token
|
|
login_response = await client.post(
|
|
"/api/v1/auth/login",
|
|
json={"email": async_test_user.email, "password": "TestPassword123!"},
|
|
)
|
|
refresh_token = login_response.json()["refresh_token"]
|
|
|
|
# Now refresh the token
|
|
response = await client.post(
|
|
"/api/v1/auth/refresh", json={"refresh_token": refresh_token}
|
|
)
|
|
|
|
assert response.status_code == status.HTTP_200_OK
|
|
data = response.json()
|
|
assert "access_token" in data
|
|
assert "refresh_token" in data
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_refresh_token_expired(self, client):
|
|
"""Test refresh with expired token."""
|
|
from app.core.auth import TokenExpiredError
|
|
|
|
with patch(
|
|
"app.services.auth_service.AuthService.refresh_tokens"
|
|
) as mock_refresh:
|
|
mock_refresh.side_effect = TokenExpiredError("Token expired")
|
|
|
|
response = await client.post(
|
|
"/api/v1/auth/refresh", json={"refresh_token": "some_token"}
|
|
)
|
|
|
|
assert response.status_code == status.HTTP_401_UNAUTHORIZED
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_refresh_token_invalid(self, client):
|
|
"""Test refresh with invalid token."""
|
|
response = await client.post(
|
|
"/api/v1/auth/refresh", json={"refresh_token": "invalid_token"}
|
|
)
|
|
|
|
assert response.status_code == status.HTTP_401_UNAUTHORIZED
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_refresh_token_unexpected_error(self, client, async_test_user):
|
|
"""Test refresh with unexpected error."""
|
|
# Get a valid refresh token first
|
|
login_response = await client.post(
|
|
"/api/v1/auth/login",
|
|
json={"email": async_test_user.email, "password": "TestPassword123!"},
|
|
)
|
|
refresh_token = login_response.json()["refresh_token"]
|
|
|
|
with patch(
|
|
"app.services.auth_service.AuthService.refresh_tokens"
|
|
) as mock_refresh:
|
|
mock_refresh.side_effect = Exception("Unexpected error")
|
|
|
|
response = await client.post(
|
|
"/api/v1/auth/refresh", json={"refresh_token": refresh_token}
|
|
)
|
|
|
|
assert response.status_code == status.HTTP_500_INTERNAL_SERVER_ERROR
|