66 lines
2.1 KiB
Perl
66 lines
2.1 KiB
Perl
|
use base "installedtest";
|
||
|
|
use strict;
|
||
|
|
use testapi;
|
||
|
|
use utils;
|
||
|
|
|
||
|
|
sub slurm_setup {
|
||
|
|
# install HPC repository
|
||
|
|
my $version = get_var("SLURM_VERSION");
|
||
|
|
assert_script_run "dnf -y install rocky-release-hpc", 120;
|
||
|
|
|
||
|
|
# Set up munge
|
||
|
|
assert_script_run "dnf -y install munge", 120;
|
||
|
|
assert_script_run "dd if=/dev/urandom bs=1 count=1024 >/etc/munge/munge.key";
|
||
|
|
assert_script_run "chmod 400 /etc/munge/munge.key";
|
||
|
|
assert_script_run "chown munge.munge /etc/munge/munge.key";
|
||
|
|
assert_script_run "systemctl enable --now munge.service";
|
||
|
|
|
||
|
|
# install slurm
|
||
|
|
if (get_var("CURRREL") eq '8') {
|
||
|
|
assert_script_run "dnf config-manager --set-enabled powertools";
|
||
|
|
}
|
||
|
|
assert_script_run "dnf install -y slurm$version-slurmdbd slurm$version-slurmrestd slurm$version-slurmctld slurm$version-slurmd";
|
||
|
|
|
||
|
|
# Since this is a single node system, we don't have to modify the conf files. We will for larger multi-node tests.
|
||
|
|
# start services
|
||
|
|
assert_script_run "systemctl enable --now slurmctld slurmdbd slurmrestd slurmd";
|
||
|
|
}
|
||
|
|
|
||
|
|
sub run {
|
||
|
|
my $self = shift;
|
||
|
|
|
||
|
|
# do all the install stuff
|
||
|
|
slurm_setup();
|
||
|
|
|
||
|
|
# if everything is configured right, sinfo should show the following output
|
||
|
|
# $ sinfo
|
||
|
|
# PARTITION AVAIL TIMELIMIT NODES STATE NODELIST
|
||
|
|
# debug* up infinite 1 idle localhost
|
||
|
|
validate_script_output "sinfo", sub { m/debug.*localhost/ };
|
||
|
|
|
||
|
|
# write a boring job script
|
||
|
|
assert_script_run "echo '#!/bin/bash\n#SBATCH --job-name=antarctica_time\nsleep 120\nTZ=NZST date' > job.sh";
|
||
|
|
|
||
|
|
## schedule a job and run it to completion
|
||
|
|
assert_script_run "sbatch job.sh";
|
||
|
|
validate_script_output "squeue", sub { m/antar/ };
|
||
|
|
sleep 121;
|
||
|
|
# after 121 seconds, job should have completed and no longer exist in the queue
|
||
|
|
validate_script_output "squeue", sub { $_ !~ m/antar/ };
|
||
|
|
|
||
|
|
## cancel a job
|
||
|
|
assert_script_run "sbatch job.sh";
|
||
|
|
validate_script_output "squeue", sub { m/antar/ };
|
||
|
|
assert_script_run "scancel 2";
|
||
|
|
# job should no longer be in the queue
|
||
|
|
validate_script_output "squeue", sub { $_ !~ m/antar/ };
|
||
|
|
}
|
||
|
|
|
||
|
|
sub test_flags {
|
||
|
|
return {fatal => 1};
|
||
|
|
}
|
||
|
|
|
||
|
|
1;
|
||
|
|
|
||
|
|
# vim: set sw=4 et:
|