Function test_submit_task_dir

tests/test_job.py:937–966

(hq_env: HqEnv, mode)

@pytest.mark.parametrize("mode", ["FINISHED", "FAILED", "CANCELED"])
def test_submit_task_dir(hq_env: HqEnv, mode):
    hq_env.start_server()
    hq_env.start_worker()

    hq_env.command(
        [
            "submit",
            "--task-dir",
            "--",
            "bash",
            "-c",
            "echo $HQ_TASK_DIR; touch $HQ_TASK_DIR/xyz; sleep 2; {}".format("exit 1" if mode == "FAILED" else ""),
        ]
    )
    wait_for_job_state(hq_env, 1, "RUNNING")

    def file_created():
        task_dir = read_file(default_task_output()).rstrip()
        if task_dir is None:
            return
        return os.path.isfile(os.path.join(task_dir, "xyz"))

    wait_until(file_created)
    task_dir = read_file(default_task_output()).rstrip()

    if mode == "CANCELED":
        hq_env.command(["job", "cancel", "all"])

    wait_for_job_state(hq_env, 1, mode)
    assert not os.path.exists(task_dir)


def test_custom_error_message(hq_env: HqEnv):