#!/usr/bin/expect -f
# BEGIN_ICS_COPYRIGHT8 ****************************************
# 
# Copyright (c) 2015-2017, Intel Corporation
# 
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
# 
#     * Redistributions of source code must retain the above copyright notice,
#       this list of conditions and the following disclaimer.
#     * Redistributions in binary form must reproduce the above copyright
#       notice, this list of conditions and the following disclaimer in the
#       documentation and/or other materials provided with the distribution.
#     * Neither the name of Intel Corporation nor the names of its contributors
#       may be used to endorse or promote products derived from this software
#       without specific prior written permission.
# 
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
# DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE
# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
# SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
# CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
# OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
# 
# END_ICS_COPYRIGHT8   ****************************************

# [ICS VERSION STRING: unknown]

#=============================================================================#
# Test Suite Description:
#-----------------------------------------------------------------------------#
## reboot
## -------
## reboot all chassis
#=============================================================================#

# manditory setup
log_user 0;	# disable detailed logging to stdout

# uncomment the following to debug the test
#strace 20
#log_user 1
#exp_internal 1
#cmd_trace on
#strace 4

set install_exit_code 0

proc suite_setup {} {
	# example of conditional setup
	#test_execute {load populate} {
		#do some installation or configuration stuff
	#}
}

proc case_setup {} {
# setup to do at start of each test case
# Usage:
#	pass as argument to start_suite
# Arguments:
#	None
# This procedure will be run at the start of every test case
# It provides an opportunity for uniform cleanup/setup for each test case

}

proc case_cleanup { failed } {
# cleanup to do after each test case
# Usage:
#	pass as argument to start_suite
# Arguments:
#	failed - set to 1 if test case failed in which case additional
#		information/files may be desired in test_tmp
#		set to 0 if test case passed
# This procedure will be run at the end of every test case
# It provides an opportunity for uniform cleanup after each test case
	global env

	test_execute {} {
		if { $failed } {
			# could put cleanup here
		}
	}
}

test_suite "upgrade" "upgrade firmware" "upgrade all chassis
Firmware Files: $env(CFG_FWFILES)
Action: $env(CFG_FWACTION)
File: /usr/lib/opa/tools/chassis_upgrade.exp" suite_setup noop {
	global env

	# Global Initializations for Test Suite:
	# --------------------------------------

	# ADD ANY GLOBAL INITIALIZATION HERE, SUCH AS:
	# set env(MY_ENV_VAR) value
	# also validate the environment
	# if the environment is bad call fail_suite "information" abort_now
	# abort_now should be 0 if it is safe to proceed and simply ignore all
	# test_case calls.  it should be 1 if that is not possible.

	# Procedures used to support Test Suite:
	# --------------------------------------

	# The actual test cases:
	# ----------------------

	if { [ test_case "check" "check firmware files" "check firmware to be pushed to all chassis
Firmware Files: $env(CFG_FWFILES)
File: /usr/lib/opa/tools/chassis_upgrade.exp" case_setup case_cleanup {
			upvar num_fw num_fw
			upvar fw_files fw_files
			upvar fw_versions fw_versions
			upvar fw_card_types fw_card_types
	
			set fw_files ""
			set fw_versions ""
			set fw_card_types ""
	
			set num_fw [llength $env(CFG_FWFILES)]
	
			puts ""
			foreach fw_file $env(CFG_FWFILES) {
				puts "$fw_file"
				set fw_version [get_file_firmware_version $fw_file]
				set card_type [get_file_firmware_type $fw_file]
				puts "\tVersion: $fw_version\tType: $card_type"
	
				lappend fw_files "$fw_file"
				lappend fw_versions "$fw_version"
				lappend fw_card_types "$card_type"
			}
		} ] } {
		fail_suite  "Invalid Firmware Files"
	}

	# We only get here if all the firmware files supplied are valid
	parallel chassis $env(CFG_CHASSIS) {
		set slots [chassis_get_slots "$chassis"]
		set chassis [chassis_strip_slots "$chassis"]

		for { set i 0 } { $i < $num_fw } { incr i } {
			set fw_file [lindex $fw_files $i]
			set fw_version [lindex $fw_versions $i]
			set card_type [lindex $fw_card_types $i]

			test_case_push_chassis_firmware "$chassis" "$fw_file" "$fw_version" "$card_type" "$env(CFG_FWACTION)" "$slots"
		}
	}
}