Skip to content

Commit 917bc6a

Browse files
fix test for entrypoint log
Signed-off-by: chrisfellowes <[email protected]>
1 parent 109ff4b commit 917bc6a

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

python/ray/dashboard/modules/job/tests/test_job_manager.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -335,8 +335,9 @@ async def test_runtime_env_setup_logged_to_job_driver_logs(
335335
gcs_client = ray._private.worker.global_worker.gcs_client
336336
job_manager = JobManager(gcs_client, tmp_path)
337337

338+
entrypoint = "echo hello 1"
338339
job_id = await job_manager.submit_job(
339-
entrypoint="echo hello 1", submission_id="test_runtime_env_setup_logs"
340+
entrypoint=entrypoint, submission_id="test_runtime_env_setup_logs"
340341
)
341342
await async_wait_for_condition(
342343
check_job_succeeded, job_manager=job_manager, job_id=job_id
@@ -347,10 +348,10 @@ async def test_runtime_env_setup_logged_to_job_driver_logs(
347348
ray._private.worker._global_node.get_logs_dir_path(),
348349
f"job-driver-{job_id}.log",
349350
)
350-
start_message = "Runtime env is setting up."
351351
with open(job_driver_log_path, "r") as f:
352352
logs = f.read()
353-
assert start_message in logs
353+
assert "Runtime env is setting up." in logs
354+
assert f"Running entrypoint for job {job_id}: {entrypoint}" in logs
354355

355356

356357
@pytest.mark.asyncio

0 commit comments

Comments
 (0)