34 lines
917 B
Python
34 lines
917 B
Python
|
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
|