Add UST test for n events using validation

[lttng-tools.git] / tests / utils.sh

diff --git a/tests/utils.sh b/tests/utils.sh
new file mode 100644 (file)
index 0000000..352a198
--- /dev/null
+++ b/tests/utils.sh
@@ -0,0 +1,139 @@
+#!/bin/bash
+#
+# Copyright (C) - 2012 David Goulet <dgoulet@efficios.com>
+#
+# This library is free software; you can redistribute it and/or modify it under
+# the terms of the GNU Lesser General Public License as published by the Free
+# Software Foundation; version 2.1 of the License.
+#
+# This library is distributed in the hope that it will be useful, but WITHOUT
+# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
+# FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more
+# details.
+#
+# You should have received a copy of the GNU Lesser General Public License
+# along with this library; if not, write to the Free Software Foundation, Inc.,
+# 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA
+
+SESSIOND_BIN="lttng-sessiond"
+LTTNG_BIN="lttng"
+BABELTRACE_BIN="babeltrace"
+
+function start_sessiond ()
+{
+       if [ -z $(pidof $SESSIOND_BIN) ]; then
+               echo -n "Starting session daemon... "
+               $TESTDIR/../lttng-sessiond/$SESSIOND_BIN --daemonize --quiet
+               if [ $? -eq 1 ]; then
+                       echo -e "\e[1;31mFAILED\e[0m"
+                       return 1
+               else
+                       echo -e "\e[1;32mOK\e[0m"
+               fi
+       fi
+}
+
+function stop_sessiond ()
+{
+       PID_SESSIOND=`pidof lt-$SESSIOND_BIN`
+
+       echo -e -n "Killing session daemon... "
+       kill $PID_SESSIOND >/dev/null 2>&1
+       if [ $? -eq 1 ]; then
+               echo -e "\e[1;31mFAILED\e[0m"
+               return 1
+       else
+               echo -e "\e[1;32mOK\e[0m"
+       fi
+}
+
+function create_lttng_session ()
+{
+       sess_name=$1
+       trace_path=$2
+
+       echo -n "Creating lttng session $SESSION_NAME in $TRACE_PATH "
+       $TESTDIR/../lttng/$LTTNG_BIN create $sess_name -o $trace_path >/dev/null 2>&1
+       if [ $? -eq 1 ]; then
+               echo -e "\e[1;31mFAILED\e[0m"
+               return 1
+       else
+               echo -e "\e[1;32mOK\e[0m"
+               #echo $out | grep "written in" | cut -d' ' -f6
+       fi
+}
+
+function enable_ust_lttng_event ()
+{
+       sess_name=$1
+       event_name=$2
+
+       echo -n "Enabling lttng event $event_name for session $sess_name "
+       $TESTDIR/../lttng/$LTTNG_BIN enable-event $event_name -s $sess_name -u >/dev/null 2>&1
+       if [ $? -eq 1 ]; then
+               echo -e '\e[1;31mFAILED\e[0m'
+               return 1
+       else
+               echo -e "\e[1;32mOK\e[0m"
+       fi
+}
+
+function start_tracing ()
+{
+       sess_name=$1
+
+       echo -n "Start lttng tracing for session $sess_name "
+       $TESTDIR/../lttng/$LTTNG_BIN start $sess_name >/dev/null 2>&1
+       if [ $? -eq 1 ]; then
+               echo -e '\e[1;31mFAILED\e[0m'
+               return 1
+       else
+               echo -e "\e[1;32mOK\e[0m"
+       fi
+}
+
+function stop_tracing ()
+{
+       sess_name=$1
+
+       echo -n "Stop lttng tracing for session $sess_name "
+       $TESTDIR/../lttng/$LTTNG_BIN stop $sess_name >/dev/null 2>&1
+       if [ $? -eq 1 ]; then
+               echo -e '\e[1;31mFAILED\e[0m'
+               return 1
+       else
+               echo -e "\e[1;32mOK\e[0m"
+       fi
+}
+
+function destroy_lttng_session ()
+{
+       sess_name=$1
+
+       echo -n "Destroy lttng session $sess_name "
+       $TESTDIR/../lttng/$LTTNG_BIN destroy $sess_name >/dev/null 2>&1 
+       if [ $? -eq 1 ]; then
+               echo -e '\e[1;31mFAILED\e[0m'
+               return 1
+       else
+               echo -e "\e[1;32mOK\e[0m"
+       fi
+}
+
+function trace_matches ()
+{
+       event_name=$1
+       nr_iter=$2
+       trace_path=$3
+
+       echo -n "Looking for $nr_iter $event_name in $trace_path "
+
+       count=$($BABELTRACE_BIN $trace_path | grep $event_name | wc -l)
+       if [ "$count" -ne "$nr_iter" ]; then
+               echo -e "$count found in trace \e[1;31mFAILED\e[0m"
+               return 1
+       else
+               echo -e "Trace is coherent \e[1;32mOK\e[0m"
+               return 0
+       fi
+}