modelcontextprotocol/python-sdk
Watch
1
0
Fork
You've already forked python-sdk
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: Replace remaining manual server polling with wait_for_server helper (#1529)

Browse source
This commit is contained in:
Max Isbey 2025-10-29 11:27:57 +00:00 • committed by GitHub
commit db9e451551
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 12 additions and 61 deletions
Download patch file Download diff file Expand all files Collapse all files

26
tests/shared/test_sse.py
View file

@ -252,19 +252,8 @@ def mounted_server(server_port: int) -> Generator[None, None, None]:
proc.start()
# Wait for server to be running
max_attempts = 20
attempt = 0
print("waiting for server to start")
while attempt < max_attempts:
try:
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
s.connect(("127.0.0.1", server_port))
break
except ConnectionRefusedError:
time.sleep(0.1)
attempt += 1
else:
raise RuntimeError(f"Server failed to start after {max_attempts} attempts")
wait_for_server(server_port)
yield
@ -367,19 +356,8 @@ def context_server(server_port: int) -> Generator[None, None, None]:
proc.start()
# Wait for server to be running
max_attempts = 20
attempt = 0
print("waiting for context server to start")
while attempt < max_attempts:
try:
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
s.connect(("127.0.0.1", server_port))
break
except ConnectionRefusedError:
time.sleep(0.1)
attempt += 1
else:
raise RuntimeError(f"Context server failed to start after {max_attempts} attempts")
wait_for_server(server_port)
yield