Fix local server for multiple applications
[anna.git] / example / diameter / launcher / resources / scripts / tinyTestcase.sh
index bd1c733..ba867b2 100755 (executable)
@@ -7,6 +7,7 @@ SCR_DIR=`readlink -f $0 | xargs dirname`
 TESTCASE_BN=testcase.txt
 WHAT=?????????
 REQUEST_STEP=
+VARIANT__dflt=safe
 
 #############
 # FUNCTIONS #
@@ -20,7 +21,7 @@ _exit() {
 
 usage() {
   echo
-  echo "Usage: $0 <source_directory> [seconds_timeout]"
+  echo "Usage: $0 <source_directory> <seconds_timeout> [variant: fast|[safe]]"
   echo
   echo "       source_directory:"
   echo "         The source directory may contain .xml and .metadata files grouped by pairs and alphabetically classified,"
@@ -52,8 +53,13 @@ usage() {
   echo "         messages and metadata for each pcap diameter frame."
   echo
   echo
+  echo "       variant:"
+  echo "         Two variants are implemented: 'fast', where wait conditions does not check the complete message, and"
+  echo "         'safe' which checks the full xml message content (good for function test). Default is '$VARIANT__dflt'."
+  echo
+  echo
   echo "       seconds_timeout:"
-  echo "         Timeout in seconds for the testcase. If not provided, no timeout will be programmed for the testcase."
+  echo "         Timeout in seconds for the testcase. If '-' provided, no timeout will be programmed for the testcase."
   echo
   echo
   echo
@@ -70,12 +76,59 @@ usage() {
   _exit
 }
 
+# Check number:
+isnumber() {
+  local re='^[0-9]+$'
+  [[ "$1" =~ $re ]] && return 0
+  return 1
+}
+
 # $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: code; $2: isrequest; $3: frame to start searching: direction depends on isrequest (1: downwards, 0: upwards)
+search_xml () {
+  local code=$1
+  local isrequest=$2
+  local this_frame=$3
+  local frames=()
+
+  if [ $isrequest -eq 1 ]
+  then
+    # requests shall be before frame
+    frames=( $(ls *.*metadata | cut -d\. -f1 | sort -rn | awk -v frame=$this_frame  '{ if($1 < frame) print $1 }') )
+  else
+    # answers shall be after frame
+    frames=( $(ls *.*metadata | cut -d\. -f1 | sort -n | awk -v frame=$this_frame  '{ if($1 > frame) print $1 }') )
+  fi
+
+  local frame=
+  local mtd=
+  local found=
+  local _code=
+  local _isrequest=
+  for frame in ${frames[@]}
+  do
+    mtd=${frame}.metadata
+    [ ! -f $mtd ] && _exit "Unexpected error: can't found '$mtd' file !!"
+
+    _code=$(grep ^code $mtd | cut -d= -f2)
+    _isrequest=$(grep ^isrequest $mtd | cut -d= -f2)
+
+    [ $_code -ne $code -o $_isrequest -ne $isrequest ] && continue
+
+    found="$(ls ${frame}.*.xml 2>/dev/null)"
+    [ -z "$found" ] && _exit "Unexpected error: can't found xml file for frame '${frame}' file !!"
+    echo $found
+    return 0
+  done
+
+  return 1
+}
+
 # $1: metadata file; $2: xml file; $3: check Result-Code indicator; $4: client/server (ADML node role)
 update_testcase () {
   # metadata aspect:
@@ -93,8 +146,10 @@ update_testcase () {
   local adml_type=$4
 
   local wait_command=waitfe
+  # wait xml variant:
+  local waitxml_command=waitfe-xml
   local send_command=sendxml2e
-  [ "$adml_type" = "server" ] && { wait_command=waitfc ; send_command=sendxml2c ; }
+  [ "$adml_type" = "server" ] && { wait_command=waitfc ; waitxml_command=waitfc-xml ; send_command=sendxml2c ; }
 
 
   # Ignore disconnect peer messages on testcase format (they have no session-id):
@@ -107,6 +162,11 @@ update_testcase () {
   local lines=$(wc -l $TESTCASE_BN | awk '{ print $1 }')
 
   local s_wait="test|1|$wait_command|$code|$((1-isrequest))"
+  # wait xml variant:
+  local this_frame=$(echo $xml | cut -d\. -f1)
+  local pairxml=$(search_xml $code $((1-isrequest)) $this_frame)
+  local s_waitxml="test|1|$waitxml_command|$pairxml"
+
   [ -n "$sessionid" ] && s_wait="${s_wait}|||${sessionid}"
 
   local s_send="test|1|$send_command|$xml"
@@ -114,8 +174,16 @@ update_testcase () {
   if [ $isrequest -eq 1 ]
   then
     # Send the request
-    # Special case for SNR/SNA (code=8388636): the Session-Id is created on client and received on SLR previously
-    [ "$code" = "8388636" -a "$adml_type" = "server" ] && s_send="test|1|$send_command|$xml|$REQUEST_STEP"
+    # Special case for SNR/SNA (code=8388636) and STR (code=275) going to SAPC: the Session-Id is created on client and received on SLR previously
+    if [ "$adml_type" = "server" ]
+    then
+      if [ "$code" = "8388636" -o "$code" = "275" ]
+      then
+        s_send="test|1|$send_command|$xml|$REQUEST_STEP"
+        s_wait="test|1|$wait_command|$code|0|||"
+      fi
+    fi
+
     echo "$s_send" >> $TESTCASE_BN
 
     if [ -n "$resultcode" ]
@@ -125,7 +193,7 @@ update_testcase () {
       local ans_xml=${hbh_matchs[1]}
       local rc=2001
       if [ -n "$ans_xml" ]
-      then 
+      then
         _rc=$(getResultCode $ans_xml)
         [ -n "$_rc" ] && rc=$_rc
       fi
@@ -133,14 +201,36 @@ update_testcase () {
       # Wait the answer:
       s_wait="${s_wait}|${rc}"
     fi
-    echo "$s_wait" >> $TESTCASE_BN
+
+    # Wait step:
+    if [ "$VARIANT" = "fast" ]
+    then
+      echo "$s_wait" >> $TESTCASE_BN
+    else
+      echo "$s_waitxml" >> $TESTCASE_BN
+    fi
+
   else
     local next_step_number=$((lines+1))
 
     # Wait the request
-    # Special case for SLR/SLA (code=8388635): the Session-Id is created on client
-    [ "$code" = "8388635" -a "$adml_type" = "server" ] && { s_wait="test|1|$wait_command|$code|1|||||$subscriber" ; REQUEST_STEP=$next_step_number ; }
-    echo "$s_wait" >> $TESTCASE_BN
+    # Special case for SLR/SLA (code=8388635) and STR (code=275) coming from SAPC: the Session-Id is created on client
+    if [ "$adml_type" = "server" ]
+    then
+      if [ "$code" = "8388635" -o "$code" = "275" ]
+      then
+        s_wait="test|1|$wait_command|$code|1"
+        REQUEST_STEP=$next_step_number
+      fi
+    fi
+
+    # Wait step:
+    if [ "$VARIANT" = "fast" ]
+    then
+      echo "$s_wait" >> $TESTCASE_BN
+    else
+      echo "$s_waitxml" >> $TESTCASE_BN
+    fi
 
     # Send the answer
     echo "test|1|$send_command|$xml|$next_step_number" >> $TESTCASE_BN
@@ -185,8 +275,8 @@ sort .involved_frames > .involved_frames_sort
 mv .involved_frames_sort .involved_frames
 
 # Messages classification:
-0> cers_4_starting
-0> ceas_4_establishing
+#0> cers_4_starting
+#0> ceas_4_establishing
 0> cers_4_starting_origin_hosts
 0> ceas_4_establishing_origin_hosts
 0> requests_4_sending
@@ -195,7 +285,12 @@ mv .involved_frames_sort .involved_frames
 
 # Optional timeout:
 TIMEOUT_SEC=$2
-[ -n "$TIMEOUT_SEC" ] && echo "test|1|timeout|$((TIMEOUT_SEC * 1000))" >> $TESTCASE_BN
+isnumber $TIMEOUT_SEC
+[ $? -eq 0 ] && echo "test|1|timeout|$((TIMEOUT_SEC * 1000))" >> $TESTCASE_BN
+
+# Variant:
+VARIANT=$VARIANT__dflt
+[ "$3" = "fast" ] && VARIANT=fast
 
 # Process frames:
 for frame in `cat .involved_frames`
@@ -209,7 +304,7 @@ do
   # Ignore keep alives:
   grep -q "^code=280$" $mtd
   [ $? -eq 0 ] && continue
-  
+
   grep -q "^isrequest=1$" $mtd
   if [ $? -eq 0 ]
   then
@@ -217,7 +312,7 @@ do
     grep -q "^code=257$" $mtd
     if [ $? -eq 0 ]
     then
-      echo $frame >> cers_4_starting
+      #echo $frame >> cers_4_starting
       [ -z "$originHost" ] &&  _exit "Missing Origin-Host (frame $frame, CER message) !!"
       echo "$originHost" >> cers_4_starting_origin_hosts
       ln -sf $xml cer.${originHost}.xml
@@ -225,7 +320,7 @@ do
     fi
     # Other requests:
     echo $frame >> requests_4_sending
-    
+
     # client or server:
     adml_type=$(grep -w "$originHost" $ohs_file | awk '{ print $NF }')
     update_testcase $mtd $xml check_result_code $adml_type
@@ -234,7 +329,7 @@ do
     grep -q "^code=257$" $mtd
     if [ $? -eq 0 ]
     then
-      echo $frame >> ceas_4_establishing
+      #echo $frame >> ceas_4_establishing
       originHost=$(grep ^originhost= ${frame}.metadata | cut -d= -f2-)
       [ -z "$originHost" ] &&  _exit "Missing Origin-Host (frame $frame, CEA message) !!"
       echo "$originHost" >> ceas_4_establishing_origin_hosts
@@ -283,7 +378,7 @@ do
     echo $oh >> cers_4_starting_origin_hosts
     echo "Missing CER: `basename $cer` (a basic template has been created, please edit & fix the unknowns)"
         cat << EOF > $cer
-<message version="1" name="CER" application-id="0" hop-by-hop-id="1" end-by-end-id="1">
+<message version="1" name="CER" application-id="0" hop-by-hop-id="1" end-to-end-id="1">
    <avp name="Origin-Host" data="$oh"/>
    <avp name="Origin-Realm" data="$(echo $oh | cut -d\. -f2-) $WHAT"/>
    <avp name="Auth-Application-Id" data="16777236 $WHAT 16777238 $WHAT"/>
@@ -306,7 +401,7 @@ do
     echo $oh >> ceas_4_establishing_origin_hosts
     echo "Missing CEA: `basename $cea` (a basic template has been created, please edit & fix the unknowns)"
         cat << EOF > $cea
-<message version="1" name="CEA" application-id="0" hop-by-hop-id="1" end-by-end-id="1">
+<message version="1" name="CEA" application-id="0" hop-by-hop-id="1" end-to-end-id="1">
    <avp name="Result-Code" data="2001" alias="DIAMETER_SUCCESS"/>
    <avp name="Origin-Host" data="$oh"/>
    <avp name="Origin-Realm" data="$(echo $oh | cut -d\. -f2-) $WHAT"/>