testsuite/python/tests/test_101_1.py

############################################################################
# Copyright (C) SchedMD LLC.
############################################################################
import re

import pytest

import atf


@pytest.fixture(scope="module", autouse=True)
def setup():
    if atf.get_version("bin/sacct") < (25, 5):
        atf.require_accounting()
        atf.require_slurm_running()


@pytest.mark.parametrize("opt", ["--help", "-h"])
def test_help(opt):
    """Verify sacct --help displays the help page"""

    output = atf.run_command_output(f"sacct {opt}", fatal=True)

    assert re.search(r"sacct \[<OPTION>\]", output) is not None
    assert re.search(r"Valid <OPTION> values are:", output) is not None
    assert re.search(r"-A, --accounts:", output) is not None
    assert re.search(r"-j, --jobs:", output) is not None
    assert re.search(r"-s, --state:", output) is not None
    assert re.search(r"-S, --starttime:", output) is not None
    assert re.search(r"valid start/end time formats are...", output) is not None
    # TODO: Ticket 25236 - remove once 26.11 is the oldest supported version
    if atf.get_version("bin/sacct") >= (26, 11):
        assert (
            "-J, --job-name" in output
        ), "sacct --help should advertise -J, --job-name (ticket 25236)"


@pytest.mark.parametrize("opt", ["--helpformat", "-e"])
def test_helpformat(opt):
    """Verify sacct --helpformat displays the expected fields"""

    output = atf.run_command_output(f"sacct {opt}", fatal=True)

    assert re.search(r"Account", output) is not None
    assert re.search(r"ExitCode", output) is not None
    assert re.search(r"JobID", output) is not None
    assert re.search(r"NodeList", output) is not None
    assert re.search(r"ReqMem", output) is not None
    assert re.search(r"Start", output) is not None
    assert re.search(r"State", output) is not None
    assert re.search(r"Timelimit", output) is not None
    assert re.search(r"User", output) is not None
    assert re.search(r"WorkDir", output) is not None

    if atf.get_version("bin/sacct") >= (26, 5):
        assert (
            re.search(r"\bExclusive\b", output) is not None
        ), "sacct --helpformat should advertise the Exclusive field (ticket 23303)"
        assert (
            re.search(r"\bOverSubscribe\b", output) is not None
        ), "sacct --helpformat should advertise the OverSubscribe field (ticket 23303)"

    if atf.get_version("bin/sacct") >= (26, 11):
        assert (
            re.search(r"\bSlurmRC\b", output) is not None
        ), "sacct --helpformat should advertise the SlurmRC field (ticket 24195)"