#!/usr/bin/expect
############################################################################
# Purpose: Test of POE functionality
#          Test of Network protocol being set correctly. 
#          (-msg_api)
#          
#
# 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.
#
############################################################################
# Copyright (C) 2002-2006 The Regents of the University of California.
# Produced at Lawrence Livermore National Laboratory (cf, DISCLAIMER).
# Written by Danny Auble <da@llnl.gov>
# UCRL-CODE-217948.
# 
# This file is part of SLURM, a resource management program.
# For details, see <http://www.llnl.gov/linux/slurm/>.
#  
# 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     "11.3"
set env(SLURM_LL_API_DEBUG) "2"
set exit_code   0
set job_id      0
set proto_count 0
set protos      0
print_header $test_id

if { ![file exists $poe] } {
	send_user "WARNING: poe must be installed on the\
	    system to run this test.\n"
	exit $exit_code
}

if { [string length $partition] == 0 } {
	set partition [default_partition]
}

set protocols { "mpi" "lapi" "mpi_lapi" }
    
foreach proto $protocols {
	#
	# execute poe with a specific node count
	#
	incr protos
	set found 0
	spawn $poe hostname -rmpool $partition -msg_api $proto\
	    -euilib us -retry wait
	expect {
		-re "LL_AdapterUsageProtocol = ($alpha_numeric_under)" {
			if {![string compare $proto $expect_out(1,string)]} {
				if { !$found } {
					incr proto_count
					incr found
				}
			} 
			exp_continue
		}
		-re "slurm job ($number)" {
			set job_id $expect_out(1,string)
			exp_continue
		}
		timeout {
			send_user "\nFAILURE: poe not responding\n"
			exec $scancel --quiet $job_id
			set exit_code 1
		}
		eof {
		}
	}
}

if { $proto_count != $protos || !$proto_count } {
	send_user "\nFAILURE: not all protocols worked got $proto_count looking for $protos\n"
	set exit_code 1
}
	
if {$exit_code == 0} {
	send_user "\nSUCCESS\n"
}
exit $exit_code
