diff --git a/crm/test/helper.sh.in b/crm/test/helper.sh.in index 2eee74c738..ad7f9a3ba4 100644 --- a/crm/test/helper.sh.in +++ b/crm/test/helper.sh.in @@ -1,468 +1,468 @@ #!/bin/bash # # Copyright (C) 2004 Andrew Beekhof # # This program 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. # # This program 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 this program; if not, write to the Free Software # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. # [[ $0 != *helper.sh ]] || { echo >&2 "helper.sh expects to be sourced!" exit 1 } # no more exit below here, use return instead; gives more flexibility ... test_type=`basename $0` test_nodes=1 dump_dir=~/BUILD/tests HALIB_DIR=@libdir@/heartbeat HAVAR_DIR=@HA_VARLIBDIR@/heartbeat HAINIT_DIR=@INITDIR@ INIT_USER=root CRMD_USER=root ADMIN_USER=root test_node_1=fish test_node_2=chips test_node_3=mayo CRM_OPTS="-VV" CRM_ERR_SHUTDOWN=0 OUTPUT_STDERR=0 OUTPUT_STDOUT=0 OUTPUT_CMDS=0 OUTPUT_ECHO=1 repeats=1 iteration=0 done=0 ip_rsc_1=192.168.9.161 ip_rsc_2=192.168.9.162 logfile=/var/log/messages if [ ! -z "$CRM_LOCAL_OPTS" -a -f "$CRM_LOCAL_OPTS" ]; then . "$CRM_LOCAL_OPTS" fi while test "$done" = "0"; do case "$1" in --debug | -x) set -x; shift;; --silent | -s) OUTPUT_ECHO=0; shift;; --stdout | -o) OUTPUT_STDOUT=1; shift;; --stderr | -e) OUTPUT_STDERR=1; shift;; --verbose | -v) OUTPUT_CMDS=1; shift;; --logfile | -l) logfile=${2?no logfile??} shift 2;; --dumpdir | -d) dump_dir=${2?no dumpdir??} shift 2;; --repeat | -r) repeats=${2?no max repetitions??} || return 1; shift 2;; --init | -i) INIT_USER=${2?no init user??} || return 1; shift 2;; --admin | -a) ADMIN_USER=${2?no admin??} || return 1; shift 2;; --crmd | -c) CRMD_USER=${2?no crmd user??} || return 1; shift 2;; --node1 | -1) test_node_1=${2?no node1??} || return 1; shift 2;; --node2 | -2) test_node_2=${2?no node2??} || return 1; shift 2;; --node3 | -3) test_node_3=${2?no node2??} || return 1; shift 2;; --ip1) ip_rsc_1=${2?no ip1??} || return 1; shift 2;; --ip2) ip_rsc_2=${2?no ip2??} || return 1; shift 2;; ""|--) done=1;; *) echo "unknown option: $1"; return 1;; esac done function test_init() { export date_str=`date "+%Y-%b-%d_%H:%M"` export test_dump_dir="$dump_dir/$test_type/$date_str" mkdir -p "$test_dump_dir" echo Test dir: $test_dump_dir } function remote_cmd() { user=$1 host=$2 shift shift `which ssh` ${user}@${host} "$*" return $?; } function is_state() { ret=1 host=$1 state=$2 output=`remote_cmd $ADMIN_USER $host $HALIB_DIR/crmadmin -S $host | grep $state` if [ ! -z "$output" ]; then echo "$host is in $state" return 0 fi echo "$host is NOT in $state" return $ret } function is_running() { rsc=$1 askhost=$2 host=$3 if [ -z $host ]; then output=`remote_cmd $ADMIN_USER $askhost $HALIB_DIR/crmadmin -W $rsc | grep -v $HALIB_DIR/crmadmin` output=`echo $output | grep -v NOT` else output=`remote_cmd $ADMIN_USER $askhost $HALIB_DIR/crmadmin -W $rsc | grep -v $HALIB_DIR/crmadmin` output=`echo $output | grep $host` fi if [ -z "$output" ]; then if [ -z $host ]; then echo "$rsc is NOT running" else echo "$rsc is NOT running on host $host" fi return 1 fi if [ -z $host ]; then echo "$rsc is running" else echo "$rsc is running on host $host" fi return 0 } function is_dc() { host=$1 output=`remote_cmd $ADMIN_USER $host $HALIB_DIR/crmadmin -S $host | grep -v S_PENDING | grep -v S_NOT_DC | grep -v S_ELECTION | grep -v S_RECOVERY` if [ ! -z "$output" ]; then echo "$host is DC" return 0 fi echo "$host is NOT DC" remote_cmd $ADMIN_USER $host $HALIB_DIR/crmadmin -S $host return 1 } function wait_for_state() { state=$1 max=$2 host=$3 target=$4 sleep 1 if [ "$host" = "DC" ]; then host=`remote_cmd $ADMIN_USER $target $HALIB_DIR/crmadmin -D` host=${host#*: } fi count=1 until is_state $host $state; do if [ $count -gt $max ]; then echo "Attempt to reach $state on $host failed" return 1 fi count=`expr $count + 1` sleep 1 done echo "$state reached on $host" return 0 } function make_node() { host=$1 uname=$2 type=$3 uuid=`uuidgen` if [ -z $type ]; then type="member" fi node_xml="''"; remote_cmd $ADMIN_USER $host "$HALIB_DIR/cibadmin -C -o node -X $node_xml" } function make_resource() { host=$1 id=$2 class=$3 type=$4 priority=$5 version=$6 stopfail=$7 shift 7 args=$* uuid=`uuidgen` node_xml="'&1 > /dev/null } function make_constraint_adv() { host=$1 shift node_xml=$* node_xml=`echo $node_xml | tr '\n' ' '` do_cmd $QUIET remote_cmd $ADMIN_USER $host "$HALIB_DIR/cibadmin -C -o constraint -X $node_xml" 2>&1 > /dev/null } function make_constraint() { host=$1 rsc=$2 result=$3 uuid1=`uuidgen` uuid2=`uuidgen` node_xml="''" make_constraint_adv $host $node_xml } function cts_assert() { do_cts_assert $? 0 "Assert failed - " $* } function cts_assert_false() { do_cts_assert $? 1 "AssertFalse failed - " $* } function do_cts_assert() { result=$1 expected=$2 shift shift if [ $result -ne $expected ]; then err $* fi } function err() { if [ ! -z $HOST ]; then if [ $CRM_ERR_SHUTDOWN -eq 1 ]; then remote_cmd $ADMIN_USER $test_node_1 killall -9 crmd remote_cmd $ADMIN_USER $test_node_2 killall -9 crmd fi fi do_cmd echo Dumping test nodes to $dump_dir dump_node 1 $test_node_1 dump_node 2 $test_node_2 dump_node 3 $test_node_3 do_cmd echo "ERROR: $*" echo "test $test_type: FAILED" echo "ERROR: $*" > $test_dump_dir/error.txt exit 1 } function dump_node() { a_node_index=$1 a_node=$2 dump_file="${a_node_index}-${a_node}.tgz" `which ssh` ${ADMIN_USER}@${a_node} "ps axf > /tmp/crm/processes" `which ssh` ${ADMIN_USER}@${a_node} \ "tar zcvf /tmp/${dump_file} /tmp/crm ~${CRMD_USER}/core" scp -r ${ADMIN_USER}@${a_node}:/tmp/${dump_file} $test_dump_dir } function do_cmd() { cmdline=$* old_OUTPUT_STDOUT=$OUTPUT_STDOUT cmd=$1 if [ $OUTPUT_STDERR -eq 1 -o $OUTPUT_STDOUT -eq 1 -o $OUTPUT_CMDS -eq 1 ]; then if [ $cmd = "remote_cmd" ]; then user=$2 host=$3 shift 3 echo `date` ": Running '$*' as ${user}@${host}" | tee -a $test_dump_dir/test.txt; elif [ $cmd = "wait_for_state" ]; then host=$4 if [ -z $host ]; then host="the DC" fi echo `date` ": Waiting for state $2 on $host (max $3 attempts)" | tee -a $test_dump_dir/test.txt; elif [ $cmd = "${HALIB_DIR}/crmtest/testutils.pl" ]; then echo `date` ": Searching '$cmdline'" | tee -a $test_dump_dir/test.txt; elif [ $cmd = "echo" ]; then echo -n "" # if [ $OUTPUT_CMDS -eq 1 -a $OUTPUT_STDOUT -eq 0 -a $OUTPUT_STDERR -eq 0 ]; then # echo -n `date` ": " # $cmdline # elif [ $OUTPUT_ECHO -eq 1 -a $OUTPUT_CMDS -eq 0 -a $OUTPUT_STDOUT -eq 0 -a $OUTPUT_STDERR -eq 0 ]; then # $cmdline # fi else echo `date` ": Running '$cmdline'" | tee -a $test_dump_dir/test.txt; fi fi if [ $cmd = "echo" ]; then if [ $OUTPUT_CMDS -eq 1 -a $OUTPUT_STDOUT -eq 0 -a $OUTPUT_STDERR -eq 0 ]; then echo -n `date` ": " | tee -a $test_dump_dir/test.txt echo -n -e "\033[01;33m" $cmdline | tee -a $test_dump_dir/test.txt echo -n -e "\033[00m" elif [ $OUTPUT_ECHO -eq 1 -a $OUTPUT_CMDS -eq 0 -a $OUTPUT_STDOUT -eq 0 -a $OUTPUT_STDERR -eq 0 ]; then echo -n `date` ": " | tee -a $test_dump_dir/test.txt echo -n -e "\033[01;33m" $cmdline | tee -a $test_dump_dir/test.txt echo -n -e "\033[00m" fi elif [ $OUTPUT_STDERR -eq 1 -a $OUTPUT_STDOUT -eq 1 ]; then $cmdline 2>&1 >> $test_dump_dir/test.txt rc=$? elif [ $OUTPUT_STDERR -eq 0 -a $OUTPUT_STDOUT -eq 0 ]; then $cmdline 2>&1 >/dev/null rc=$? elif [ $OUTPUT_STDOUT -eq 0 ]; then $cmdline >/dev/null 2>> $test_dump_dir/test.txt rc=$? elif [ $OUTPUT_STDERR -eq 0 ]; then $cmdline 2>/dev/null >> $test_dump_dir/test.txt rc=$? else $cmdline 2>&1 >> $test_dump_dir/test.txt rc=$? fi if [ $cmd = "${HALIB_DIR}/crmtest/testutils.pl" ]; then # for the logs... if [ $OUTPUT_STDERR -eq 0 -a $OUTPUT_STDOUT -eq 0 -a $OUTPUT_CMDS -eq 0 ]; then echo `date` ": Searching '$cmdline'" >> $test_dump_dir/test.txt; fi $cmdline -v 2>&1 >> $test_dump_dir/test.txt if [ $rc -lt 0 ]; then echo `date` ": Searching '$cmdline'" $cmdline fi fi OUTPUT_STDOUT=$old_OUTPUT_STDOUT # echo Result: $rc return $rc } function crm-cleanup() { if [ -z $1 ]; then test_init fi do_cmd echo Cleaning up on $test_node_1, $test_node_2 and $test_node_3 crm-cleanup-node $test_node_1 $1 crm-cleanup-node $test_node_2 $1 crm-cleanup-node $test_node_3 $1 } function crm-cleanup-node() { ha_host=$1 do_cmd echo Cleaning up on $ha_host # stop all running HAs do_cmd remote_cmd $INIT_USER $ha_host $HALIB_DIR/heartbeat "-k" "2>&1 >/dev/null" # be *very* sure everything has stopped do_cmd remote_cmd $INIT_USER $ha_host "killall -q9 heartbeat ccm lrmd crmd" # resources too do_cmd remote_cmd $INIT_USER $ha_host "/etc/ha.d/resource.d/IPaddr $ip_rsc_1 stop" 2>&1 > /dev/null do_cmd remote_cmd $INIT_USER $ha_host "/etc/ha.d/resource.d/IPaddr $ip_rsc_2 stop" 2>&1 > /dev/null if [ -z $2 ]; then # make *sure* theres nothing left over from last time do_cmd remote_cmd $INIT_USER $ha_host "rm -f $HAVAR_DIR/crm/cib*.xml" remote_cmd $INIT_USER $ha_host "rm -f /tmp/crm/*" 2>&1 > /dev/null - do_cmd remote_cmd $CRMD_USER $ha_host "touch /tmp/crm/cluster.log 2>&1 > /dev/null" - do_cmd remote_cmd $CRMD_USER $ha_host "rm -f ~/core 2>&1 > /dev/null" - do_cmd remote_cmd $INIT_USER $ha_host "/etc/init.d/syslog restart 2>&1 > /dev/null" + remote_cmd $CRMD_USER $ha_host "touch /tmp/crm/cluster.log" + remote_cmd $CRMD_USER $ha_host "rm -f /tmp/crm/*/core" 2>&1 > /dev/null + do_cmd remote_cmd $INIT_USER $ha_host "/etc/init.d/syslog restart" # zero out logs, maybe if [ ${test_node_1} = "fish" ]; then do_cmd remote_cmd $INIT_USER $ha_host "cat /dev/null > /var/log/messages" fi fi } function mark_log() { export crm_log_pos=$(stat -L -c %s $logfile) } # successfully sourced true