tests/regression/ust/exit-fast/test_exit-fast.py

   1 #!/usr/bin/env python3
   2 #
   3 # Copyright (C) 2013 Jérémie Galarneau <jeremie.galarneau@efficios.com>
   4 #
   5 # SPDX-License-Identifier: GPL-2.0-only
   6
   7 import os
   8 import subprocess
   9 import re
  10 import shutil
  11 import sys
  12
  13 test_path = os.path.dirname(os.path.abspath(__file__)) + "/"
  14 test_utils_path = test_path
  15 for i in range(4):
  16     test_utils_path = os.path.dirname(test_utils_path)
  17 test_utils_path = test_utils_path + "/utils"
  18 sys.path.append(test_utils_path)
  19 from test_utils import *
  20
  21
  22 normal_exit_message = "exit-fast tracepoint normal exit"
  23 suicide_exit_message = "exit-fast tracepoint suicide"
  24 NR_TESTS = 5
  25 current_test = 1
  26 print("1..{0}".format(NR_TESTS))
  27
  28 # Check if a sessiond is running... bail out if none found.
  29 if session_daemon_alive() == 0:
  30     bail(
  31         '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.'
  32     )
  33
  34 session_info = create_session()
  35 enable_ust_tracepoint_event(session_info, "ust_tests_exitfast*")
  36 start_session(session_info)
  37
  38 test_env = os.environ.copy()
  39 test_env["LTTNG_UST_REGISTER_TIMEOUT"] = "-1"
  40
  41 exit_fast_process = subprocess.Popen(
  42     test_path + "exit-fast",
  43     stdout=subprocess.DEVNULL,
  44     stderr=subprocess.DEVNULL,
  45     env=test_env,
  46 )
  47 exit_fast_process.wait()
  48
  49 print_test_result(
  50     exit_fast_process.returncode == 0, current_test, "Test application exited normally"
  51 )
  52 current_test += 1
  53
  54 exit_fast_process = subprocess.Popen(
  55     [test_path + "exit-fast", "suicide"],
  56     stdout=subprocess.DEVNULL,
  57     stderr=subprocess.DEVNULL,
  58     env=test_env,
  59 )
  60 exit_fast_process.wait()
  61
  62 stop_session(session_info)
  63
  64 # Check both events (normal exit and suicide messages) are present in the resulting trace
  65 try:
  66     babeltrace_process = subprocess.Popen(
  67         [BABELTRACE_BIN, session_info.trace_path],
  68         stdout=subprocess.PIPE,
  69         stderr=subprocess.PIPE,
  70     )
  71 except FileNotFoundError:
  72     bail("Could not open {}. Please make sure it is installed.".format(BABELTRACE_BIN))
  73
  74 event_lines = []
  75 for event_line in babeltrace_process.stdout:
  76     event_line = event_line.decode("utf-8").replace("\n", "")
  77     event_lines.append(event_line)
  78 babeltrace_process.wait()
  79
  80 print_test_result(
  81     babeltrace_process.returncode == 0, current_test, "Resulting trace is readable"
  82 )
  83 current_test += 1
  84
  85 if babeltrace_process.returncode != 0:
  86     bail("Unreadable trace; can't proceed with analysis.")
  87
  88 print_test_result(
  89     len(event_lines) == 2,
  90     current_test,
  91     "Correct number of events found in resulting trace",
  92 )
  93 current_test += 1
  94
  95 if len(event_lines) != 2:
  96     bail(
  97         "Unexpected number of events found in resulting trace ("
  98         + session_info.trace_path
  99         + ")."
 100     )
 101
 102 match = re.search(r".*message = \"(.*)\"", event_lines[0])
 103 print_test_result(
 104     match is not None and match.group(1) == normal_exit_message,
 105     current_test,
 106     "Tracepoint message generated during normal exit run is present in trace and has the expected value",
 107 )
 108 current_test += 1
 109
 110 match = re.search(r".*message = \"(.*)\"", event_lines[1])
 111 print_test_result(
 112     match is not None and match.group(1) == suicide_exit_message,
 113     current_test,
 114     "Tracepoint message generated during suicide run is present in trace and has the expected value",
 115 )
 116 current_test += 1
 117
 118 shutil.rmtree(session_info.tmp_directory)