mirror of
https://github.com/pewdiepie-archdaemon/odysseus.git
synced 2026-10-27 16:22:20 +01:00
Scope backend checks to backend behavior and retain ownership regression coverage
This commit is contained in:
@@ -1,51 +0,0 @@
|
||||
"""Graceful-shutdown signalling tests (HARNESS-owned).
|
||||
|
||||
SSE / streaming responses never complete on their own, so uvicorn's graceful
|
||||
shutdown would wait for an open browser stream up to the whole
|
||||
``timeout_graceful_shutdown`` window. src.shutdown is a process-wide flag that
|
||||
the SIGTERM/SIGINT handler flips and every streaming generator checks, so a
|
||||
restart ends open streams promptly while ordinary in-flight requests still
|
||||
finish.
|
||||
"""
|
||||
from pathlib import Path
|
||||
|
||||
from src import shutdown
|
||||
|
||||
ROOT = Path(__file__).resolve().parent.parent
|
||||
|
||||
|
||||
def test_request_shutdown_is_sticky_and_resettable():
|
||||
shutdown.reset()
|
||||
try:
|
||||
assert shutdown.is_shutting_down() is False
|
||||
shutdown.request_shutdown()
|
||||
assert shutdown.is_shutting_down() is True
|
||||
# idempotent
|
||||
shutdown.request_shutdown()
|
||||
assert shutdown.is_shutting_down() is True
|
||||
finally:
|
||||
shutdown.reset()
|
||||
assert shutdown.is_shutting_down() is False
|
||||
|
||||
|
||||
def test_app_installs_its_own_shutdown_handler():
|
||||
src = (ROOT / "app.py").read_text(encoding="utf-8")
|
||||
# This uvicorn version has no install_signal_handlers kwarg, so app.py wraps
|
||||
# server.handle_exit (which capture_signals wires SIGINT/SIGTERM to).
|
||||
assert "install_signal_handlers=False" not in src
|
||||
assert "server.handle_exit = _handle_shutdown_signal" in src
|
||||
assert "_shutdown.request_shutdown()" in src
|
||||
assert "server.run()" in src
|
||||
|
||||
|
||||
def test_streaming_generators_check_the_shutdown_flag():
|
||||
# editor bridge SSE + chat stream/resume + research + email events + shell.
|
||||
for rel in (
|
||||
"routes/editor_agent_routes.py",
|
||||
"routes/chat_routes.py",
|
||||
"routes/research/research_routes.py",
|
||||
"routes/email/email_routes.py",
|
||||
"routes/shell_routes.py",
|
||||
):
|
||||
src = (ROOT / rel).read_text(encoding="utf-8")
|
||||
assert "_shutdown.is_shutting_down()" in src, rel
|
||||
Reference in New Issue
Block a user