learningcircuit--local-deep-research/tests/storage/conftest.py

"""Fixtures for storage module tests."""

import pytest
from unittest.mock import MagicMock


@pytest.fixture
def mock_session():
    """Mock SQLAlchemy session."""
    session = MagicMock()
    return session


@pytest.fixture
def mock_research_history():
    """Mock ResearchHistory model instance."""
    record = MagicMock()
    record.id = "test-uuid-123"
    record.report_content = "# Test Report\n\nThis is a test report."
    record.research_meta = {"key": "value"}
    record.query = "test query"
    record.mode = "detailed"
    record.created_at = "2024-01-01T10:00:00"
    record.completed_at = "2024-01-01T10:30:00"
    record.duration_seconds = 1800
    return record


@pytest.fixture
def sample_report_content():
    """Sample report content for testing."""
    return """# Research Report

## Summary
This is a test research report.

## Findings
1. Finding one
2. Finding two

## Conclusion
Test conclusion.
"""


@pytest.fixture
def sample_metadata():
    """Sample metadata for testing."""
    return {
        "query": "test query",
        "mode": "detailed",
        "sources_count": 5,
        "timestamp": "2024-01-01T10:00:00",
    }