tests/tools/streaming/run-ust

   1 #!/bin/bash
   2 #
   3 # Copyright (C) - 2012 David Goulet <dgoulet@efficios.com>
   4 #
   5 # This library is free software; you can redistribute it and/or modify it under
   6 # the terms of the GNU Lesser General Public License as published by the Free
   7 # Software Foundation; version 2.1 of the License.
   8 #
   9 # This library is distributed in the hope that it will be useful, but WITHOUT
  10 # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  11 # FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more
  12 # details.
  13 #
  14 # You should have received a copy of the GNU Lesser General Public License
  15 # along with this library; if not, write to the Free Software Foundation, Inc.,
  16 # 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA
  17
  18 CURDIR=$(dirname $0)/
  19 TESTDIR=$CURDIR/../..
  20 BIN_NAME="gen-ust-events"
  21 SESSION_NAME="stream"
  22 EVENT_NAME="tp:tptest"
  23 PID_RELAYD=0
  24
  25 source $TESTDIR/utils.sh
  26
  27 echo -e "\n-------------------------------"
  28 echo -e " Streaming - User space tracing "
  29 echo -e "--------------------------------"
  30
  31 if [ ! -x "$CURDIR/$BIN_NAME" ]; then
  32         echo -e "No UST nevents binary detected. Passing."
  33         exit 0
  34 fi
  35
  36 function lttng_create_session
  37 {
  38         # Create session with default path
  39         $TESTDIR/../src/bin/lttng/$LTTNG_BIN create $SESSION_NAME >/dev/null 2>&1
  40 }
  41
  42 function lttng_enable_consumer
  43 {
  44         # Create session with default path
  45         $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-consumer -u net://localhost >/dev/null 2>&1
  46 }
  47
  48 function wait_apps
  49 {
  50         echo -n "Waiting for applications to end"
  51         while [ -n "$(pidof $BIN_NAME)" ]; do
  52                 echo -n "."
  53                 sleep 0.5
  54         done
  55         echo ""
  56 }
  57
  58 # MUST set TESTDIR before calling those functions
  59
  60 function test_ust_before_start ()
  61 {
  62         echo -e "\n=== Testing UST streaming BEFORE tracing starts\n"
  63         lttng_create_session
  64         lttng_enable_consumer
  65         enable_ust_lttng_event $SESSION_NAME $EVENT_NAME
  66
  67         # Run 5 times with a 1 second delay
  68         ./$CURDIR/$BIN_NAME 5 1000000 >/dev/null 2>&1 &
  69
  70         start_tracing $SESSION_NAME
  71
  72         wait_apps
  73         stop_tracing $SESSION_NAME
  74         destroy_lttng_session $SESSION_NAME
  75 }
  76
  77 function test_ust_after_start ()
  78 {
  79         echo -e "\n=== Testing UST streaming AFTER tracing starts\n"
  80         lttng_create_session
  81         lttng_enable_consumer
  82         enable_ust_lttng_event $SESSION_NAME $EVENT_NAME
  83         start_tracing $SESSION_NAME
  84
  85         # Run 5 times with a 1 second delay
  86         ./$CURDIR/$BIN_NAME 5 1000000 >/dev/null 2>&1 &
  87
  88         wait_apps
  89         stop_tracing $SESSION_NAME
  90         destroy_lttng_session $SESSION_NAME
  91 }
  92
  93 start_sessiond
  94 lttng_start_relayd
  95
  96 tests=( test_ust_before_start test_ust_after_start )
  97
  98 for fct_test in ${tests[@]};
  99 do
 100         SESSION_NAME=$(randstring 16 0)
 101         ${fct_test}
 102
 103         # Validate test
 104         validate_trace $EVENT_NAME ~/lttng-traces/$HOSTNAME/$SESSION_NAME*
 105         if [ $? -eq 0 ]; then
 106                 # Only delete if successful
 107                 rm -rf ~/lttng-traces/$HOSTNAME/$SESSION_NAME*
 108                 rm -rf ~/lttng-traces/$SESSION_NAME*
 109         else
 110                 break
 111         fi
 112 done
 113
 114 echo ""
 115 stop_sessiond
 116 lttng_stop_relayd
 117
 118 exit $out