tests: Add probe_read_kernel to known types

bpftrace · Aug 28, 2020 · 5441336 · 5441336
1 parent 1038d07
commit 5441336
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/tests/runtime/engine/parser.py b/tests/runtime/engine/parser.py
@@ -98,7 +98,7 @@ def __read_test_struct(test, test_suite):
                 arch = [x.strip() for x in line.split("|")]
             elif item_name == 'REQUIRES_FEATURE':
                 feature_requirement = {x.strip() for x in line.split(" ")}
-                unknown = feature_requirement - {"loop", "btf"}
+                unknown = feature_requirement - {"loop", "btf", "probe_read_kernel"}
                 if len(unknown) > 0:
                     raise UnknownFieldError('%s is invalid for REQUIRES_FEATURE. Suite: %s' % (','.join(unknown), test_suite))
             else:

diff --git a/tests/runtime/engine/utils.py b/tests/runtime/engine/utils.py
@@ -96,6 +96,7 @@ def __get_bpffeature():
         output = p.communicate()[0]
         bpffeature = {}
         bpffeature["loop"] = output.find("Loop support: yes") != -1
+        bpffeature["probe_read_kernel"] = output.find("probe_read_kernel: yes") != -1
         bpffeature["btf"] = output.find("btf (depends on Build:libbpf): yes") != -1
         return bpffeature