From: Simon Marchi <simon.marchi@efficios.com>
Date: Thu, 15 Aug 2019 15:17:11 +0000 (-0400)
Subject: tests: add tests for CLI params parsing
X-Git-Url: http://git.efficios.com/?p=babeltrace.git;a=commitdiff_plain;h=7b8c5132ba2d578e6df2fbe1375d3ce6d23ad051

tests: add tests for CLI params parsing

Add tests that checks various forms of valid parameters passed to the
CLI.

Change-Id: I81eb0ceefb6e4326fa001febbecfee5c66ad18c1
Signed-off-by: Simon Marchi <simon.marchi@efficios.com>
Reviewed-on: https://review.lttng.org/c/babeltrace/+/1943
Tested-by: jenkins <jenkins@lttng.org>
Reviewed-by: Philippe Proulx <eeppeliteloop@gmail.com>
---

diff --git a/tests/cli/params/test_params b/tests/cli/params/test_params
new file mode 100755
index 00000000..273b6244
--- /dev/null
+++ b/tests/cli/params/test_params
@@ -0,0 +1,69 @@
+#!/bin/bash
+#
+# Copyright (C) 2019 Simon Marchi <simon.marchi@efficios.com>
+#
+# This program is free software; you can redistribute it and/or modify it
+# under the terms of the GNU General Public License, version 2 only, as
+# published by the Free Software Foundation.
+#
+# This program 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
+# this program; if not, write to the Free Software Foundation, Inc., 51
+# Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+
+# Test how parameters are applied to sources auto-discovered by the convert
+# command.
+
+if [ "x${BT_TESTS_SRCDIR:-}" != "x" ]; then
+	UTILSSH="$BT_TESTS_SRCDIR/utils/utils.sh"
+else
+	UTILSSH="$(dirname "$0")/../../utils/utils.sh"
+fi
+
+# shellcheck source=../../utils/utils.sh
+SH_TAP=1 source "$UTILSSH"
+
+NUM_TESTS=8
+
+plan_tests $NUM_TESTS
+
+data_dir="${BT_TESTS_DATADIR}/cli/params"
+plugin_dir="${data_dir}"
+
+expected_file=$(mktemp -t expected.XXXXXX)
+
+function expect_success
+{
+	local test_name="$1"
+	local params_str="$2"
+	local expected_str="$3"
+
+	echo "$expected_str" > "$expected_file"
+
+	bt_diff_cli "$expected_file" /dev/null --plugin-path "$plugin_dir" -c "src.text.dmesg" \
+		-c "sink.params.SinkThatPrintsParams" --params "$params_str"
+	ok "$?" "$test_name"
+}
+
+expect_success 'null' 'a=null,b=nul,c=NULL' \
+	'{a=None, b=None, c=None}'
+expect_success 'bool' 'a=true,b=TRUE,c=yes,d=YES,e=false,f=FALSE,g=no,h=NO' \
+	'{a=True, b=True, c=True, d=True, e=False, f=False, g=False, h=False}'
+expect_success 'signed integer' 'a=0b110, b=022, c=22, d=0x22' \
+	'{a=6, b=18, c=22, d=34}'
+expect_success 'unsigned integer' 'a=+0b110, b=+022, c=+22, d=+0x22' \
+	'{a=6u, b=18u, c=22u, d=34u}'
+expect_success 'string' 'a="avril lavigne", b=patata, c="This\"is\\escaped"' \
+	'{a=avril lavigne, b=patata, c=This"is\escaped}'
+expect_success 'float' 'a=1.234, b=17., c=.28, d=-18.28' \
+	'{a=1.234, b=17.0, c=0.28, d=-18.28}'
+expect_success 'float scientific notation' 'a=10.5e6, b=10.5E6, c=10.5e-6, d=10.5E-6' \
+	'{a=10500000.0, b=10500000.0, c=1.05e-05, d=1.05e-05}'
+expect_success 'array' 'a=[1, [["hi",]]]' \
+	'{a=[1, [[hi]]]}'
+
+rm -f "$expected_file"
diff --git a/tests/data/cli/params/bt_plugin_params.py b/tests/data/cli/params/bt_plugin_params.py
new file mode 100644
index 00000000..4d750137
--- /dev/null
+++ b/tests/data/cli/params/bt_plugin_params.py
@@ -0,0 +1,51 @@
+#
+# Copyright (C) 2019 EfficiOS Inc.
+#
+# This program 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; only version 2
+# of the License.
+#
+# This program 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 this program; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
+#
+
+import bt2
+
+
+def to_string(p):
+    # Print BT values in a predictable way (the order of map entries) and with
+    # additional information (u suffix to differentiate unsigned integers from
+    # signed integers).
+
+    if type(p) is bt2.ArrayValue:
+        s = '[{}]'.format(', '.join([to_string(x) for x in p]))
+    elif type(p) is bt2.MapValue:
+        s = '{{{}}}'.format(
+            ', '.join([k + '=' + to_string(p[k]) for k in sorted(p.keys())])
+        )
+    elif type(p) is bt2.UnsignedIntegerValue:
+        s = str(p) + 'u'
+    else:
+        s = str(p)
+
+    return s
+
+
+@bt2.plugin_component_class
+class SinkThatPrintsParams(bt2._UserSinkComponent):
+    def __init__(self, params, obj):
+        self._add_input_port('in')
+        print(to_string(params))
+
+    def _user_consume(self):
+        raise bt2.Stop
+
+
+bt2.register_plugin(__name__, "params")