Import CStringView from the Babeltrace tree
[lttng-tools.git] / tests / regression / ust / fork / test_fork.py
index 3baa8cc389dc6aa88e4f21de8880b2ea324b19d2..40f0b4073f96cc410f6f36f3c81b0980050348a8 100644 (file)
@@ -1,19 +1,8 @@
 #!/usr/bin/env python3
 #
-# Copyright (C) 2013 Jérémie Galarneau <jeremie.galarneau@efficios.com>
+# Copyright (C) 2013 Jérémie Galarneau <jeremie.galarneau@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.
+# SPDX-License-Identifier: GPL-2.0-only
 
 import os
 import subprocess
@@ -36,19 +25,23 @@ print("1..{0}".format(NR_TESTS))
 
 # Check if a sessiond is running... bail out if none found.
 if session_daemon_alive() == 0:
-    bail("No sessiond running. Please make sure you are running this test with the \"run\" shell script and verify that the lttng tools are properly installed.")
+    bail(
+        'No sessiond running. Please make sure you are running this test with the "run" shell script and verify that the lttng tools are properly installed.'
+    )
 
 session_info = create_session()
 enable_ust_tracepoint_event(session_info, "ust_tests_fork*")
 start_session(session_info)
 
-fork_process = subprocess.Popen([test_path + "fork", test_path + "fork2"], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
-fork_process.wait()
-
+fork_process = subprocess.Popen(
+    [test_path + "fork", test_path + "fork2"],
+    stdout=subprocess.PIPE,
+    stderr=subprocess.PIPE,
+)
 parent_pid = -1
 child_pid = -1
 for line in fork_process.stdout:
-    line = line.decode('utf-8').replace("\n", "")
+    line = line.decode("utf-8").replace("\n", "")
     match = re.search(r"child_pid (\d+)", line)
     if match:
         child_pid = match.group(1)
@@ -56,28 +49,41 @@ for line in fork_process.stdout:
     if match:
         parent_pid = match.group(1)
 
-print_test_result(fork_process.returncode == 0, current_test, "Fork test application exited normally")
+fork_process.wait()
+
+print_test_result(
+    fork_process.returncode == 0, current_test, "Fork test application exited normally"
+)
 current_test += 1
 
 stop_session(session_info)
 
 # Check both events (normal exit and suicide messages) are present in the resulting trace
 try:
-    babeltrace_process = subprocess.Popen(["babeltrace", session_info.trace_path], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+    babeltrace_process = subprocess.Popen(
+        [BABELTRACE_BIN, session_info.trace_path],
+        stdout=subprocess.PIPE,
+        stderr=subprocess.PIPE,
+    )
 except FileNotFoundError:
-    bail("Could not open babeltrace. Please make sure it is installed.", session_info)
+    bail("Could not open {}. Please make sure it is installed.".format(BABELTRACE_BIN))
 
 event_lines = []
 for event_line in babeltrace_process.stdout:
-    event_line = event_line.decode('utf-8').replace("\n", "")
-    if re.search(r"warning", event_line) is not None or re.search(r"error", event_line) is not None:
-        print( "# " + event_line )
+    event_line = event_line.decode("utf-8").replace("\n", "")
+    if (
+        re.search(r"warning", event_line) is not None
+        or re.search(r"error", event_line) is not None
+    ):
+        print("# " + event_line)
     else:
         event_lines.append(event_line)
 
 babeltrace_process.wait()
 
-print_test_result(babeltrace_process.returncode == 0, current_test, "Resulting trace is readable")
+print_test_result(
+    babeltrace_process.returncode == 0, current_test, "Resulting trace is readable"
+)
 current_test += 1
 
 if babeltrace_process.returncode != 0:
@@ -96,21 +102,37 @@ for event_line in event_lines:
         continue
 
     if re.search(r"before_fork", event_line):
-        event_before_fork = (event_pid == parent_pid)
+        event_before_fork = event_pid == parent_pid
     if re.search(r"after_fork_parent", event_line):
-        event_after_fork_parent = (event_pid == parent_pid)
+        event_after_fork_parent = event_pid == parent_pid
     if re.search(r"after_fork_child", event_line):
-        event_after_fork_child = (event_pid == child_pid)
+        event_after_fork_child = event_pid == child_pid
     if re.search(r"after_exec", event_line):
-        event_after_exec = (event_pid == child_pid)
+        event_after_exec = event_pid == child_pid
 
-print_test_result(event_before_fork, current_test, "before_fork event logged by parent process found in trace")
+print_test_result(
+    event_before_fork,
+    current_test,
+    "before_fork event logged by parent process found in trace",
+)
 current_test += 1
-print_test_result(event_after_fork_parent, current_test, "after_fork_parent event logged by parent process found in trace")
+print_test_result(
+    event_after_fork_parent,
+    current_test,
+    "after_fork_parent event logged by parent process found in trace",
+)
 current_test += 1
-print_test_result(event_after_fork_child, current_test, "after_fork_child event logged by child process found in trace")
+print_test_result(
+    event_after_fork_child,
+    current_test,
+    "after_fork_child event logged by child process found in trace",
+)
 current_test += 1
-print_test_result(event_after_exec, current_test, "after_exec event logged by child process found in trace")
+print_test_result(
+    event_after_exec,
+    current_test,
+    "after_exec event logged by child process found in trace",
+)
 current_test += 1
 
 shutil.rmtree(session_info.tmp_directory)
This page took 0.026484 seconds and 4 git commands to generate.