Improvs.
[anna.git] / example / diameter / launcher / resources / scripts / tinyTestcase.sh
index f2b30d5..330d08e 100755 (executable)
@@ -49,8 +49,9 @@ usage() {
   echo
   echo "         This script will create a basic testcase based on frames timeline (11, 13, 15, etc.), from the desired"
   echo "         end-point list. For example, from the 'gentraf' point of view, outgoing requests will be translated as"
-  echo "         sendxml operations with corresponding waits for answers and successful result code (2001). CER/A are"
-  echo "         detected creating a symbolic link to better reference them, and DPR/A are ignored."
+  echo "         sendxml operations with corresponding waits for answers and corresponding result codes (normally 2001)"
+  echo "         if those answers are provided (2001 will be assigned as result-code condition if no answer is found)."
+  echo "         CER/A are detected creating a symbolic link to better reference them, and DPR/A are ignored."
   echo
   echo "         The provided test end point list shall match metadata 'src' field in order to filter the valid testcase"
   echo "         messages. In case of nothing provided (empty string) a list with the possible values from the source"
@@ -59,7 +60,7 @@ usage() {
   echo
   echo "       timeout:"
   echo
-  echo "         Timeout in seconds for the testcase. By default, no timeout will be programmed for the testcase."
+  echo "         Timeout in seconds for the testcase. If not provided, no timeout will be programmed for the testcase."
   echo
   echo
   echo
@@ -74,8 +75,19 @@ usage() {
   echo "       will be detected and symbolically linked with the name 'cer.<origin host>.xml'."
   _exit
 }
+
+# $1: xml file
+getOriginHost () {
+  grep "<avp name=\"Origin-Host\"" $1 | cut -d\" -f4
+}
+# $1: xml file
+# Nested Result-Code's not supported here (i.e. charging application)
+getResultCode () {
+  grep "<avp name=\"Result-Code\"" $1 | awk -F'data=' '{ print $2 }' | cut -d\" -f2
+}
  
-# $1: frame number (there must be .metadata and .xml files); $2: check Result-Code indicator
+# $1: metadata file; $2: xml file; $3: check Result-Code indicator
 update_testcase () {
   # metadata aspect:
   #
@@ -86,27 +98,38 @@ update_testcase () {
   # code=258
   # isrequest=0
   # applicationid=16777238
-  mtd=${1}.*metadata
-  xml=( `ls ${1}.*xml 2>/dev/null` )
-  [ ${#xml[@]} -ne 1 ] && _exit "There must be one xml message file corresponding to '$mtd' !!"
-  resultcode=$2
+  local mtd=$1
+  local xml=$2
+  local resultcode=$3
 
   # Ignore disconnect peer messages on testcase format (they have no session-id):
   grep "^code=282$" $mtd > /dev/null
   [ $? -eq 0 ] && return
 
-  code=$(grep ^code $mtd | cut -d= -f2)
-  isrequest=$(grep ^isrequest $mtd | cut -d= -f2)
-  sessionid=$(grep Session-Id $xml | cut -d\" -f4)
-  lines=$(wc -l $TESTCASE_BN | awk '{ print $1 }')
+  local code=$(grep ^code $mtd | cut -d= -f2)
+  local isrequest=$(grep ^isrequest $mtd | cut -d= -f2)
+  local sessionid=$(grep Session-Id $xml | cut -d\" -f4)
+  local lines=$(wc -l $TESTCASE_BN | awk '{ print $1 }')
 
-  s_waitfe="test|1|waitfe|$code|$((1-isrequest))"
+  local s_waitfe="test|1|waitfe|$code|$((1-isrequest))"
   [ -n "$sessionid" ] && s_waitfe="${s_waitfe}|||${sessionid}"
 
   if [ $isrequest -eq 1 ]
   then
     echo "test|1|sendxml2e|$xml" >> $TESTCASE_BN
-    [ -n "$resultcode" ] && s_waitfe="${s_waitfe}|2001"
+    if [ -n "$resultcode" ]
+    then
+      local hbh="$(grep -o "hop-by-hop-id=\"[0-9]*\"" $xml)"
+      local hbh_matchs=( $(grep -l "$hbh" *.xml) )
+      local ans_xml=${hbh_matchs[1]}
+      local rc=2001
+      if [ -n "$ans_xml" ]
+      then 
+        _rc=$(getResultCode $ans_xml)
+        [ -n "$_rc" ] && rc=$_rc
+      fi
+      s_waitfe="${s_waitfe}|${rc}"
+    fi
     echo "$s_waitfe" >> $TESTCASE_BN
   else
     echo "$s_waitfe" >> $TESTCASE_BN
@@ -119,6 +142,7 @@ update_testcase () {
 #############
 echo
 SOURCE_DIR=$1
+[ -z "$SOURCE_DIR" ] && usage
 SOURCE_DIR=`readlink -f $SOURCE_DIR`
 [ ! -d "$SOURCE_DIR" ] && _exit "Can't found provided directory '$SOURCE_DIR'"
 # Work on source directory:
@@ -134,10 +158,16 @@ if [ -z "$END_POINTS" ]
 then
   src_eps=( $(grep "^src=" *.*metadata | cut -d= -f2 | sort -u) )
   [ ${#src_eps[@]} -eq 0 ] && _exit "No source end-points detected: metadata must have an 'src=xxx' line."
-  echo "Input a space-separated list for desired end-points to be processed as test side: "
-  echo " (available source end-points: ${src_eps[*]})"
-  read END_POINTS
-  [ -z "$END_POINTS" ] && _exit "Invalid empty input !!"
+  if [ ${#src_eps[@]} -eq 1 ]
+  then
+    END_POINTS=$src_eps
+    echo "Detected a unique end-point: $END_POINTS"
+  else
+    echo "Input a space-separated list for desired end-points to be processed as test side: "
+    echo " (available source end-points: ${src_eps[*]})"
+    read END_POINTS
+    [ -z "$END_POINTS" ] && _exit "Invalid empty input !!"
+  fi
 fi
 
 # Identify useful frames: those which are created at test-bed side (all except frames coming from tested systems):
@@ -155,6 +185,7 @@ ls *.needed >/dev/null 2>/dev/null
 
 # Messages classification:
 0> cers_4_starting
+0> cers_4_starting_origin_hosts
 0> requests_4_sending
 0> answers_4_programming
 0> $TESTCASE_BN
@@ -166,26 +197,36 @@ TIMEOUT_SEC=$3
 # Process frames:
 for frame in `ls *.needed | cut -d\. -f1 | sort -n`
 do
-  file=( `ls ${frame}.*metadata 2>/dev/null` )
-  [ ${#file[@]} -ne 1 ] && _exit "There must be one metadata file corresponding to frame '$frame' !!"
-  
+  mtd=( `ls ${frame}.*metadata 2>/dev/null` )
+  [ ${#mtd[@]} -ne 1 ] && _exit "There must be one metadata file corresponding to frame '$frame' !!"
+  xml=( `ls ${frame}.*xml 2>/dev/null` )
+  [ ${#xml[@]} -ne 1 ] && _exit "There must be one xml message file corresponding to '$mtd' (frame '$frame') !!"
+
   # Ignore keep alives:
-  grep -q "^code=280$" $file
+  grep -q "^code=280$" $mtd
   [ $? -eq 0 ] && continue
   
-  grep -q "^isrequest=1$" $file
+  grep -q "^isrequest=1$" $mtd
   if [ $? -eq 0 ]
   then
-    grep -q "^code=257$" $file
-    [ $? -eq 0 ] && { echo $frame >> cers_4_starting ; continue ; }
+    grep -q "^code=257$" $mtd
+    if [ $? -eq 0 ]
+    then
+      echo $frame >> cers_4_starting
+      originHost=$(getOriginHost $xml)
+      [ -z "$originHost" ] &&  _exit "Missing Origin-Host (frame $frame, CER message) !!"
+      echo "$originHost" >> cers_4_starting_origin_hosts
+      ln -sf $xml cer.${originHost}.xml
+      continue
+    fi
     echo $frame >> requests_4_sending
-    update_testcase $frame check_result_code
+    update_testcase $mtd $xml check_result_code
   else
     # Ignore CEA's:
-    grep -q "^code=257$" $file
+    grep -q "^code=257$" $mtd
     [ $? -eq 0 ] && continue
     echo $frame >> answers_4_programming
-    update_testcase $frame
+    update_testcase $mtd $xml
   fi
 done
 rm -f *.needed
@@ -211,3 +252,6 @@ do
   count=$((count+1))
 done
 
+echo
+exit 0
+