| ############################################################################ |
| # Copyright (C) SchedMD LLC. |
| ############################################################################ |
| import atf |
| import pytest |
| |
| |
| # Setup/Teardown |
| @pytest.fixture(scope="module", autouse=True) |
| def setup(): |
| atf.require_config_parameter("PriorityType", "priority/basic") |
| atf.require_slurm_running() |
| |
| |
| @pytest.fixture(scope="module") |
| def section_description(): |
| """Submit three jobs (with one held) and evaluate the priorities""" |
| |
| |
| @pytest.fixture(scope="module") |
| def first_job(): |
| """Submit a non-held job""" |
| return atf.submit_job_sbatch(fatal=True) |
| |
| |
| @pytest.fixture(scope="module") |
| def second_job(): |
| """Submit a second non-held job""" |
| return atf.submit_job_sbatch(fatal=True) |
| |
| |
| @pytest.fixture(scope="module") |
| def held_job(): |
| """Submit a held job""" |
| return atf.submit_job_sbatch('--hold --wrap="sleep 60"', fatal=True) |
| |
| |
| @pytest.fixture(scope="module") |
| def jobs_dict(section_description, first_job, second_job, held_job): |
| """Return dictionary from get_jobs""" |
| return atf.get_jobs() |
| |
| |
| def test_relative_job_priorities(jobs_dict, first_job, second_job): |
| """Verify that first job higher priority""" |
| assert jobs_dict[first_job]["Priority"] > jobs_dict[second_job]["Priority"] |
| |
| |
| def test_held_job_has_zero_priority(jobs_dict, held_job): |
| """Verify that the held job has zero priority""" |
| assert jobs_dict[held_job]["Priority"] == 0 |
