| ############################################################################ |
| # Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved. |
| ############################################################################ |
| import re |
| import atf |
| |
| |
| def setup_module(): |
| atf.require_version( |
| (26, 5), |
| "bin/scontrol", |
| reason="Ticket 22180: SLUID availability in scontrol added in 26.05+", |
| ) |
| atf.require_nodes(1, [("CPUs", 1)]) |
| atf.require_config_parameter_includes("JobAcctGatherType", "jobacct_gather/cgroup") |
| atf.require_slurm_running() |
| |
| |
| def test_cgroup_path_contains_sluid(): |
| """Verify that a job step's cgroup path contains the job's SLUID. |
| |
| Expected cgroup line format: |
| 0::/.../<sluid>/step_0/user/task_0 |
| """ |
| |
| file_out = atf.module_tmp_path / "cgroup_output" |
| job_id = atf.submit_job_sbatch( |
| f"-n1 --output={file_out} --wrap 'srun cat /proc/self/cgroup'", |
| fatal=True, |
| ) |
| atf.wait_for_job_state(job_id, "DONE", fatal=True) |
| |
| sluid = atf.get_job_parameter(job_id, "SLUID") |
| assert sluid is not None, f"Job {job_id} has no SLUID" |
| |
| atf.wait_for_file(file_out, fatal=True) |
| |
| cgroup_output = atf.run_command_output(f"cat {file_out}", fatal=True) |
| |
| assert re.search( |
| rf"/{re.escape(sluid)}/step_0/user/task_0", cgroup_output |
| ), f"Expected SLUID '{sluid}' in cgroup path, got: {cgroup_output}" |
