testsuite/python/expect/test_21_30.py

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

import atf

pytestmark = pytest.mark.slow


@pytest.fixture(scope="module", autouse=True)
def setup():
    atf.require_expect()

    atf.require_accounting()
    atf.require_config_parameter_includes("AccountingStorageEnforce", "limits")
    atf.require_config_parameter_includes("AccountingStorageEnforce", "safe")
    atf.require_config_parameter("SelectType", "select/cons_tres")
    atf.require_config_parameter("SelectTypeParameters", "CR_Core_Memory")

    # Avoid usage decay to simplify test and avoiding false failures
    atf.require_config_parameter("PriorityDecayHalfLife", "0")
    atf.require_config_parameter("PriorityUsageResetPeriod", "YEARLY")

    # Allow 20 parallel jobs of 3s to speed up test to reach 1min usage
    atf.require_nodes(
        20,
        [
            ("CPUS", 8),
            ("Sockets", 2),
            ("CoresPerSocket", 2),
            ("ThreadsPerCore", 2),
            ("RealMemory", 2048),
        ],
    )
    atf.require_slurm_running()

    atf.run_command(
        f"sacctmgr -i create user {atf.properties['slurm-user']} account=root",
        user=atf.properties["slurm-user"],
        fatal=True,
    )


def test_expect():
    atf.run_expect_test()