""" Pytest configuration and fixtures for backend tests. Provides comprehensive mocking and fixtures for all test scenarios. """ import os import sys from datetime import datetime, timezone from pathlib import Path from unittest.mock import MagicMock, patch import pytest from fastapi.testclient import TestClient from sqlalchemy import create_engine from sqlalchemy.orm import sessionmaker # Set test environment variables BEFORE any application imports os.environ.setdefault("ENVIRONMENT", "test") os.environ.setdefault("SECRET_KEY", "test-secret-key-min-32-chars-long-for-security") os.environ.setdefault("JWT_SECRET_KEY", "test-jwt-secret-key-min-32-chars") os.environ.setdefault("ENCRYPTION_KEY", "6SayK9nR4HO8REkIp3FhAaax8hwvLhsngvVd5OZa9Qc=") os.environ.setdefault("DATABASE_URL", "sqlite:///:memory:") os.environ.setdefault("CORS_ALLOWED_ORIGINS", "http://localhost:3000,http://localhost:5173") os.environ["TESTING"] = "True" # Add backend directory to Python path backend_dir = Path(__file__).parent.parent sys.path.insert(0, str(backend_dir)) # Import these after sys.path is updated from app.modules.auth.service import get_current_user # noqa: E402 from core.database import get_db # noqa: E402 # ===== Database Fixtures ===== @pytest.fixture(scope="session") def engine(): """Create test database engine with in-memory SQLite.""" from sqlalchemy.pool import StaticPool engine = create_engine( "sqlite:///:memory:", connect_args={"check_same_thread": False}, poolclass=StaticPool, ) # Import all models to ensure they are registered with Base # Importing from core.models ensures all models are loaded from core.models.base import Base Base.metadata.create_all(bind=engine) return engine @pytest.fixture(scope="session") def session_local(engine): """Create test session factory.""" return sessionmaker(autocommit=False, autoflush=False, bind=engine) @pytest.fixture def db_session(session_local): """Provide a clean database session for each test.""" session = session_local() try: yield session finally: session.close() @pytest.fixture def db_session_with_data(session_local): """Provide a database session with test data.""" from sqlalchemy import text from core.models import Case, FraudAlert, Project, Team, User session = session_local() # Clean up any existing data from previous tests to avoid IntegrityErrors session.execute(text("DELETE FROM fraud_alerts")) session.execute(text("DELETE FROM case_notes")) session.execute(text("DELETE FROM case_activities")) session.execute(text("DELETE FROM cases")) session.execute(text("DELETE FROM projects")) session.execute(text("DELETE FROM teams")) session.execute(text("DELETE FROM users")) session.commit() # Create test user test_user = User( id="test-user-001", username="testuser", email="test@example.com", full_name="Test User", role="ANALYST", is_active=True, mfa_enabled=False, password_hash="$pbkdf2_sha256$600000$test", preferences='{}', # Add preferences field created_at=datetime.now(timezone.utc), ) session.add(test_user) session.flush() # Create default team default_team = Team( id="default-team-001", name="Default Team", description="Default team for testing", ) session.add(default_team) session.flush() # Create default project default_project = Project( id="default", name="Default Project", description="Default project for testing", created_by="test-user-001", ) session.add(default_project) session.flush() # Create test case test_case = Case( id="test-case-001", project_id="default", title="Test Fraud Case", description="Test case for unit testing", status="OPEN", priority="HIGH", risk_score=75.0, assignee_id="test-user-001", team_id="default-team-001", created_at=datetime.now(timezone.utc), ) session.add(test_case) # Create test alert test_alert = FraudAlert( id="test-alert-001", case_id="test-case-001", alert_type="suspicious_transaction", severity="HIGH", title="Suspicious Activity Detected", is_acknowledged=False, description="Test alert for unit testing", created_at=datetime.now(timezone.utc), ) session.add(test_alert) session.commit() try: yield session finally: session.close() # ===== Authentication Fixtures ===== @pytest.fixture def mock_current_user(): """Mock authenticated user for tests.""" return { "id": "test-user-001", "username": "testuser", "email": "test@example.com", "role": "analyst", "mfa_verified": True, } @pytest.fixture def mock_admin_user(): """Mock admin user for tests.""" return { "id": "admin-user-001", "username": "admin", "email": "admin@example.com", "role": "admin", "mfa_verified": True, } @pytest.fixture def auth_headers(mock_current_user): """Generate authentication headers.""" from app.services.infrastructure.auth_service import auth_service token = auth_service.create_access_token(mock_current_user) return {"Authorization": f"Bearer {token}"} @pytest.fixture def admin_auth_headers(mock_admin_user): """Generate admin authentication headers.""" from app.services.infrastructure.auth_service import auth_service token = auth_service.create_access_token(mock_admin_user) return {"Authorization": f"Bearer {token}"} # ===== Service Mocks ===== @pytest.fixture def mock_auth_service(): """Mock auth service for tests.""" with patch("app.services.infrastructure.auth_service.auth_service") as mock: mock.authenticate_user.return_value = None mock.get_current_user.return_value = None mock.create_access_token.return_value = "test-jwt-token" yield mock @pytest.fixture def mock_db_service(): """Mock database service for tests.""" mock = MagicMock() mock.get_user_by_username.return_value = None mock.get_user_by_email.return_value = None mock.create_user.return_value = None mock.get_cases.return_value = [] mock.get_case_by_id.return_value = None return mock @pytest.fixture def mock_cache_service(): """Mock cache service for tests.""" mock = MagicMock() mock.get.return_value = None mock.set.return_value = True mock.delete.return_value = True mock.get_stats.return_value = { "l1_cache": {"entries": 0, "max_entries": 500}, "l2_cache": {"entries": 0, "max_entries": 1500}, "metrics": {"hits": 0, "misses": 0, "hit_rate": 0.0}, } return mock # ===== Application Fixtures ===== @pytest.fixture(scope="session") def app(): """Create test application instance.""" from app.factory import create_app app_instance = create_app() return app_instance @pytest.fixture(scope="session") def client(app): """Create test client with exceptions enabled for debugging.""" # Ensure performance_monitor is mocked even for simple client with patch("app.services.infrastructure.performance_monitor.performance_monitor.get_baselines") as mock_baselines: mock_baselines.return_value = { "monitoring_active": True, "metrics_collected": 100, } with patch( "app.services.infrastructure.performance_monitor.performance_monitor.get_current_metrics" ) as mock_metrics: mock_metrics.return_value = {"cpu_percent": 10.0, "memory_percent": 20.0} with patch( "app.services.infrastructure.performance_monitor.performance_monitor.check_thresholds" ) as mock_alerts: mock_alerts.return_value = [] with patch("app.services.infrastructure.circuit_breaker.get_all_circuit_breakers") as mock_cb: mock_cb.return_value = {} return TestClient(app, raise_server_exceptions=True) @pytest.fixture def client_with_db(app, db_session_with_data): """Create test client with database data and dependency overrides.""" from core.models import User # 1. Override database session def override_get_db(): try: yield db_session_with_data finally: pass # 2. Override get_current_user to return our test user def override_get_current_user(): return db_session_with_data.query(User).filter(User.id == "test-user-001").first() app.dependency_overrides[get_db] = override_get_db app.dependency_overrides[get_current_user] = override_get_current_user # 3. Patch db_service to use test session for health checks and internal service calls with ( patch("app.services.infrastructure.storage.database_service.db_service.get_db") as mock_get_db, patch("app.services.infrastructure.storage.database_service.db_service.health_check") as mock_health, patch("app.services.infrastructure.cache_service.cache_manager.get_stats") as mock_cache_stats, patch("app.services.infrastructure.performance_monitor.performance_monitor.get_baselines") as mock_baselines, patch( "app.services.infrastructure.performance_monitor.performance_monitor.get_current_metrics" ) as mock_metrics, patch("app.services.infrastructure.performance_monitor.performance_monitor.check_thresholds") as mock_alerts, patch("app.services.infrastructure.circuit_breaker.get_all_circuit_breakers") as mock_cb, ): mock_get_db.return_value = db_session_with_data # Mock health check to return healthy mock_health.return_value = { "status": "healthy", "checks": {"connectivity": {"status": "healthy"}}, "response_time_ms": 1.0, } # Mock cache service health check mock_cache_stats.return_value = { "l1_cache": {"entries": 0, "max_entries": 500}, "l2_cache": {"entries": 0, "max_entries": 1500}, "metrics": {"hits": 0, "misses": 0, "hit_rate": 0.0}, "l3_cache": {"available": True}, } # Mock performance monitor mock_baselines.return_value = { "monitoring_active": True, "metrics_collected": 100, } mock_metrics.return_value = {"cpu_percent": 5.0, "memory_percent": 10.0} mock_alerts.return_value = [] # Mock circuit breakers mock_cb.return_value = { "database_connection": {"state": "closed"}, "api_responsiveness": {"state": "closed"}, } client = TestClient(app, raise_server_exceptions=True) yield client app.dependency_overrides.clear() # ===== Helper Fixtures ===== @pytest.fixture def sample_case_data(): """Sample case data for create/update tests.""" return { "title": "New Test Case", "description": "Description for test case", "priority": "high", "tags": ["test", "automation"], } @pytest.fixture def sample_transaction_data(): """Sample transaction data for fraud analysis tests.""" return { "transaction_id": "txn-test-001", "amount": 15000.00, "currency": "USD", "merchant": "Test Merchant", "timestamp": datetime.now(timezone.utc).isoformat(), "account_id": "acc-001", } @pytest.fixture def sample_user_registration(): """Sample user registration data.""" return { "username": "newuser", "email": "newuser@example.com", "password": "SecureP@ss123!", "full_name": "New User", } @pytest.fixture def bulk_case_data(): """Fixture for bulk case data.""" return [ { "title": f"Bulk Case {i}", "description": f"Description for bulk case {i}", "priority": "Medium", "tags": ["bulk", "test"], } for i in range(5) ] @pytest.fixture def complex_case_with_notes(db_session_with_data): """Fixture for a case with multiple notes and attachments.""" from core.models import CaseNote case_id = "test-case-001" notes = [ CaseNote( id=f"note-{i}", case_id=case_id, content=f"Complex note content {i}", user_id="test-user-001", is_internal=(i % 2 == 0), created_at=datetime.now(timezone.utc), ) for i in range(3) ] for note in notes: db_session_with_data.add(note) db_session_with_data.commit() return case_id # ===== Environment Fixtures ===== @pytest.fixture(autouse=True) def reset_env(): """Reset environment between tests.""" yield @pytest.fixture def set_production_env(): """Set production environment variables.""" with patch.dict( os.environ, { "ENVIRONMENT": "production", "DEBUG": "false", }, ): yield # ===== Async Fixtures ===== @pytest.fixture async def async_client(): """Create async test client for async endpoints.""" from httpx import AsyncClient from app.factory import create_app app = create_app() async with AsyncClient(app=app, base_url="http://test") as client: yield client # ===== Export commonly used items ===== # Store reference to tracer provider for cleanup _tracer_provider = None @pytest.fixture(scope="session", autouse=True) def _setup_telemetry_cleanup(): """Setup OpenTelemetry span processor cleanup after tests.""" global _tracer_provider try: from opentelemetry.sdk.resources import Resource from opentelemetry.sdk.trace import TracerProvider from opentelemetry.sdk.trace.export import ConsoleSpanExporter, SimpleSpanProcessor _tracer_provider = TracerProvider(resource=Resource.create({})) _tracer_provider.add_span_processor(SimpleSpanProcessor(ConsoleSpanExporter())) except ImportError: pass yield # Cleanup after all tests complete if _tracer_provider is not None: try: _tracer_provider.shutdown() except Exception: pass @pytest.fixture(autouse=True) def disable_telemetry_for_tests(): """Disable OpenTelemetry tracing during tests to avoid span export errors.""" import os # Disable any OTEL exporters that might try to write to closed files os.environ["OTEL_TRACES_EXPORTER"] = "none" os.environ["OTEL_METRICS_EXPORTER"] = "none" os.environ["OTEL_LOGS_EXPORTER"] = "none" yield # Clean up after test os.environ.pop("OTEL_TRACES_EXPORTER", None) os.environ.pop("OTEL_METRICS_EXPORTER", None) os.environ.pop("OTEL_LOGS_EXPORTER", None) __all__ = [ "engine", "session_local", "db_session", "db_session_with_data", "mock_current_user", "mock_admin_user", "auth_headers", "admin_auth_headers", "mock_auth_service", "mock_db_service", "mock_cache_service", "app", "client", "client_with_db", "sample_case_data", "sample_transaction_data", "sample_user_registration", "async_client", ]