/tests/conftest.py

"""Shared pytest fixtures for Excalibur tests."""

import tempfile
from pathlib import Path

import pytest

from excalibur.core.backend import AgentBackend, AgentMessage
from excalibur.core.config import ExcaliburConfig
from excalibur.core.events import EventBus

# =============================================================================
# Pytest Markers
# =============================================================================


def pytest_configure(config: pytest.Config) -> None:
    """Configure custom pytest markers."""
    config.addinivalue_line("markers", "unit: Unit tests (fast, no external dependencies)")
    config.addinivalue_line("markers", "integration: Integration tests (may use mocks)")
    config.addinivalue_line("markers", "docker: Docker tests (requires Docker daemon)")
    config.addinivalue_line("markers", "slow: Slow tests (skip with -m 'not slow')")


# =============================================================================
# EventBus Fixtures
# =============================================================================


@pytest.fixture(autouse=True)
def reset_event_bus():
    """Reset EventBus singleton before and after each test."""
    EventBus.reset()
    yield
    EventBus.reset()


# =============================================================================
# Directory Fixtures
# =============================================================================


@pytest.fixture
def temp_sessions_dir():
    """Create a temporary directory for session storage."""
    with tempfile.TemporaryDirectory() as tmpdir:
        yield Path(tmpdir)


@pytest.fixture
def temp_working_dir():
    """Create a temporary working directory."""
    with tempfile.TemporaryDirectory() as tmpdir:
        yield Path(tmpdir)


# =============================================================================
# Configuration Fixtures
# =============================================================================


@pytest.fixture
def sample_config(temp_working_dir: Path) -> ExcaliburConfig:
    """Create a sample configuration for testing."""
    return ExcaliburConfig(
        target="test.example.com",
        working_directory=temp_working_dir,
    )


# =============================================================================
# Mock Backend
# =============================================================================


class MockBackend(AgentBackend):
    """Mock backend for testing agent controller."""

    def __init__(self) -> None:
        self._connected = False
        self._messages: list[AgentMessage] = []
        self._session_id = "mock-session-123"

    async def connect(self) -> None:
        """Simulate connection."""
        self._connected = True

    async def disconnect(self) -> None:
        """Simulate disconnection."""
        self._connected = False

    async def query(self, prompt: str) -> None:
        """Simulate sending a query."""
        pass

    async def receive_messages(self):
        """Yield preset messages."""
        for msg in self._messages:
            yield msg

    @property
    def session_id(self) -> str:
        """Get mock session ID."""
        return self._session_id

    @property
    def supports_resume(self) -> bool:
        """Mock does not support resume."""
        return False

    async def resume(self, session_id: str) -> bool:
        """Mock resume always fails."""
        return False

    def set_messages(self, messages: list[AgentMessage]) -> None:
        """Set messages to be returned by receive_messages."""
        self._messages = messages


@pytest.fixture
def mock_backend() -> MockBackend:
    """Create a mock backend for testing."""
    return MockBackend()


# =============================================================================
# Planner Fixtures
# =============================================================================


@pytest.fixture
def attack_tree():
    """Create a sample attack tree for testing."""
    from excalibur.planner.models import AttackNode, AttackTree, EvidenceLevel, NodeType

    root = AttackNode(
        node_type=NodeType.OBSERVATION,
        description="Initial recon of target",
        host="10.10.10.1",
        evidence_level=EvidenceLevel.VERIFIED,
        promise_score=0.5,
    )
    tree = AttackTree(root_id=root.id)
    tree.add_node(root)
    return tree


@pytest.fixture
def state_store():
    """Create an in-memory state store for testing."""
    from excalibur.memory.state_store import StateStore

    store = StateStore(db_path=":memory:")
    yield store
    store.close()