"""
Pytest configuration and fixtures
"""

import pytest
from fastapi.testclient import TestClient
from main import app

@pytest.fixture
def client():
    """FastAPI test client"""
    return TestClient(app)

@pytest.fixture
def mock_models_response():
    """Mock response from Ollama API"""
    return {
        "models": [
            {
                "name": "llama2",
                "digest": "91ab89b1b9117e34fb2ff4a5bff07b2e1fa1f1d2d3e4f5a6b7c8d9e0f1a2b3c",
                "size": 3825922048,
                "modified_at": "2024-01-15T10:30:00.000Z"
            },
            {
                "name": "mistral",
                "digest": "a1b2c3d4e5f6a7b8c9d0e1f2a3b4c5d6e7f8a9b0c1d2e3f4a5b6c7d8e9f0a1",
                "size": 4096000000,
                "modified_at": "2024-01-14T15:45:00.000Z"
            }
        ]
    }