Enhance the bpftrace function to raise an exception if the underlying bpftrace command returns a non-zero exit code. This helps detect and surface errors from bpftrace execution in test scripts, improving robustness and debugging capabilities.
Signed-off-by: Breno Leitao leitao@debian.org --- tools/testing/selftests/net/lib/py/utils.py | 3 +++ 1 file changed, 3 insertions(+)
diff --git a/tools/testing/selftests/net/lib/py/utils.py b/tools/testing/selftests/net/lib/py/utils.py index 760ccf6fccccc..c4e26567ee6fb 100644 --- a/tools/testing/selftests/net/lib/py/utils.py +++ b/tools/testing/selftests/net/lib/py/utils.py @@ -203,6 +203,9 @@ def bpftrace(expr, json=None, ns=None, host=None, timeout=None): expr += ' interval:s:' + str(timeout) + ' { exit(); }' cmd_arr += ['-e', expr] cmd_obj = cmd(cmd_arr, ns=ns, host=host, shell=False) + if cmd_obj.ret != 0: + raise Exception("Warning: bpftrace command returned a non-zero exit code.") + if json: # bpftrace prints objects as lines ret = {}