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

import atf


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

    atf.require_accounting()
    atf.require_config_parameter("AccountingStorageTres", "gres/gpu")
    atf.require_config_parameter_includes("AccountingStorageEnforce", "limits")

    atf.require_config_parameter("SelectType", "select/cons_tres")
    atf.require_config_parameter("SelectTypeParameters", "CR_Core")

    atf.require_config_parameter_includes("GresTypes", "gpu")
    atf.require_nodes(1, [("CPUs", 8), ("Gres", "gpu:4")])

    atf.require_tty(0)
    atf.require_tty(1)
    atf.require_tty(2)
    atf.require_tty(3)
    atf.require_config_file(
        "gres.conf",
        "Name=gpu File=/dev/tty[0-3]",
    )

    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()