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 TRACE_PATH=$(mktemp -d)
  26
  27 source $TESTDIR/utils.sh
  28
  29 echo -e "\n-------------------------------"
  30 echo -e " Streaming - User space tracing "
  31 echo -e "--------------------------------"
  32
  33 if [ ! -x "$CURDIR/$BIN_NAME" ]; then
  34         echo -e "No UST nevents binary detected. Passing."
  35         exit 0
  36 fi
  37
  38 function lttng_create_session
  39 {
  40         # Create session with default path
  41         $TESTDIR/../src/bin/lttng/$LTTNG_BIN create $SESSION_NAME >/dev/null 2>&1
  42 }
  43
  44 function lttng_enable_consumer
  45 {
  46         # Create session with default path
  47         $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-consumer -u net://localhost >/dev/null 2>&1
  48 }
  49
  50 function wait_apps
  51 {
  52         echo -n "Waiting for applications to end"
  53         while [ -n "$(pidof $BIN_NAME)" ]; do
  54                 echo -n "."
  55                 sleep 0.5
  56         done
  57         echo ""
  58 }
  59
  60 # MUST set TESTDIR before calling those functions
  61
  62 function test_ust_before_start ()
  63 {
  64         echo -e "\n=== Testing UST streaming BEFORE tracing starts\n"
  65         lttng_create_session
  66         lttng_enable_consumer
  67         enable_ust_lttng_event $SESSION_NAME $EVENT_NAME
  68
  69         # Run 5 times with a 1 second delay
  70         ./$CURDIR/$BIN_NAME 5 1000000 >/dev/null 2>&1 &
  71
  72         start_tracing $SESSION_NAME
  73
  74         wait_apps
  75         stop_tracing $SESSION_NAME
  76         destroy_lttng_session $SESSION_NAME
  77 }
  78
  79 function test_ust_after_start ()
  80 {
  81         echo -e "\n=== Testing UST streaming AFTER tracing starts\n"
  82         lttng_create_session
  83         lttng_enable_consumer
  84         enable_ust_lttng_event $SESSION_NAME $EVENT_NAME
  85         start_tracing $SESSION_NAME
  86
  87         # Run 5 times with a 1 second delay
  88         ./$CURDIR/$BIN_NAME 5 1000000 >/dev/null 2>&1 &
  89
  90         wait_apps
  91         stop_tracing $SESSION_NAME
  92         destroy_lttng_session $SESSION_NAME
  93 }
  94
  95 start_sessiond
  96 lttng_start_relayd "-o $TRACE_PATH"
  97
  98 tests=( test_ust_before_start test_ust_after_start )
  99
 100 for fct_test in ${tests[@]};
 101 do
 102         SESSION_NAME=$(randstring 16 0)
 103         ${fct_test}
 104
 105         # Validate test
 106         validate_trace $EVENT_NAME $TRACE_PATH/$HOSTNAME/$SESSION_NAME*
 107         if [ $? -eq 0 ]; then
 108                 # Only delete if successful
 109                 rm -rf $TRACE_PATH
 110         else
 111                 break
 112         fi
 113 done
 114
 115 echo ""
 116 stop_sessiond
 117 lttng_stop_relayd
 118
 119 exit $out