X-Git-Url: https://git.lttng.org/?a=blobdiff_plain;f=tests%2Ftest_functions.sh;fp=tests%2Ftest_functions.sh;h=d27a734176124a75f317dc476d02169d3609fc4d;hb=dd7a064c5953d06fbd6010f0a0f4a6f94df01a9f;hp=0000000000000000000000000000000000000000;hpb=72d022d60207b479177401cfdeddeb685d5aa85f;p=ust.git

diff --git a/tests/test_functions.sh b/tests/test_functions.sh
new file mode 100644
index 0000000..d27a734
--- /dev/null
+++ b/tests/test_functions.sh
@@ -0,0 +1,92 @@
+#!/bin/bash
+#
+# Copyright 2010 Ericsson AB
+#
+#    This file is part of LTTng-UST.
+#
+#    LTTng-UST is free software: you can redistribute it and/or modify
+#    it under the terms of the GNU General Public License as published by
+#    the Free Software Foundation, either version 3 of the License, or
+#    (at your option) any later version.
+#
+#    LTTng-UST 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 General Public License for more details.
+#
+#    You should have received a copy of the GNU General Public License
+#    along with LTTng-UST.  If not, see <http://www.gnu.org/licenses/>.
+
+function starttest() {
+
+	echo "------------------------------------"
+	echo "Starting test: $1"
+	echo "------------------------------------"
+}
+
+function check_trace_logs() {
+	TRACE=$1
+
+	for f in $(ls $1/*.log); do
+		NLINES=$(egrep "Warning|Error" $f | wc -l)
+		if [ "$NLINES" -ne "0" ]; then
+			fail "Errors/warnings found in $f"
+			return 1;
+		fi
+	done
+	pass "$f was consistent"
+	return 0;
+}
+
+
+function trace_matches() {
+
+    RUNLTTV=~/devel/lttv/runlttv
+
+    if [ ! -x "$RUNLTTV" ]; then
+	echo "$0: $RUNLTTV not executable. Edit \$RUNLTTV to point to your lttv source directory." >/dev/stderr
+	exit 1;
+    fi
+
+    while getopts ":n:N:" options; do
+	case "$options" in
+	    n) expected_count=$OPTARG;;
+	    N) name=$OPTARG;;
+	    *) echo "Invalid option to trace_matches"
+		exit 1;;
+	esac
+    done
+    shift $(($OPTIND - 1))
+
+    pattern=$1
+    if [ -z "$pattern" ]; then
+	error "no pattern specified"
+	usage
+	exit 1
+    fi
+
+    if [ -z "$2" ]; then
+	error "no trace directory specified"
+	return 1
+    fi
+    traces=$(find "$2" -mindepth 1 -maxdepth 1 -type d)
+
+    cnt=$($RUNLTTV -m text "$traces" | grep "$pattern" | wc -l)
+    if [ -z "$expected_count" ]; then
+	if [ "$cnt" -eq "0" ]; then
+	    fail "Did not find at least one instance of $name in trace"
+	    return 1
+	else
+	    pass "Found at least one instance of $name in trace."
+	    return 0
+	fi
+    else
+	if [ "$cnt" -ne "$expected_count" ]; then
+	    fail "Found $cnt instances of $name in trace, expected $expected_count"
+	    return 1
+	else
+	    pass "Found $cnt instances of $name in trace."
+	    return 0
+	fi
+    fi
+}