1 files changed, 104 insertions, 0 deletions
diff --git a/tools/perf/tests/shell/test_netdev_times_python.sh b/tools/perf/tests/shell/test_netdev_times_python.sh
new file mode 100755
index 000000000000..0d8d10a42cab
--- /dev/null
+++ b/tools/perf/tests/shell/test_netdev_times_python.sh
@@ -0,0 +1,104 @@
+#!/bin/bash
+# SPDX-License-Identifier: GPL-2.0
+# netdev_times python test
+
+set -e
+
+shelldir=$(dirname "$0")
+# shellcheck source=lib/setup_python.sh
+. "${shelldir}"/lib/setup_python.sh
+
+if ! "$PYTHON" -c 'import perf' > /dev/null 2>&1; then
+	echo "Skipping test, perf python module not found"
+	exit 2
+fi
+
+script_dir="$(dirname "$0")/../../python"
+script_path="${script_dir}/netdev-times.py"
+
+if ! perf check feature -q libtraceevent > /dev/null 2>&1; then
+	echo "Skipping test, libtraceevent is disabled"
+	exit 2
+fi
+
+if [ ! -f "$script_path" ]; then
+	echo "Skipping test, netdev-times.py not found at $script_path"
+	exit 2
+fi
+
+err=0
+temp_data=""
+temp_out=""
+
+cleanup() {
+	rm -f "${temp_data}" "${temp_out}"
+}
+trap 'cleanup' EXIT TERM INT
+
+temp_data=$(mktemp /tmp/perf.data.XXXXXX)
+temp_out=$(mktemp /tmp/perf.out.XXXXXX)
+
+echo "Testing netdev-times.py..."
+
+if ! "$PYTHON" - "$script_path" << 'EOF'
+import argparse
+import contextlib
+import importlib.util
+import io
+import sys
+
+spec = importlib.util.spec_from_file_location("netdev_times", sys.argv[1])
+mod = importlib.util.module_from_spec(spec)
+spec.loader.exec_module(mod)
+
+cfg = argparse.Namespace(tx=True, rx=False, dev=None, debug=False, input="perf.data")
+analyzer = mod.NetdevTimes(cfg)
+analyzer.handle_net_dev_queue({
+    "time": 1_000_000_000,
+    "skbaddr": 0xdeadbeef,
+    "skblen": 1500,
+    "dev_name": "eth0",
+})
+analyzer.handle_net_dev_xmit({
+    "time": 1_000_500_000,
+    "skbaddr": 0xdeadbeef,
+    "rc": 0,
+})
+analyzer.handle_kfree_skb({
+    "time": 1_000_750_000,
+    "skbaddr": 0xdeadbeef,
+    "comm": "ping",
+    "pid": 42,
+    "location": 0xffffffff81000000,
+})
+
+buf = io.StringIO()
+with contextlib.redirect_stdout(buf):
+    analyzer.print_summary()
+out = buf.getvalue()
+assert "eth0" in out and "1500" in out and "1.000000sec" in out, out
+assert "0.500msec" in out and "0.250msec" in out, out
+EOF
+then
+	echo "netdev-times.py synthetic unit test failed"
+	exit 1
+fi
+
+# Create a perf.data file. Force dropping a packet if tracepoint is available!
+if ! perf record -B -N --no-bpf-event -e skb:kfree_skb -a -o "${temp_data}" \
+	-- ping -c 1 -W 1 127.0.0.1 >/dev/null 2>&1; then
+	perf record -B -N --no-bpf-event -e cycles -o "${temp_data}" \
+		-- perf test -w noploop >/dev/null 2>&1 || \
+		{ echo "Skipping test, perf record failed"; exit 2; }
+fi
+
+# Check that the script executes
+if ! perf script netdev-times -i "${temp_data}" > "${temp_out}"; then
+	echo "netdev-times.py test failed"
+	err=1
+else
+	echo "netdev-times test passed."
+fi
+rm -f "${temp_out}"
+
+exit $err