AIPass/src/aipass/memory/tests/conftest.py

# ===================AIPASS====================
# META DATA HEADER
# Name: tests/conftest.py
# Date: 2026-03-24
# Version: 1.0.0
# Category: memory/tests
# =============================================

"""Shared pytest fixtures for memory tests."""
import json
import pytest
import shutil
import tempfile
from pathlib import Path
from typing import Generator
from unittest.mock import MagicMock


@pytest.fixture(autouse=True)
def _mock_infrastructure(monkeypatch):
    """Mock heavy infrastructure imports to avoid live dependencies."""
    import sys

    # Mock prax logger
    mock_logger = MagicMock()
    prax_mod = MagicMock()
    prax_mod.logger = mock_logger
    prax_modules_mod = MagicMock()
    prax_modules_mod.logger = MagicMock()
    prax_modules_mod.logger.get_system_logger = MagicMock(return_value=mock_logger)
    monkeypatch.setitem(sys.modules, "aipass.prax", prax_mod)
    monkeypatch.setitem(sys.modules, "aipass.prax.apps", MagicMock())
    monkeypatch.setitem(sys.modules, "aipass.prax.apps.modules", prax_modules_mod)
    monkeypatch.setitem(sys.modules, "aipass.prax.apps.modules.logger", prax_modules_mod.logger)

    # Mock json handler
    mock_json_handler = MagicMock()
    mock_json_handler.log_operation = MagicMock(return_value=True)
    json_pkg = MagicMock()
    json_pkg.json_handler = mock_json_handler
    monkeypatch.setitem(sys.modules, "aipass.memory.apps.handlers.json", json_pkg)
    monkeypatch.setitem(sys.modules, "aipass.memory.apps.handlers.json.json_handler", mock_json_handler)

    # Mock trigger
    mock_trigger = MagicMock()
    mock_trigger.fire = MagicMock()
    trigger_mod = MagicMock()
    trigger_mod.Trigger = mock_trigger
    monkeypatch.setitem(sys.modules, "aipass.trigger", MagicMock())
    monkeypatch.setitem(sys.modules, "aipass.trigger.apps", MagicMock())
    monkeypatch.setitem(sys.modules, "aipass.trigger.apps.modules", MagicMock())
    monkeypatch.setitem(sys.modules, "aipass.trigger.apps.modules.core", trigger_mod)


@pytest.fixture
def temp_test_dir() -> Generator[Path, None, None]:
    """Creates temporary directory for testing, cleans up after."""
    test_dir = Path(tempfile.mkdtemp())
    yield test_dir
    if test_dir.exists():
        shutil.rmtree(test_dir)


@pytest.fixture
def sample_test_data() -> dict:
    """Provides reusable sample data for general test assertions."""
    return {
        "created": "2026-01-01",
        "last_updated": "2026-01-15",
        "entries": [
            {"id": 1, "name": "alpha", "status": "active"},
            {"id": 2, "name": "beta", "status": "pending"},
        ],
        "metadata": {
            "source": "test_fixture",
            "version": "1.0.0",
        },
    }


@pytest.fixture
def sample_memory_data() -> dict:
    """Provides sample memory file data (v2 schema)."""
    return {
        "document_metadata": {
            "document_type": "session_history",
            "document_name": "TEST.LOCAL",
            "version": "2.0.0",
            "schema_version": "2.0.0",
            "created": "2026-01-01",
            "last_updated": "2026-01-01",
            "managed_by": "TEST",
            "tags": ["test"],
            "limits": {"max_sessions": 20, "max_key_learnings": 25},
            "status": {"health": "healthy", "current_lines": 50}
        },
        "key_learnings": {"test_learning": "This is a test."},
        "sessions": [
            {"session_number": 1, "date": "2026-01-01", "summary": "Test session", "status": "completed"}
        ]
    }


@pytest.fixture
def sample_registry_data() -> dict:
    """Provides sample AIPASS_REGISTRY.json data."""
    return {
        "branches": {
            "test_branch": {
                "name": "TEST_BRANCH",
                "path": "src/aipass/test_branch",
                "module": "aipass.test_branch",
                "email": "@test_branch",
                "status": "active"
            },
            "memory": {
                "name": "MEMORY",
                "path": "src/aipass/memory",
                "module": "aipass.memory",
                "email": "@memory",
                "status": "active"
            }
        }
    }


@pytest.fixture
def temp_branch(tmp_path, sample_memory_data):
    """Create a minimal branch structure with .trinity/ files."""
    branch_dir = tmp_path / "src" / "aipass" / "test_branch"
    trinity = branch_dir / ".trinity"
    trinity.mkdir(parents=True)
    (trinity / "local.json").write_text(json.dumps(sample_memory_data, indent=2), encoding="utf-8")
    (trinity / "passport.json").write_text(json.dumps({
        "branch_info": {"branch_name": "test_branch", "path": "src/aipass/test_branch"},
        "identity": {"role": "test", "purpose": "testing"},
        "citizenship": {"registered": True}
    }, indent=2), encoding="utf-8")
    (trinity / "observations.json").write_text(json.dumps({
        "document_metadata": {"document_type": "collaboration_patterns"},
        "observations": []
    }, indent=2), encoding="utf-8")
    return branch_dir


@pytest.fixture
def temp_registry(tmp_path, sample_registry_data):
    """Create a temporary AIPASS_REGISTRY.json."""
    registry_path = tmp_path / "AIPASS_REGISTRY.json"
    registry_path.write_text(json.dumps(sample_registry_data, indent=2), encoding="utf-8")
    return registry_path