1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192
|
#!/usr/bin/env expect
############################################################################
# Purpose: Test of Slurm functionality
# Test routing all signals through slurmctld rather than slurmd
# (undocumented --ctld option).
############################################################################
# Copyright (C) 2002-2007 The Regents of the University of California.
# Produced at Lawrence Livermore National Laboratory (cf, DISCLAIMER).
# Written by Morris Jette <jette1@llnl.gov>
# UCRL-CODE-217948.
#
# This file is part of Slurm, a resource management program.
# For details, see <https://slurm.schedmd.com/>.
# Please also read the included file: DISCLAIMER.
#
# Slurm is free software; you can redistribute it and/or modify it under
# the terms of the GNU General Public License as published by the Free
# Software Foundation; either version 2 of the License, or (at your option)
# any later version.
#
# Slurm is distributed in the hope that it will be useful, but WITHOUT ANY
# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
# FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
# details.
#
# You should have received a copy of the GNU General Public License along
# with Slurm; if not, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
############################################################################
source ./globals
set file_in "$test_dir/input"
set file_out "$test_dir/output"
set file_prog "$test_name.prog"
set job_id 0
#
# Cannot run the test if OverTimeLimit is set, since we test time limits.
#
regexp "($number)" [get_config_param "OverTimeLimit"] {} overtimelim
if {$overtimelim != 0} {
skip "Cannot run this test when OverTimeLimit is set. Exiting now"
}
proc cleanup {} {
global job_id file_prog
cancel_job $job_id
file delete $file_prog
}
#
# Delete left-over programs and rebuild them.
# We use our own program to get ulimit values since the output
# of the ulimit program is inconsistent across systems.
#
exec $bin_cc -O -o $file_prog ${file_prog}.c
make_bash_script $file_in "
$srun ./$file_prog
"
#
# Submit a job so we have something to work with
#
set job_id [submit_job -fail "--output=$file_out -t1 $file_in"]
wait_for_job -fail $job_id RUNNING
# Allow time for poe to start the step
# This could take a while with launch/poe
sleep 30
#
# Test verbose scancel
#
set matches 0
spawn $scancel --signal=USR1 --ctld --verbose $job_id
expect {
-re "Signal ($number) to job $job_id" {
set sig_num $expect_out(1,string)
if {$sig_num != 9} {
incr matches
} else {
incr matches -1
}
exp_continue
}
-re "Job is in transitional state, retrying" {
incr matches -1
exp_continue
}
timeout {
fail "scancel not responding"
}
eof {
wait
}
}
subtest {$matches == 1} "Test verbose scancel"
wait_for_job -fail $job_id DONE
#
# Test output file
#
wait_for_file -fail $file_out
set got_signal 0
spawn cat $file_out
expect {
-re "ERROR" {
fail "Failure running test program"
}
-re "Received SIGUSR1" {
set got_signal 1
exp_continue
}
eof {
wait
}
}
subtest {$got_signal == 1} "Verify SIGUSR1 received"
#
# Variation of above to test signal of batch script processes
# Note that job termination sends SIGCONT/SIGTERM, hence the
# duplicate sleep call
#
exec rm -f $file_in $file_out
make_bash_script $file_in "
echo BEGIN
$bin_sleep 10
$bin_sleep 10
$bin_sleep 10
echo FINI
"
#
# Submit a job so we have something to work with
#
set job_id [submit_job -fail "--output=$file_out -t1 $file_in"]
wait_for_job -fail $job_id RUNNING
sleep 5
#
# Test verbose scancel
#
set matches 0
spawn $scancel --batch --signal=STOP --ctld --verbose $job_id
expect {
-re "Signal ($number) to batch job $job_id" {
set sig_num $expect_out(1,string)
if {$sig_num != 9} {
incr matches
} else {
incr matches -1
}
exp_continue
}
-re "Job is in transitional state, retrying" {
incr matches -1
exp_continue
}
timeout {
fail "scancel not responding"
}
eof {
wait
}
}
subtest {$matches == 1} "Verify SIGSTOP received"
wait_for_job -fail $job_id DONE
#
# Test output file
#
wait_for_file -fail $file_out
set matches 0
spawn cat $file_out
expect {
-re "FINI" {
fail "Job not stopped"
}
-re "CANCELLED" {
set matches 1
exp_continue
}
eof {
wait
}
}
subtest {$matches == 1} "Expecting CANCELLED error message"
|