+#!/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
+
+CURDIR=$(dirname $0)/
+TESTDIR=$CURDIR/../..
+EVENT_NAME="sched_switch"
+PID_RELAYD=0
+SESSION_NAME=""
+
+source $TESTDIR/utils.sh
+
+echo -e "\n---------------------------"
+echo -e " Streaming - Kernel tracing "
+echo -e "----------------------------"
+
+if [ "$(id -u)" != "0" ]; then
+ echo "This test must be running as root. Aborting"
+ # Exit status 0 so the tests can continue
+ exit 0
+fi
+
+# LTTng kernel modules check
+out=`ls /lib/modules/$(uname -r)/extra | grep lttng`
+if [ -z "$out" ]; then
+ echo "LTTng modules not detected. Aborting kernel tests!"
+ echo ""
+ # Exit status 0 so the tests can continue
+ exit 0
+fi
+
+function lttng_create_session
+{
+ echo -n "Creating session $SESSION_NAME... "
+ # Create session with default path
+ $TESTDIR/../src/bin/lttng/$LTTNG_BIN create $SESSION_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 lttng_enable_consumer_localhost
+{
+ echo -n "Enabling network consumer... "
+ # Create session with default path
+ $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-consumer -k net://localhost >/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 test_kernel_before_start ()
+{
+ echo -e "\n=== Testing kernel streaming with event enable BEFORE start\n"
+ lttng_create_session
+ lttng_enable_consumer_localhost
+ lttng_enable_kernel_event $SESSION_NAME $EVENT_NAME
+ start_tracing $SESSION_NAME
+ # Give a second
+ sleep 1
+ stop_tracing $SESSION_NAME
+ destroy_lttng_session $SESSION_NAME
+}
+
+function test_kernel_after_start ()
+{
+ echo -e "\n=== Testing kernel streaming with event enable AFTER start\n"
+ lttng_create_session
+ lttng_enable_consumer_localhost
+ start_tracing $SESSION_NAME
+ lttng_enable_kernel_event $SESSION_NAME $EVENT_NAME
+ # Give a second
+ sleep 1
+ stop_tracing $SESSION_NAME
+ destroy_lttng_session $SESSION_NAME
+}
+
+start_sessiond
+lttng_start_relayd
+
+tests=( test_kernel_before_start test_kernel_after_start )
+
+for fct_test in ${tests[@]};
+do
+ SESSION_NAME=$(randstring 16 0)
+ ${fct_test}
+
+ # Validate test
+ validate_trace $EVENT_NAME ~/lttng-traces/$HOSTNAME/$SESSION_NAME*
+ if [ $? -eq 0 ]; then
+ # Only delete if successful
+ rm -rf ~/lttng-traces/$HOSTNAME/$SESSION_NAME*
+ rm -rf ~/lttng-traces/$SESSION_NAME*
+ else
+ break
+ fi
+done
+
+echo ""
+stop_sessiond
+lttng_stop_relayd
+
+
+exit $out