196 lines
4.9 KiB
Python
196 lines
4.9 KiB
Python
"""Pytest fixtures and configuration."""
|
|
|
|
import os
|
|
from pathlib import Path
|
|
|
|
# Load test environment variables BEFORE importing app
|
|
test_env_path = Path(__file__).parent.parent / ".env.test"
|
|
if test_env_path.exists():
|
|
from dotenv import load_dotenv
|
|
load_dotenv(test_env_path)
|
|
|
|
import pytest
|
|
from unittest.mock import AsyncMock, MagicMock
|
|
from fastapi.testclient import TestClient
|
|
from datetime import datetime, timedelta
|
|
|
|
from app.main import app
|
|
from app.dependencies import get_db_client, get_current_user
|
|
from app.interfaces.db_api_client import DBApiClient
|
|
from app.models.auth import UserResponse
|
|
from app.models.settings import UserSettings, EnvironmentSettings
|
|
from app.utils.security import create_access_token
|
|
|
|
|
|
# ============================================
|
|
# Mock DB Client
|
|
# ============================================
|
|
|
|
@pytest.fixture
|
|
def mock_db_client():
|
|
"""Mock DB API client."""
|
|
client = AsyncMock(spec=DBApiClient)
|
|
return client
|
|
|
|
|
|
# ============================================
|
|
# Test User
|
|
# ============================================
|
|
|
|
@pytest.fixture
|
|
def test_user():
|
|
"""Test user data."""
|
|
return {
|
|
"user_id": "test-user-123",
|
|
"login": "12345678"
|
|
}
|
|
|
|
|
|
@pytest.fixture
|
|
def test_user_response(test_user):
|
|
"""Test user response from DB API."""
|
|
return UserResponse(
|
|
user_id=test_user["user_id"],
|
|
login=test_user["login"],
|
|
last_login_at=datetime.utcnow().isoformat() + "Z",
|
|
created_at=datetime.utcnow().isoformat() + "Z"
|
|
)
|
|
|
|
|
|
# ============================================
|
|
# JWT Token
|
|
# ============================================
|
|
|
|
@pytest.fixture
|
|
def test_token(test_user):
|
|
"""Generate test JWT token."""
|
|
return create_access_token(test_user)
|
|
|
|
|
|
@pytest.fixture
|
|
def expired_token(test_user):
|
|
"""Generate expired JWT token."""
|
|
return create_access_token(
|
|
test_user,
|
|
expires_delta=timedelta(seconds=-10) # Expired 10 seconds ago
|
|
)
|
|
|
|
|
|
# ============================================
|
|
# Test Settings
|
|
# ============================================
|
|
|
|
@pytest.fixture
|
|
def test_settings():
|
|
"""Test user settings."""
|
|
env_settings = EnvironmentSettings(
|
|
apiMode="bench",
|
|
bearerToken="test-bearer-token",
|
|
systemPlatform="test-platform",
|
|
systemPlatformUser="test-user",
|
|
platformUserId="platform-user-123",
|
|
platformId="platform-123",
|
|
withClassify=False,
|
|
resetSessionMode=True
|
|
)
|
|
|
|
return UserSettings(
|
|
user_id="test-user-123",
|
|
settings={
|
|
"ift": env_settings,
|
|
"psi": env_settings.model_copy(),
|
|
"prod": env_settings.model_copy()
|
|
},
|
|
updated_at=datetime.utcnow().isoformat() + "Z"
|
|
)
|
|
|
|
|
|
# ============================================
|
|
# FastAPI Test Client
|
|
# ============================================
|
|
|
|
@pytest.fixture
|
|
def client(mock_db_client, test_user):
|
|
"""FastAPI test client with mocked dependencies."""
|
|
|
|
# Override get_current_user to return test user
|
|
async def mock_get_current_user():
|
|
return test_user
|
|
|
|
# Override get_db_client to return mock
|
|
def mock_get_db_client():
|
|
return mock_db_client
|
|
|
|
app.dependency_overrides[get_current_user] = mock_get_current_user
|
|
app.dependency_overrides[get_db_client] = mock_get_db_client
|
|
|
|
with TestClient(app) as test_client:
|
|
yield test_client
|
|
|
|
# Clear overrides after test
|
|
app.dependency_overrides.clear()
|
|
|
|
|
|
@pytest.fixture
|
|
def unauthenticated_client():
|
|
"""FastAPI test client without authentication."""
|
|
with TestClient(app) as test_client:
|
|
yield test_client
|
|
|
|
|
|
# ============================================
|
|
# Mock RAG Responses
|
|
# ============================================
|
|
|
|
@pytest.fixture
|
|
def mock_bench_response():
|
|
"""Mock RAG backend bench response."""
|
|
return {
|
|
"answers": [
|
|
{
|
|
"question_id": 1,
|
|
"answer": "Test answer 1",
|
|
"confidence": 0.95,
|
|
"docs": []
|
|
},
|
|
{
|
|
"question_id": 2,
|
|
"answer": "Test answer 2",
|
|
"confidence": 0.87,
|
|
"docs": []
|
|
}
|
|
]
|
|
}
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_backend_response():
|
|
"""Mock RAG backend single response."""
|
|
return {
|
|
"answer": "Test answer",
|
|
"confidence": 0.92,
|
|
"docs": []
|
|
}
|
|
|
|
|
|
# ============================================
|
|
# Mock httpx Client
|
|
# ============================================
|
|
|
|
@pytest.fixture
|
|
def mock_httpx_client():
|
|
"""Mock httpx.AsyncClient."""
|
|
client = AsyncMock()
|
|
|
|
# Mock response
|
|
mock_response = MagicMock()
|
|
mock_response.status_code = 200
|
|
mock_response.json.return_value = {"test": "response"}
|
|
mock_response.raise_for_status = MagicMock()
|
|
|
|
client.post.return_value = mock_response
|
|
client.get.return_value = mock_response
|
|
client.aclose = AsyncMock()
|
|
|
|
return client
|