EngineX/xc-llm-ascend
3
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[Test] Add the always_check_nodes parameter to the _wait_for_multiple_servers function in conftest.py for the EPD test case. (#7410)

Browse Source 
### What this PR does / why we need it?
This PR add the always_check_nodes parameter to the
_wait_for_multiple_servers function in conftest.py for the EPD test
case.

### Does this PR introduce _any_ user-facing change?
no

### How was this patch tested?
1.by running the test
`pytest -sv test_disaggregated_encoder.py`

2.by running ci

- vLLM version: v0.17.0
- vLLM main:
4497431df6

---------

Signed-off-by: yenuo26 <410167048@qq.com>
This commit is contained in:
wangyu
2026-03-20 11:33:48 +08:00
committed by GitHub
parent eb92e7d50e
commit 7be66cec75
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/e2e/conftest.py
View File

@@ -349,7 +349,9 @@ class RemoteOpenAIServer:
# Then wait for all api_server nodes # Then wait for all api_server nodes
self._wait_for_multiple_servers(targets=targets, timeout=timeout) self._wait_for_multiple_servers(targets=targets, timeout=timeout)
def _wait_for_multiple_servers(self, targets, timeout: float, log_interval: float = 30.0): def _wait_for_multiple_servers(
self, targets, timeout: float, log_interval: float = 30.0, always_check_nodes: bool = False
):
""" """
targets: List[(node_ip, url)] targets: List[(node_ip, url)]
log_interval log_interval
@@ -367,14 +369,14 @@ class RemoteOpenAIServer:
should_log = (now - last_log_time) >= log_interval should_log = (now - last_log_time) >= log_interval
for node_ip, url in targets: for node_ip, url in targets:
if ready[node_ip]: if ready[node_ip] and not always_check_nodes:
continue continue
try: try:
resp = client.get(url) resp = client.get(url)
if resp.status_code == 200: if resp.status_code == 200:
ready[node_ip] = True ready[node_ip] = True
logger.info(f"[READY] Node {node_ip} is ready.") logger.info(f"[READY] Node {node_ip}: {url} is ready.")
except RequestException: except RequestException:
all_ready = False all_ready = False
if should_log: if should_log:
@@ -498,7 +500,9 @@ class RemoteEPDServer(RemoteOpenAIServer):
self._proc_list.append(proc) self._proc_list.append(proc)
timeout_value = float(max_wait_seconds) if max_wait_seconds is not None else 2800.0 timeout_value = float(max_wait_seconds) if max_wait_seconds is not None else 2800.0
super()._wait_for_multiple_servers([(self.host, url) for url in self.health_url_list], timeout=timeout_value) super()._wait_for_multiple_servers(
[(self.host, url) for url in self.health_url_list], timeout=timeout_value, always_check_nodes=True
)
def _poll(self) -> int | None: def _poll(self) -> int | None:
return None return None