Reduce backend test warning noise

Author: terrywangcode

server/alembic.ini

@@ -1,6 +1,7 @@
 [alembic]
 script_location = hive/db/migrations
 prepend_sys_path = .
+path_separator = os
 sqlalchemy.url = postgresql+asyncpg://hive:hive@localhost:5432/hive
 
 [loggers]

server/conftest.py

@@ -1,9 +1,31 @@
 """Root conftest: add repo root to sys.path so packages.shared is importable."""
 
 import sys
+import warnings
 from pathlib import Path
 
 # Repo root is one level up from server/
 _REPO_ROOT = str(Path(__file__).parent.parent)
 if _REPO_ROOT not in sys.path:
     sys.path.insert(0, _REPO_ROOT)
+
+warnings.filterwarnings(
+    "ignore",
+    message=r".*datetime\.datetime\.utcfromtimestamp\(\) is deprecated.*",
+    category=DeprecationWarning,
+)
+warnings.filterwarnings(
+    "ignore",
+    message=r".*There is no current event loop.*",
+    category=DeprecationWarning,
+)
+warnings.filterwarnings(
+    "ignore",
+    message=r".*websockets\.InvalidStatusCode is deprecated.*",
+    category=DeprecationWarning,
+)
+warnings.filterwarnings(
+    "ignore",
+    message=r".*websockets\.legacy is deprecated.*",
+    category=DeprecationWarning,
+)

server/hive/gateway/feishu_ws.py

@@ -71,6 +71,9 @@ def schedule_on_loop(
     closed (e.g. during server shutdown).  Never raises.
     """
     if loop.is_closed():
+        close = getattr(coro, "close", None)
+        if callable(close):
+            close()
         logger.warning("feishu_ws.loop_closed_dropping_event")
         return None
     return asyncio.run_coroutine_threadsafe(coro, loop)

server/hive/gateway/project_ws.py

@@ -20,7 +20,6 @@
 from hive.db import queries
 from hive.db.models import ProjectChannel
 from hive.gateway.feishu import FeishuMessage
-from hive.gateway.feishu_ws import build_event_handler, start_ws_client
 from hive.tracing import generate_trace_id, get_trace_id, set_trace_id
 
 logger = structlog.get_logger("hive.gateway.project_ws")
@@ -90,6 +89,8 @@ def start_channel(
         If False, private messages on this channel are silently ignored — the channel
         is used only for Scout group messages.
         """
+        from hive.gateway.feishu_ws import build_event_handler, start_ws_client
+
         self.stop_channel(channel_id)
 
         pool = self._pool

server/hive/runtime/command_runner.py

@@ -74,9 +74,11 @@ class CommandRunner:
 
     def __init__(self, *, shell_bin: str = "/bin/sh") -> None:
         self._shell_bin = shell_bin
+        self._active_processes: dict[str, subprocess.Popen[str]] = {}
 
     async def start(self, spec: CommandRunSpec) -> CommandRunHandle:
         command = self._normalize_command(spec.command)
+        run_id = uuid.uuid4().hex
         async with span(
             "sandbox.command.start",
             cwd=spec.cwd,
@@ -91,10 +93,11 @@ async def start(self, spec: CommandRunSpec) -> CommandRunHandle:
                 start_new_session=True,
                 text=True,
             )
+            self._active_processes[run_id] = process
             pid = process.pid
             Path(spec.pid_path).write_text(f"{pid}\n", encoding="utf-8")
             return CommandRunHandle(
-                run_id=uuid.uuid4().hex,
+                run_id=run_id,
                 pid=pid,
                 command=command,
                 cwd=spec.cwd,
@@ -121,6 +124,7 @@ async def snapshot(
             effective_started_at = started_at or handle.started_at
 
             if result_payload is not None:
+                self._reap_process(handle.run_id)
                 exit_code = self._as_int(result_payload.get("exit_code"))
                 timed_out = bool(result_payload.get("timed_out", False))
                 interrupted = bool(result_payload.get("interrupted", False))
@@ -177,6 +181,7 @@ async def cancel(self, handle: CommandRunHandle) -> None:
                     '{"exit_code":null,"duration_ms":0,"timed_out":false,"interrupted":true}',
                     encoding="utf-8",
                 )
+            self._reap_process(handle.run_id)
 
     @staticmethod
     def _normalize_command(command: list[str]) -> list[str]:
@@ -258,6 +263,18 @@ def _build_script(self, spec: CommandRunSpec, command: list[str]) -> str:
 EOF
 """
 
+    def _reap_process(self, run_id: str) -> None:
+        process = self._active_processes.get(run_id)
+        if process is None:
+            return
+        try:
+            process.wait(timeout=0)
+        except subprocess.TimeoutExpired:
+            # The shell wrapper is still unwinding; keep the handle so a later
+            # snapshot or cancel can reap it without leaking a ResourceWarning.
+            return
+        self._active_processes.pop(run_id, None)
+
     @staticmethod
     def _read_pid(path: Path) -> int | None:
         if not path.exists():

server/pyproject.toml

@@ -37,6 +37,12 @@ testpaths = ["tests"]
 markers = [
     "integration: end-to-end tests that require a running server (deselect with -m 'not integration')",
 ]
+filterwarnings = [
+    "ignore:datetime\\.datetime\\.utcfromtimestamp\\(\\) is deprecated.*:DeprecationWarning",
+    "ignore:There is no current event loop:DeprecationWarning",
+    "ignore:websockets\\.InvalidStatusCode is deprecated:DeprecationWarning",
+    "ignore:websockets\\.legacy is deprecated.*:DeprecationWarning",
+]
 
 [tool.ruff]
 target-version = "py312"
@@ -56,4 +62,3 @@ build-backend = "hatchling.build"
 
 [tool.hatch.build.targets.wheel]
 packages = ["hive"]
-

server/tests/test_agent/test_checkpoint.py

@@ -96,7 +96,9 @@ async def _fake_upsert(table: str, **kwargs):
 
     @asynccontextmanager
     async def fake_sf():
-        yield AsyncMock()
+        session = AsyncMock()
+        session.add = MagicMock()
+        yield session
 
     with patch("hive.agent.llm.create_provider", return_value=MagicMock(
         context_window=200_000,

server/tests/test_agent/test_llm_usage.py

@@ -13,6 +13,15 @@
 from hive.agent.scope import AgentScope
 
 
+def _session_factory() -> MagicMock:
+    session = AsyncMock()
+    session.__aenter__ = AsyncMock(return_value=session)
+    session.__aexit__ = AsyncMock(return_value=None)
+    session.add = MagicMock()
+    session.commit = AsyncMock()
+    return MagicMock(return_value=session)
+
+
 def _make_config(tmp_path: Path) -> AgentConfig:
     (tmp_path / "MEMORY.md").write_text("", encoding="utf-8")
     return AgentConfig(
@@ -46,7 +55,7 @@ async def test_llm_usage_inserted_on_run(tmp_path):
     async def fake_record_llm_usage(**kwargs):
         inserted_rows.append(kwargs)
 
-    agent.db._session_factory = MagicMock()
+    agent.db._session_factory = _session_factory()
     agent.db.record_llm_usage = fake_record_llm_usage  # type: ignore[method-assign]
     agent.db.update_one = AsyncMock(return_value=None)
 
@@ -94,7 +103,7 @@ async def test_llm_usage_insert_failure_does_not_raise(tmp_path):
     """A DB write failure for llm_usage is logged as warning, not raised."""
     config = _make_config(tmp_path)
     agent = HiveAgent(config)
-    agent.db._session_factory = MagicMock()
+    agent.db._session_factory = _session_factory()
 
     async def failing_record_llm_usage(**kwargs):
         raise RuntimeError("db error")
@@ -116,7 +125,7 @@ async def test_llm_usage_trace_id_captured(tmp_path):
 
     config = _make_config(tmp_path)
     agent = HiveAgent(config)
-    agent.db._session_factory = MagicMock()
+    agent.db._session_factory = _session_factory()
 
     captured: list[dict] = []
 

server/tests/test_agent/test_runtime.py

@@ -11,6 +11,15 @@
 from hive.agent.scope import AgentScope
 
 
+def _session_factory() -> MagicMock:
+    session = AsyncMock()
+    session.__aenter__ = AsyncMock(return_value=session)
+    session.__aexit__ = AsyncMock(return_value=None)
+    session.add = MagicMock()
+    session.commit = AsyncMock()
+    return MagicMock(return_value=session)
+
+
 def _make_config(tmp_path: Path, tools: list[str] | None = None) -> AgentConfig:
     (tmp_path / "MEMORY.md").write_text("", encoding="utf-8")
     return AgentConfig(
@@ -179,6 +188,8 @@ async def test_run_writes_session_metrics_to_db(tmp_path):
     )
 
     mock_session = _AsyncMock()
+    mock_session.add = MagicMock()
+    mock_session.commit = _AsyncMock()
 
     @asynccontextmanager
     async def fake_factory():
@@ -211,7 +222,7 @@ async def test_run_writes_agent_run_and_events(tmp_path):
     """agent.run() should create a run spine with ordered runtime events."""
     config = _make_config(tmp_path)
     agent = HiveAgent(config)
-    agent.db._session_factory = MagicMock()
+    agent.db._session_factory = _session_factory()
 
     inserts: list[tuple[str, dict]] = []
     updates: list[tuple[str, object, dict]] = []
@@ -247,7 +258,7 @@ async def test_run_marks_scheduler_invocation_source(tmp_path):
     """Scheduler-triggered runs must get a separate runtime-context channel marker."""
     config = _make_config(tmp_path)
     agent = HiveAgent(config)
-    agent.db._session_factory = MagicMock()
+    agent.db._session_factory = _session_factory()
 
     captured_runs: list[dict] = []
 

server/tests/test_gateway/test_api_channel.py

@@ -74,6 +74,8 @@ def _build_app(role: str = "pm", user_id: str = "pm_alice") -> tuple[FastAPI, As
     app = FastAPI()
     app.include_router(router)
     session = AsyncMock()
+    session.add = MagicMock()
+    session.commit = AsyncMock()
     user = {"user_id": user_id, "role": role}
     app.dependency_overrides[get_current_user] = lambda: user
     app.dependency_overrides[get_db_session] = _async_gen(session)