"""Tests for admin (cross-user) deletion of scheduled-task runs.

Regular users may only delete runs they own; admins may delete any run. This
covers the service/store layer that the admin-aware ``DELETE /runs/{id}``
route relies on.
"""

from __future__ import annotations

from datetime import UTC, datetime

import pytest

from deerflow.persistence.scheduled_tasks import MemoryScheduledTaskStore
from deerflow.runtime.scheduler.service import ScheduledTaskService


@pytest.fixture
def store() -> MemoryScheduledTaskStore:
    return MemoryScheduledTaskStore()


def _build_service(store: MemoryScheduledTaskStore) -> ScheduledTaskService:
    service = ScheduledTaskService(app=None, store=store, poll_interval_seconds=60.0)
    service._element_client = None  # noqa: SLF001
    return service


async def _seed_run(store: MemoryScheduledTaskStore, *, run_id: str, task_id: str, user_id: str) -> None:
    await store.create_run_once(
        {
            "id": run_id,
            "task_id": task_id,
            "user_id": user_id,
            "scheduler_thread_id": f"thread-{user_id}",
            "scheduled_for": datetime.now(UTC),
            "status": "succeeded",
            "started_at": datetime.now(UTC),
        }
    )


@pytest.mark.asyncio
async def test_admin_delete_run_removes_any_owner_run(store):
    await _seed_run(store, run_id="r1", task_id="t1", user_id="alice")
    service = _build_service(store)

    assert await service.admin_delete_run("r1") is True
    assert await store.get_run_any("r1") is None


@pytest.mark.asyncio
async def test_admin_delete_run_missing_returns_false(store):
    service = _build_service(store)
    assert await service.admin_delete_run("nope") is False


@pytest.mark.asyncio
async def test_owner_delete_run_still_scoped_to_owner(store):
    """A non-owner, non-subscriber deleting another user's run gets False."""
    await _seed_run(store, run_id="r1", task_id="t1", user_id="alice")
    service = _build_service(store)

    assert await service.delete_run("bob", "r1") is False
    assert await store.get_run_any("r1") is not None  # untouched

    # The owner can still delete their own run.
    assert await service.delete_run("alice", "r1") is True
    assert await store.get_run_any("r1") is None


@pytest.mark.asyncio
async def test_store_delete_run_any_independent_of_owner(store):
    await _seed_run(store, run_id="r1", task_id="t1", user_id="alice")

    assert await store.delete_run_any("r1") is True
    assert await store.delete_run_any("r1") is False