vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

server tests: handle TimeoutExpired exception

Browse source
This commit is contained in:
Jared Van Bortel 2024-03-19 11:03:11 -04:00
parent 731134958f
commit 4479b85b8a
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
examples/server/tests/features/environment.py
View file

@ -5,6 +5,7 @@ import sys
import time import time
import traceback import traceback
from contextlib import closing from contextlib import closing
from subprocess import TimeoutExpired
def before_scenario(context, scenario): def before_scenario(context, scenario):
@ -38,7 +39,9 @@ def after_scenario(context, scenario):
server_graceful_shutdown(context) # SIGINT server_graceful_shutdown(context) # SIGINT
if context.server_process.wait(0.5) is None: try:
context.server_process.wait(0.5)
except TimeoutExpired:
print(f"server still alive after 500ms, force-killing pid={context.server_process.pid} ...") print(f"server still alive after 500ms, force-killing pid={context.server_process.pid} ...")
context.server_process.kill() # SIGKILL context.server_process.kill() # SIGKILL
context.server_process.wait() context.server_process.wait()