Add test7.12 Test of slurm_job_step_stat() API call.

testsuite/expect/Makefile.am

@@ -182,6 +182,8 @@ EXTRA_DIST = \
 	test7.10			\
 	test7.11			\
 	test7.11.prog.c			\
+	test7.12			\
+	test7.12.prog.c			\
 	test8.1				\
 	test8.2				\
 	test8.3				\

testsuite/expect/Makefile.in

@@ -452,6 +452,8 @@ EXTRA_DIST = \
 	test7.10			\
 	test7.11			\
 	test7.11.prog.c			\
+	test7.12			\
+	test7.12.prog.c			\
 	test8.1				\
 	test8.2				\
 	test8.3				\

testsuite/expect/README

@@ -307,6 +307,7 @@ test7.9    Test that no files are open in spawned tasks (except stdin,
 test7.10   Test if we can trick SLURM into using the wrong user ID
            through an LD_PRELOAD option.
 test7.11   Test of SPANK plugin.
+test7.12   Test of slurm_job_step_stat() API call.
 
 
 test8.#    Test of Blue Gene specific functionality.

testsuite/expect/test7.12

+#!/usr/bin/expect
+############################################################################
+# Purpose:  Test of slurm_job_step_stat() API call.
+#
+# Output:  "TEST: #.#" followed by "SUCCESS" if test was successful, OR
+#          "FAILURE: ..." otherwise with an explanation of the failure, OR
+#          anything else indicates a failure mode that must be investigated.
+#
+# Note:    This script generates and then deletes a file in the working
+#          directory named test7.12.prog
+############################################################################
+# Copyright (C) 2010 Lawrence Livermore National Security.
+# Produced at Lawrence Livermore National Laboratory (cf, DISCLAIMER).
+# Written by Morris Jette <jette1@llnl.gov>
+# CODE-OCEC-09-009. All rights reserved.
+#
+# This file is part of SLURM, a resource management program.
+# For details, see <https://computing.llnl.gov/linux/slurm/>.
+# 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 test_id     "7.12"
+set exit_code   0
+set file_in     "test$test_id.input"
+set job_id      0
+set test_prog   "test$test_id.prog"
+
+print_header $test_id
+
+if {[test_front_end] != 0} {
+        send_user "\nWARNING: This test is incompatible with front-end systems\n"
+        exit 0
+}
+
+#
+# Delete left-over program and rebuild it
+#
+file delete $file_in $test_prog
+make_bash_script $file_in "
+  $srun $bin_sleep 120
+"
+
+send_user "slurm_dir is $slurm_dir\n"
+if {[test_aix]} {
+	send_user "$bin_cc ${test_prog}.c -Wl,-brtl -g -pthread -o ${test_prog} -I${slurm_dir}/include  -L${slurm_dir}/lib -lslurm -lntbl\n"
+	exec       $bin_cc ${test_prog}.c -Wl,-brtl -g -pthread -o ${test_prog} -I${slurm_dir}/include  -L${slurm_dir}/lib -lslurm -lntbl
+} elseif [file exists ${slurm_dir}/lib64/libslurm.so] {
+	send_user "$bin_cc ${test_prog}.c -g -pthread -o ${test_prog} -I${slurm_dir}/include -Wl,--rpath=${slurm_dir}/lib64 -L${slurm_dir}/lib64 -lslurm\n"
+	exec       $bin_cc ${test_prog}.c -g -pthread -o ${test_prog} -I${slurm_dir}/include -Wl,--rpath=${slurm_dir}/lib64 -L${slurm_dir}/lib64 -lslurm
+} else {
+	send_user "$bin_cc ${test_prog}.c -g -pthread -o ${test_prog} -I${slurm_dir}/include -Wl,--rpath=${slurm_dir}/lib -L${slurm_dir}/lib -lslurm\n"
+	exec       $bin_cc ${test_prog}.c -g -pthread -o ${test_prog} -I${slurm_dir}/include -Wl,--rpath=${slurm_dir}/lib -L${slurm_dir}/lib -lslurm
+}
+exec $bin_chmod 700 $test_prog
+
+#
+# Spawn program via sbatch
+#
+spawn $sbatch -N1 -t1 --output=/dev/null $file_in
+expect {
+	-re "Submitted batch job ($number)" {
+		set job_id $expect_out(1,string)
+		exp_continue
+	}
+	timeout {
+		send_user "\nFAILURE: sbatch not responding\n"
+		set exit_code 1
+		exp_continue
+	}
+	eof {
+		wait
+	}
+}
+if { $job_id == 0 } {
+	send_user "\nFAILURE: failed to submit job\n"
+	exit 1
+}
+
+#
+# Wait for job to start
+#
+if {[wait_for_job $job_id "RUNNING"] != 0} {
+	send_user "\nFAILURE: waiting for job to start\n"
+	cancel_job $job_id
+	exit 1
+}
+# wait for job step to start
+sleep 2
+
+set matches 0
+spawn ./$test_prog $job_id
+expect {
+	-re "job_id:$job_id" {
+		incr matches
+		exp_continue
+	}
+	-re "pid:$number" {
+		incr matches
+		exp_continue
+	}
+	timeout {
+		send_user "\nFAILURE: spawn IO not responding\n"
+		cancel_job $job_id
+		set exit_code 1
+	}
+	eof {
+		wait
+	}
+}
+
+if {$matches != 2} {
+	send_user "\nFAILURE: error running slurm_job_step_stat() program\n"
+	set exit_code 1
+}
+
+cancel_job $job_id
+if {$exit_code == 0} {
+	file delete $file_in $test_prog
+	send_user "\nSUCCESS\n"
+}
+exit $exit_code

testsuite/expect/test7.12.prog.c

+/*****************************************************************************\
+ *  test7.12.prog.c - Test of slurm_job_step_stat() API call.
+ *****************************************************************************
+ *  Copyright (C) 2010 Lawrence Livermore National Security.
+ *  Produced at Lawrence Livermore National Laboratory (cf, DISCLAIMER).
+ *  Written by Morris Jette <jette1@llnl.gov>
+ *  CODE-OCEC-09-009. All rights reserved.
+ *
+ *  This file is part of SLURM, a resource management program.
+ *  For details, see <https://computing.llnl.gov/linux/slurm/>.
+ *  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.
+\*****************************************************************************/
+#include <stdio.h>
+#include <stdlib.h>
+#include <slurm/slurm.h>
+#include <slurm/slurm_errno.h>
+
+main(int argc, char **argv)
+{
+	int i, rc = 0;
+	uint32_t job_id=0, step_id=0;
+	job_step_stat_response_msg_t *resp = NULL;
+	job_step_stat_t *step_stat = NULL;
+	ListIterator itr;
+
+	if (argc < 2) {
+		printf("Usage: job_id [step_id]\n");
+		exit(1);
+	}
+	job_id = atoi(argv[1]);
+	if (argc > 2)
+		step_id = atoi(argv[2]);
+	printf("job_id:%u step_id:%u\n", job_id, step_id);
+
+	rc = slurm_job_step_stat(job_id, step_id, NULL, &resp);
+	if (rc != SLURM_SUCCESS) {
+		slurm_perror("slurm_job_step_stat");
+		exit(1);
+	}
+
+	itr = slurm_list_iterator_create(resp->stats_list);
+	while ((step_stat = slurm_list_next(itr))) {
+		for (i=0; i<step_stat->step_pids->pid_cnt; i++)
+			printf("pid:%u\n", step_stat->step_pids->pid[i]);
+	}
+	slurm_list_iterator_destroy(itr);
+	slurm_job_step_pids_response_msg_free(resp);
+	exit(0);
+}