aljaz/mcp-python-sdk
1
0
Fork 0
mirror of https://github.com/aljazceru/mcp-python-sdk.git synced 2025-12-19 14:54:24 +01:00
Code Issues Packages Projects Releases Wiki Activity

attempt at server in process

Browse Source
This commit is contained in:
Nick Merrill
2025-01-14 11:15:05 -05:00
parent 7ab1fc71aa
commit 8d90a3afa3
1 changed files with 12 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
tests/shared/test_sse.py
View File

@@ -1,5 +1,6 @@
# test_sse.py # test_sse.py
import re import re
import multiprocessing
import socket import socket
import time import time
import json import json
@@ -94,10 +95,15 @@ def space_around_test():
@pytest.fixture() @pytest.fixture()
def server(server_app: Starlette, server_port: int): def server(server_app: Starlette, server_port: int):
server = uvicorn.Server(config=uvicorn.Config(app=server_app, host="127.0.0.1", port=server_port, log_level="error")) proc = multiprocessing.Process(target=uvicorn.run, daemon=True, kwargs={
server_thread = threading.Thread( target=server.run, daemon=True ) "app": server_app,
"host": "127.0.0.1",
"port": server_port,
"log_level": "error"
})
print(f'starting server on {server_port}') print(f'starting server on {server_port}')
server_thread.start() proc.start()
# Give server time to start # Give server time to start
while not server.started: while not server.started:
print('waiting for server to start') print('waiting for server to start')
@@ -117,8 +123,9 @@ def server(server_app: Starlette, server_port: int):
s.close() s.close()
# Wait for thread to finish # Wait for thread to finish
server_thread.join(timeout=2) proc.terminate()
if server_thread.is_alive(): proc.join(timeout=2)
if proc.is_alive():
print("Warning: Server thread did not exit cleanly") print("Warning: Server thread did not exit cleanly")
# Optionally, you could add more aggressive cleanup here # Optionally, you could add more aggressive cleanup here
import _thread import _thread