Files
nousresearch--hermes-agent/tests/gateway/test_compression_in_flight_check.py
wehub-resource-sync b4fbd6fe9f
Deploy Site / deploy-vercel (push) Has been skipped
Deploy Site / deploy-docs (push) Has been skipped
Build Skills Index / build-index (push) Has been skipped
CI / Deny unrelated histories (push) Has been skipped
CI / Detect affected areas (push) Successful in 27m35s
CI / OSV scan (push) Failing after 4s
CI / Build&Test Docker image (push) Successful in 9s
CI / Supply-chain scan (push) Has been skipped
CI / Lint Docker scripts (push) Failing after 5m13s
CI / Check contributors (push) Failing after 12m8s
CI / Docs Site (push) Failing after 12m8s
CI / TypeScript (push) Failing after 12m8s
CI / Python lints (push) Failing after 12m9s
CI / Python tests (push) Failing after 12m9s
CI / Check uv.lock (push) Failing after 23m22s
CI / CI timing report (push) Has been cancelled
Build Skills Index / trigger-deploy (push) Has been cancelled
CI / All required checks pass (push) Has been cancelled
chore: import upstream snapshot with attribution
2026-07-13 11:56:03 +08:00

74 lines
2.5 KiB
Python

"""#5 regression: _session_has_compression_in_flight must offload both blocking sources to thread pool."""
import inspect
import threading
from unittest.mock import MagicMock
import pytest
def _make_runner(holder_value=None, record_thread=False, thread_sink=None):
from gateway.run import GatewayRunner
runner = GatewayRunner.__new__(GatewayRunner)
store = MagicMock()
store._lock = threading.Lock()
store._loaded = True
store._entries = {"k": MagicMock(session_id="sess-123")}
store._ensure_loaded_locked = lambda: None
runner.session_store = store
raw_db = MagicMock()
if record_thread and thread_sink is not None:
def _holder(sid):
thread_sink["thread"] = threading.get_ident()
return holder_value
raw_db.get_compression_lock_holder = _holder
else:
raw_db.get_compression_lock_holder = MagicMock(return_value=holder_value)
session_db = MagicMock()
session_db._db = raw_db
runner._session_db = session_db
return runner
def test_method_is_coroutine():
from gateway.run import GatewayRunner
assert inspect.iscoroutinefunction(
GatewayRunner._session_has_compression_in_flight
), "#5: method must be async, blocking calls offloaded"
@pytest.mark.asyncio
async def test_returns_true_when_lock_held():
runner = _make_runner(holder_value="agent-1")
assert await runner._session_has_compression_in_flight("k") is True
@pytest.mark.asyncio
async def test_returns_false_when_no_lock():
runner = _make_runner(holder_value=None)
assert await runner._session_has_compression_in_flight("k") is False
@pytest.mark.asyncio
async def test_returns_false_when_no_session_store():
from gateway.run import GatewayRunner
runner = GatewayRunner.__new__(GatewayRunner)
runner.session_store = None
runner._session_db = MagicMock()
assert await runner._session_has_compression_in_flight("k") is False
@pytest.mark.asyncio
async def test_db_call_runs_off_event_loop():
"""Regression core: get_compression_lock_holder MUST execute in non-event-loop thread."""
sink = {}
runner = _make_runner(holder_value="agent-1", record_thread=True, thread_sink=sink)
loop_thread = threading.get_ident()
await runner._session_has_compression_in_flight("k")
assert "thread" in sink, "underlying db.get_compression_lock_holder was not called"
assert sink["thread"] != loop_thread, (
"DB call still on event loop thread — #5 NOT fixed (to_thread not applied)"
)