976fd1d4ad
- Introduced new unit tests for session CRUD operations, including `update_refresh_token`, `cleanup_expired`, and multi-user session handling. - Added comprehensive tests for `CRUDBase` methods, covering edge cases, error handling, and UUID validation. - Reduced default test session creation from 5 to 2 for performance optimization. - Enhanced pagination, filtering, and sorting validations in `get_multi_with_total`. - Improved error handling with descriptive assertions for database exceptions. - Introduced tests for eager-loaded relationships in user sessions for comprehensive coverage.
85 lines
3.8 KiB
Python
85 lines
3.8 KiB
Python
# tests/test_init_db.py
|
|
"""
|
|
Tests for database initialization script.
|
|
"""
|
|
import pytest
|
|
import pytest_asyncio
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
from app.init_db import init_db
|
|
from app.core.config import settings
|
|
|
|
|
|
class TestInitDb:
|
|
"""Tests for init_db functionality."""
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_init_db_creates_superuser_when_not_exists(self, async_test_db):
|
|
"""Test that init_db creates a superuser when one doesn't exist."""
|
|
test_engine, SessionLocal = async_test_db
|
|
|
|
# Mock the SessionLocal to use our test database
|
|
with patch('app.init_db.SessionLocal', SessionLocal):
|
|
# Mock settings to provide test credentials
|
|
with patch.object(settings, 'FIRST_SUPERUSER_EMAIL', 'test_admin@example.com'):
|
|
with patch.object(settings, 'FIRST_SUPERUSER_PASSWORD', 'TestAdmin123!'):
|
|
# Run init_db
|
|
user = await init_db()
|
|
|
|
# Verify superuser was created
|
|
assert user is not None
|
|
assert user.email == 'test_admin@example.com'
|
|
assert user.is_superuser is True
|
|
assert user.first_name == 'Admin'
|
|
assert user.last_name == 'User'
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_init_db_returns_existing_superuser(self, async_test_db, async_test_user):
|
|
"""Test that init_db returns existing superuser instead of creating duplicate."""
|
|
test_engine, SessionLocal = async_test_db
|
|
|
|
# Mock the SessionLocal to use our test database
|
|
with patch('app.init_db.SessionLocal', SessionLocal):
|
|
# Mock settings to match async_test_user's email
|
|
with patch.object(settings, 'FIRST_SUPERUSER_EMAIL', 'testuser@example.com'):
|
|
with patch.object(settings, 'FIRST_SUPERUSER_PASSWORD', 'TestPassword123!'):
|
|
# Run init_db
|
|
user = await init_db()
|
|
|
|
# Verify it returns the existing user
|
|
assert user is not None
|
|
assert user.id == async_test_user.id
|
|
assert user.email == 'testuser@example.com'
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_init_db_uses_default_credentials(self, async_test_db):
|
|
"""Test that init_db uses default credentials when env vars not set."""
|
|
test_engine, SessionLocal = async_test_db
|
|
|
|
# Mock the SessionLocal to use our test database
|
|
with patch('app.init_db.SessionLocal', SessionLocal):
|
|
# Mock settings to have None values (not configured)
|
|
with patch.object(settings, 'FIRST_SUPERUSER_EMAIL', None):
|
|
with patch.object(settings, 'FIRST_SUPERUSER_PASSWORD', None):
|
|
# Run init_db
|
|
user = await init_db()
|
|
|
|
# Verify superuser was created with defaults
|
|
assert user is not None
|
|
assert user.email == 'admin@example.com'
|
|
assert user.is_superuser is True
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_init_db_handles_database_errors(self, async_test_db):
|
|
"""Test that init_db handles database errors gracefully."""
|
|
test_engine, SessionLocal = async_test_db
|
|
|
|
# Mock user_crud.get_by_email to raise an exception
|
|
with patch('app.init_db.user_crud.get_by_email', side_effect=Exception("Database error")):
|
|
with patch('app.init_db.SessionLocal', SessionLocal):
|
|
with patch.object(settings, 'FIRST_SUPERUSER_EMAIL', 'test@example.com'):
|
|
with patch.object(settings, 'FIRST_SUPERUSER_PASSWORD', 'TestPassword123!'):
|
|
# Run init_db and expect it to raise
|
|
with pytest.raises(Exception, match="Database error"):
|
|
await init_db()
|