Fix: test: handle env variables undefined
[lttng-tools.git] / tests / regression / ust / ust-dl / test_ust-dl.py
1 #!/usr/bin/env python3
2 #
3 # Copyright (C) - 2013 Jérémie Galarneau <jeremie.galarneau@efficios.com>
4 # Copyright (C) - 2015 Antoine Busque <abusque@efficios.com>
5 #
6 # This program is free software; you can redistribute it and/or modify it
7 # under the terms of the GNU General Public License, version 2 only, as
8 # published by the Free Software Foundation.
9 #
10 # This program is distributed in the hope that it will be useful, but WITHOUT
11 # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
12 # FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
13 # more details.
14 #
15 # You should have received a copy of the GNU General Public License along with
16 # this program; if not, write to the Free Software Foundation, Inc., 51
17 # Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
18
19 import os
20 import subprocess
21 import re
22 import shutil
23 import sys
24
25 test_path = os.path.dirname(os.path.abspath(__file__)) + "/"
26 test_utils_path = test_path
27 for i in range(4):
28 test_utils_path = os.path.dirname(test_utils_path)
29 test_utils_path = test_utils_path + "/utils"
30 sys.path.append(test_utils_path)
31 from test_utils import *
32
33
34 NR_TESTS = 6
35 current_test = 1
36 print("1..{0}".format(NR_TESTS))
37
38 # Check if a sessiond is running... bail out if none found.
39 if session_daemon_alive() == 0:
40 bail("""No sessiond running. Please make sure you are running this test
41 with the "run" shell script and verify that the lttng tools are
42 properly installed.""")
43
44 session_info = create_session()
45 enable_ust_tracepoint_event(session_info, "*")
46 start_session(session_info)
47
48 test_env = os.environ.copy()
49 test_env["LD_PRELOAD"] = test_env.get("LD_PRELOAD", "") + ":liblttng-ust-dl.so"
50 test_env["LD_LIBRARY_PATH"] = test_env.get("LD_LIBRARY_PATH", "") + ":" + test_path
51 test_process = subprocess.Popen(test_path + "prog",
52 stdout=subprocess.PIPE, stderr=subprocess.PIPE,
53 env=test_env)
54
55 if sys.version_info >= (3, 3):
56 try:
57 test_process.wait(5)
58 except subprocess.TimeoutExpired:
59 test_process.kill()
60 bail("Failed to run ust-dl test application.", session_info)
61 else:
62 test_process.wait()
63
64 print_test_result(test_process.returncode == 0, current_test, "Test application exited normally")
65 current_test += 1
66
67 stop_session(session_info)
68
69 # Check for dl events in the resulting trace
70 try:
71 babeltrace_process = subprocess.Popen(["babeltrace", session_info.trace_path], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
72 except FileNotFoundError:
73 bail("Could not open babeltrace. Please make sure it is installed.", session_info)
74
75 dlopen_event_found = False
76 build_id_event_found = False
77 debug_link_event_found = False
78 dlclose_event_found = False
79
80 for event_line in babeltrace_process.stdout:
81 # Let babeltrace finish to get the return code
82 if dlopen_event_found and build_id_event_found and \
83 debug_link_event_found and dlclose_event_found:
84 continue
85
86 event_line = event_line.decode('utf-8').replace("\n", "")
87 if re.search(r".*lttng_ust_dl:dlopen.*", event_line) is not None:
88 dlopen_event_found = True
89 elif re.search(r".*lttng_ust_dl:build_id.*", event_line) is not None:
90 build_id_event_found = True
91 elif re.search(r".*lttng_ust_dl:debug_link.*", event_line) is not None:
92 debug_link_event_found = True
93 elif re.search(r".*lttng_ust_dl:dlclose.*", event_line) is not None:
94 dlclose_event_found = True
95
96 babeltrace_process.wait()
97
98 print_test_result(babeltrace_process.returncode == 0, current_test, "Resulting trace is readable")
99 current_test += 1
100
101 print_test_result(dlopen_event_found, current_test, "lttng_ust_dl:dlopen event found in resulting trace")
102 current_test += 1
103
104 print_test_result(build_id_event_found, current_test, "lttng_ust_dl:build_id event found in resulting trace")
105 current_test += 1
106
107 print_test_result(debug_link_event_found, current_test, "lttng_ust_dl:debug_link event found in resulting trace")
108 current_test += 1
109
110 print_test_result(dlclose_event_found, current_test, "lttng_ust_dl:dlclose event found in resulting trace")
111 current_test += 1
112
113 shutil.rmtree(session_info.tmp_directory)
This page took 0.044207 seconds and 4 git commands to generate.