from __future__ import annotations

import pytest

from v1.agent import router as agent_router


@pytest.mark.asyncio
async def test_allow_run_request_fails_closed_when_redis_unavailable(
    monkeypatch: pytest.MonkeyPatch,
) -> None:
    async def _raise_redis_error():
        raise RuntimeError("redis unavailable")

    monkeypatch.setattr(agent_router, "get_or_init_redis_client", _raise_redis_error)

    allowed = await agent_router._allow_run_request(user_id="user-1")

    assert allowed is False


@pytest.mark.asyncio
async def test_acquire_sse_slot_fails_closed_when_redis_unavailable(
    monkeypatch: pytest.MonkeyPatch,
) -> None:
    async def _raise_redis_error():
        raise RuntimeError("redis unavailable")

    monkeypatch.setattr(agent_router, "get_or_init_redis_client", _raise_redis_error)

    allowed = await agent_router._acquire_sse_slot(user_id="user-1")

    assert allowed is False