|
|
"""Pytest configuration and fixtures.""" |
|
|
|
|
|
import pytest |
|
|
from unittest.mock import AsyncMock, MagicMock |
|
|
|
|
|
|
|
|
@pytest.fixture |
|
|
def sample_queries(): |
|
|
"""Sample queries for testing.""" |
|
|
return { |
|
|
"simple": [ |
|
|
"Hello", |
|
|
"What is 2 + 2?", |
|
|
"What is the capital of France?", |
|
|
"Hi there", |
|
|
"Thanks!", |
|
|
], |
|
|
"medium": [ |
|
|
"Explain the difference between TCP and UDP.", |
|
|
"What are the main features of Python 3.11?", |
|
|
"How does garbage collection work in Java?", |
|
|
"Describe the MVC architecture pattern.", |
|
|
], |
|
|
"complex": [ |
|
|
"Write a Python function that implements a binary search tree.", |
|
|
"Create a REST API with authentication using FastAPI.", |
|
|
"Implement a neural network from scratch in Python.", |
|
|
"Design a distributed caching system with Redis.", |
|
|
"Write a compiler for a simple programming language.", |
|
|
], |
|
|
} |
|
|
|
|
|
|
|
|
@pytest.fixture |
|
|
def mock_openai_response(): |
|
|
"""Mock OpenAI API response.""" |
|
|
return { |
|
|
"id": "chatcmpl-123", |
|
|
"object": "chat.completion", |
|
|
"created": 1234567890, |
|
|
"model": "gpt-4o", |
|
|
"choices": [ |
|
|
{ |
|
|
"index": 0, |
|
|
"message": { |
|
|
"role": "assistant", |
|
|
"content": "This is a test response.", |
|
|
}, |
|
|
"finish_reason": "stop", |
|
|
} |
|
|
], |
|
|
"usage": { |
|
|
"prompt_tokens": 10, |
|
|
"completion_tokens": 20, |
|
|
"total_tokens": 30, |
|
|
}, |
|
|
} |
|
|
|
|
|
|
|
|
@pytest.fixture |
|
|
def mock_redis_client(): |
|
|
"""Mock Redis client.""" |
|
|
client = AsyncMock() |
|
|
client.get.return_value = None |
|
|
client.setex.return_value = True |
|
|
client.delete.return_value = 1 |
|
|
client.ping.return_value = True |
|
|
return client |
|
|
|
|
|
|
|
|
@pytest.fixture |
|
|
def mock_qdrant_client(): |
|
|
"""Mock Qdrant client.""" |
|
|
client = MagicMock() |
|
|
client.search.return_value = [] |
|
|
client.upsert.return_value = None |
|
|
return client |
|
|
|
