tracee_test: Add tests for execveat_flags_to_str

Signed-off-by: Simarpreet Singh <simar@linux.com>
aquasecurity · Nov 5, 2019 · efc2f14 · efc2f14
1 parent d0f474f
commit efc2f14
Showing 1 changed file with 29 additions and 0 deletions.
diff --git a/test_container_tracer.py b/test_container_tracer.py
@@ -14,6 +14,35 @@ def test_load_bpf_program(self):
         actualbpf = tracee.container_tracer.load_bpf_program()
         self.assertEqual(expectedbpf, actualbpf, "should be a valid bpf program")
 
+    def test_execveat_flags_to_str(self):
+        self.longMessage = True
+
+        test_cases = [
+            {
+                "name": "invalid flags",
+                "input": 0x666,
+                "expected": "",
+            },
+            {
+                "name": "empty path",
+                "input": 0x1000,
+                "expected": "AT_EMPTY_PATH",
+            },
+            {
+                "name": "symlink no follow",
+                "input": 0x100,
+                "expected": "AT_SYMLINK_NOFOLLOW",
+            },
+            {
+                "name": "symlink no follow with empty path",
+                "input": 0x1100,
+                "expected": "AT_EMPTY_PATH|AT_SYMLINK_NOFOLLOW",
+            },
+        ]
+        for test_case in test_cases:
+            self.assertEqual(test_case["expected"], tracee.container_tracer.execveat_flags_to_str(test_case["input"]),
+                             test_case["name"])
+
 
 if __name__ == '__main__':
     unittest.main()