Function test_slurm_allocation_name

tests/autoalloc/test_autoalloc.py:612–632

(hq_env: HqEnv)

def test_slurm_allocation_name(hq_env: HqEnv):
    handler = ExtractSubmitScriptPath(SlurmManagerFlavor().create_adapter())

    def check_name(path: str, name: str):
        with open(path) as f:
            data = f.read()
            slurm_args = extract_script_args(data, "#SBATCH")
            for arg in slurm_args:
                if "--job-name=" in arg:
                    assert arg[len("--job-name=") :] == name
                    return
            raise Exception(f"Slurm name {name} not found in {path}")

    with MockJobManager(hq_env, handler):
        start_server_with_quick_refresh(hq_env)
        prepare_tasks(hq_env)

        add_queue(hq_env, manager="slurm", name="foo", backlog=2)
        check_name(handler.get_script_path(), "foo-1")
        handler.add_worker(hq_env, default_job_id())
        check_name(handler.get_script_path(), "foo-2")


@all_flavors