29 lines
695 B
Python
29 lines
695 B
Python
"""Minimal test session cleanup to prevent hanging background servers.
|
|
|
|
This keeps the test suite simple while ensuring any stray embedding server
|
|
processes are terminated at session start and end.
|
|
"""
|
|
|
|
import subprocess
|
|
|
|
|
|
def _kill_embedding_servers() -> None:
|
|
patterns = [
|
|
"hnsw_embedding_server",
|
|
"diskann_embedding_server",
|
|
"embedding_server",
|
|
]
|
|
for pat in patterns:
|
|
try:
|
|
subprocess.run(["pkill", "-9", "-f", pat], timeout=2, capture_output=True)
|
|
except Exception:
|
|
pass
|
|
|
|
|
|
def pytest_sessionstart(session):
|
|
_kill_embedding_servers()
|
|
|
|
|
|
def pytest_sessionfinish(session, exitstatus):
|
|
_kill_embedding_servers()
|