From 3a4b05af8367a39b7505b6ffe7a7b71ccea5f1cb Mon Sep 17 00:00:00 2001
From: =?utf8?q?Zden=C4=9Bk=20=C5=A0ustr?= <sustr4@cesnet.cz>
Date: Wed, 20 Jan 2010 12:56:59 +0000
Subject: [PATCH] Test the HTTPS interface of an LB server. Initial (but fully
 functional) version.

---
 org.glite.testsuites.ctb/LB/tests/lb-common.sh     |   1 +
 org.glite.testsuites.ctb/LB/tests/lb-test-https.sh | 194 +++++++++++++++++++++
 2 files changed, 195 insertions(+)
 create mode 100755 org.glite.testsuites.ctb/LB/tests/lb-test-https.sh

diff --git a/org.glite.testsuites.ctb/LB/tests/lb-common.sh b/org.glite.testsuites.ctb/LB/tests/lb-common.sh
index 4ba0623..cd7fa6f 100755
--- a/org.glite.testsuites.ctb/LB/tests/lb-common.sh
+++ b/org.glite.testsuites.ctb/LB/tests/lb-common.sh
@@ -75,6 +75,7 @@ SYS_PING=ping
 SYS_AWK=awk
 SYS_ECHO=echo
 SYS_DOMAINNAME=domainname
+SYS_CURL=curl
 
 # not used at the moment
 DEBUG=2
diff --git a/org.glite.testsuites.ctb/LB/tests/lb-test-https.sh b/org.glite.testsuites.ctb/LB/tests/lb-test-https.sh
new file mode 100755
index 0000000..d4160b6
--- /dev/null
+++ b/org.glite.testsuites.ctb/LB/tests/lb-test-https.sh
@@ -0,0 +1,194 @@
+#!/bin/bash
+
+# show help and usage
+progname=`basename $0`
+showHelp()
+{
+cat << EndHelpHeader
+Script for testing correct job registration
+
+Prerequisities:
+   - LB server
+   - environment variables set:
+
+     GLITE_LB_SERVER_PORT - if nondefault port (9000) is used
+
+Tests called:
+
+    job registration
+
+Returned values:
+    Exit TEST_OK: Test Passed
+    Exit TEST_ERROR: Test Failed
+    Exit 2: Wrong Input
+
+EndHelpHeader
+
+	echo "Usage: $progname [OPTIONS]"
+	echo "Options:"
+	echo " -h | --help            Show this help message."
+	echo " -o | --output 'file'   Redirect all output to the 'file' (stdout by default)."
+	echo " -t | --text            Format output as plain ASCII text."
+	echo " -c | --color           Format output as text with ANSI colours (autodetected by default)."
+	echo " -x | --html            Format output as html."
+}
+
+# read common definitions and functions
+COMMON=lb-common.sh
+if [ ! -r ${COMMON} ]; then
+	printf "Common definitions '${COMMON}' missing!"
+	exit 2
+fi
+source ${COMMON}
+
+logfile=$$.tmp
+flag=0
+while test -n "$1"
+do
+	case "$1" in
+		"-h" | "--help") showHelp && exit 2 ;;
+		"-o" | "--output") shift ; logfile=$1 flag=1 ;;
+		"-t" | "--text")  setOutputASCII ;;
+		"-c" | "--color") setOutputColor ;;
+		"-x" | "--html")  setOutputHTML ;;
+	esac
+	shift
+done
+
+# redirecting all output to $logfile
+touch $logfile
+if [ ! -w $logfile ]; then
+	echo "Cannot write to output file $logfile"
+	exit $TEST_ERROR
+fi
+
+DEBUG=2
+
+##
+#  Starting the test
+#####################
+
+{
+test_start
+
+
+# check_binaries
+printf "Testing if all binaries are available"
+check_binaries $GRIDPROXYINFO $SYS_GREP $SYS_SED $SYS_AWK $SYS_CURL
+if [ $? -gt 0 ]; then
+	test_failed
+else
+	test_done
+fi
+
+printf "Testing credentials"
+
+timeleft=`${GRIDPROXYINFO} | ${SYS_GREP} -E "^timeleft" | ${SYS_SED} "s/timeleft\s*:\s//"`
+
+if [ "$timeleft" = "" ]; then
+        test_failed
+        print_error "No credentials"
+else
+        if [ "$timeleft" = "0:00:00" ]; then
+                test_failed
+                print_error "Credentials expired"
+        else
+                test_done
+
+		# Get path to the proxy cert
+		printf "Getting proxy cert path... "
+
+		PROXYCERT=`${GRIDPROXYINFO} | ${SYS_GREP} -E "^path" | ${SYS_SED} "s/path\s*:\s//"`
+
+	        if [ "$PROXYCERT" = "" ]; then
+        	        test_failed
+                	print_error "Unable to identify the path to your proxy certificate"
+	        else
+			printf "$PROXYCERT"
+        	        test_done
+	
+			# Register job:
+			printf "Registering testing job "
+			jobid=`${LBJOBREG} -m ${GLITE_WMS_QUERY_SERVER} -s application | $SYS_GREP "new jobid" | ${SYS_AWK} '{ print $3 }'`
+
+			if [ -z $jobid  ]; then
+				test_failed
+				print_error "Failed to register job"
+			else
+				test_done
+
+				# Get list of jobs
+				printf "Evaluating job list... "
+
+				$SYS_CURL -3 --silent --key $PROXYCERT --cert $PROXYCERT --capath /etc/grid-security/certificates --output https.$$.tmp https://${GLITE_WMS_QUERY_SERVER}/
+
+				if [ "$?" != "0" ]; then
+					test_failed
+					print_error "Job list not returned"
+				else
+					test_done
+
+					printf "Looking up the test job..."
+
+					$SYS_GREP $jobid https.$$.tmp > /dev/null 2> /dev/null
+
+					if [ "$?" != "0" ]; then
+						test_failed
+						print_error "Test job not found in the list"
+					else
+						test_done
+					fi
+
+					rm https.$$.tmp
+
+				fi
+
+				# Get job status
+				printf "Evaluating job status listing... "
+
+				$SYS_CURL -3 --silent --key $PROXYCERT --cert $PROXYCERT --capath /etc/grid-security/certificates --output https.$$.tmp "${jobid}"
+
+				if [ "$?" != "0" ]; then
+					test_failed
+					print_error "Job status not returned"
+				else
+					test_done
+
+					printf "Checking for jobid (verifying content)..."
+
+					$SYS_GREP $jobid https.$$.tmp > /dev/null 2> /dev/null
+
+					if [ "$?" != "0" ]; then
+						test_failed
+						print_error "JobID not found among data returned"
+					else
+						test_done
+					fi
+
+					rm https.$$.tmp
+
+				fi
+
+				#Purge test job
+				joblist=$$_jobs_to_purge.txt
+				echo $jobid > ${joblist}
+				try_purge ${joblist}
+
+			fi
+
+
+		fi
+		
+		
+	fi
+fi
+
+test_end
+} &> $logfile
+
+if [ $flag -ne 1 ]; then
+ 	cat $logfile
+ 	rm $logfile
+fi
+exit $TEST_OK
+
-- 
1.8.2.3