| 1 | import pytest |
| 2 | import os |
| 3 | import shutil |
| 4 | import signal |
| 5 | import subprocess |
| 6 | |
| 7 | import lttng_ivc.utils.ProjectFactory as ProjectFactory |
| 8 | import lttng_ivc.utils.utils as utils |
| 9 | import lttng_ivc.utils.runtime as Run |
| 10 | import lttng_ivc.settings as Settings |
| 11 | |
| 12 | """ |
| 13 | |
| 14 | FC: Fully Compatible |
| 15 | BC: Feature of the smallest version number will works. |
| 16 | TU: Tracing unavailable |
| 17 | |
| 18 | Note: actual testing is limited by lttng-ust and lttng-tools abi/api. |
| 19 | |
| 20 | +----------------------------------------------------------------------------------+ |
| 21 | | LTTng UST Java agent protocol vs LTTng session daemon | |
| 22 | +--------------------------+------------+------------+------------+----------------+ |
| 23 | | LTTng UST Java/ LTTng Tools | 2.7 (1.0) | 2.8 (2.0) | 2.9 (2.0) | 2.10 (2.0) | |
| 24 | +--------------------------+------------+------------+------------+----------------+ |
| 25 | | 2.7 (1.0) | FC | TU | TU | TU | |
| 26 | | 2.8 (2.0) | TU | FC | BC | BC | |
| 27 | | 2.9 (2.0) | TU | BC | FC | BC | |
| 28 | | 2.10 (2.0) | TU | BC | BC | FC | |
| 29 | +--------------------------+------------+------------+------------+----------------+ |
| 30 | |
| 31 | """ |
| 32 | |
| 33 | """ |
| 34 | First tuple member: lttng-ust label |
| 35 | Second tuple member: lttng-tool label |
| 36 | Third tuple member: app version to use |
| 37 | Fourth tuple member: expected scenario |
| 38 | """ |
| 39 | |
| 40 | version_to_app = {1: Settings.apps_jul_1, |
| 41 | 2: Settings.apps_jul_2} |
| 42 | |
| 43 | test_matrix_tracing_available = [ |
| 44 | ("lttng-ust-2.7", "lttng-tools-2.7", 1, True), |
| 45 | ("lttng-ust-2.7", "lttng-tools-2.8", 1, False), |
| 46 | ("lttng-ust-2.7", "lttng-tools-2.9", 1, False), |
| 47 | ("lttng-ust-2.7", "lttng-tools-2.10", 1, False), |
| 48 | ("lttng-ust-2.8", "lttng-tools-2.7", 2, False), |
| 49 | ("lttng-ust-2.8", "lttng-tools-2.8", 2, True), |
| 50 | ("lttng-ust-2.8", "lttng-tools-2.9", 2, False), |
| 51 | ("lttng-ust-2.8", "lttng-tools-2.10", 2, False), |
| 52 | ("lttng-ust-2.9", "lttng-tools-2.7", 2, False), |
| 53 | ("lttng-ust-2.9", "lttng-tools-2.8", 2, False), |
| 54 | ("lttng-ust-2.9", "lttng-tools-2.9", 2, True), |
| 55 | ("lttng-ust-2.9", "lttng-tools-2.10", 2, True), |
| 56 | ("lttng-ust-2.10", "lttng-tools-2.7", 2, False), |
| 57 | ("lttng-ust-2.10", "lttng-tools-2.8", 2, False), |
| 58 | ("lttng-ust-2.10", "lttng-tools-2.9", 2, True), |
| 59 | ("lttng-ust-2.10", "lttng-tools-2.10", 2, True), |
| 60 | ] |
| 61 | |
| 62 | test_matrix_agent_interface = [ |
| 63 | ("lttng-ust-2.7", "lttng-tools-2.7", 1, "Success"), |
| 64 | ("lttng-ust-2.7", "lttng-tools-2.8", 1, "Unsupported protocol"), |
| 65 | ("lttng-ust-2.7", "lttng-tools-2.9", 1, "Unsupported protocol"), |
| 66 | ("lttng-ust-2.7", "lttng-tools-2.10", 1, "Unsupported protocol"), |
| 67 | ("lttng-ust-2.8", "lttng-tools-2.7", 2, "Unsupported protocol"), |
| 68 | ("lttng-ust-2.8", "lttng-tools-2.8", 2, "Success"), |
| 69 | ("lttng-ust-2.8", "lttng-tools-2.9", 2, "Success"), |
| 70 | ("lttng-ust-2.8", "lttng-tools-2.10", 2, "Success"), |
| 71 | ("lttng-ust-2.9", "lttng-tools-2.7", 2, "Unsupported protocol"), |
| 72 | ("lttng-ust-2.9", "lttng-tools-2.8", 2, "Success"), |
| 73 | ("lttng-ust-2.9", "lttng-tools-2.9", 2, "Success"), |
| 74 | ("lttng-ust-2.9", "lttng-tools-2.10", 2, "Success"), |
| 75 | ("lttng-ust-2.10", "lttng-tools-2.7", 2, "Unsupported protocol"), |
| 76 | ("lttng-ust-2.10", "lttng-tools-2.8", 2, "Success"), |
| 77 | ("lttng-ust-2.10", "lttng-tools-2.9", 2, "Success"), |
| 78 | ("lttng-ust-2.10", "lttng-tools-2.10", 2, "Success"), |
| 79 | ] |
| 80 | |
| 81 | runtime_matrix_tracing_available = [] |
| 82 | runtime_matrix_agent_interface = [] |
| 83 | |
| 84 | if not Settings.test_only: |
| 85 | runtime_matrix_tracing_available = test_matrix_tracing_available |
| 86 | runtime_matrix_agent_interface = test_matrix_agent_interface |
| 87 | else: |
| 88 | for tup in test_matrix_tracing_available: |
| 89 | if (tup[0] in Settings.test_only or tup[1] in |
| 90 | Settings.test_only): |
| 91 | runtime_matrix_tracing_available.append(tup) |
| 92 | for tup in test_matrix_agent_interface: |
| 93 | if (tup[0] in Settings.test_only or tup[1] in |
| 94 | Settings.test_only): |
| 95 | runtime_matrix_agent_interface.append(tup) |
| 96 | |
| 97 | |
| 98 | @pytest.mark.parametrize("ust_label,tools_label,app_version,should_trace", runtime_matrix_tracing_available) |
| 99 | def test_ust_java_agent_tracing_available(tmpdir, ust_label, tools_label, app_version, should_trace): |
| 100 | |
| 101 | nb_iter = 100 |
| 102 | nb_events = 3 * nb_iter |
| 103 | |
| 104 | # Prepare environment |
| 105 | ust = ProjectFactory.get_precook(ust_label) |
| 106 | tools = ProjectFactory.get_precook(tools_label) |
| 107 | babeltrace = ProjectFactory.get_precook(Settings.default_babeltrace) |
| 108 | |
| 109 | tools_runtime_path = os.path.join(str(tmpdir), "tools") |
| 110 | ust_runtime_path = os.path.join(str(tmpdir), "ust") |
| 111 | app_path = os.path.join(str(tmpdir), "app") |
| 112 | |
| 113 | with Run.get_runtime(ust_runtime_path) as runtime_app, Run.get_runtime(tools_runtime_path) as runtime_tools: |
| 114 | runtime_tools.add_project(tools) |
| 115 | runtime_tools.add_project(babeltrace) |
| 116 | |
| 117 | runtime_app.add_project(ust) |
| 118 | runtime_app.lttng_home = runtime_tools.lttng_home |
| 119 | |
| 120 | trace_path = os.path.join(runtime_tools.lttng_home, 'trace') |
| 121 | |
| 122 | # Make application using the ust runtime |
| 123 | shutil.copytree(version_to_app[app_version], app_path) |
| 124 | runtime_app.run("javac App.java", cwd=app_path) |
| 125 | |
| 126 | # Start lttng-sessiond |
| 127 | sessiond = utils.sessiond_spawn(runtime_tools) |
| 128 | |
| 129 | # Create session using mi to get path and session name |
| 130 | runtime_tools.run('lttng create trace --output={}'.format(trace_path)) |
| 131 | |
| 132 | runtime_tools.run('lttng enable-event -j jello') |
| 133 | runtime_tools.run('lttng start') |
| 134 | |
| 135 | # Run application |
| 136 | cmd = 'java App {}'.format(nb_iter) |
| 137 | runtime_app.run(cmd, cwd=app_path) |
| 138 | |
| 139 | # Stop tracing |
| 140 | runtime_tools.run('lttng stop') |
| 141 | runtime_tools.run('lttng destroy -a') |
| 142 | cp = runtime_tools.subprocess_terminate(sessiond) |
| 143 | if cp.returncode != 0: |
| 144 | pytest.fail("Sessiond return code") |
| 145 | |
| 146 | # Read trace with babeltrace and check for event count via number of line |
| 147 | cmd = 'babeltrace {}'.format(trace_path) |
| 148 | if should_trace: |
| 149 | cp_process, cp_out, cp_err = runtime_tools.run(cmd) |
| 150 | assert(utils.line_count(cp_out) == nb_events) |
| 151 | else: |
| 152 | with pytest.raises(subprocess.CalledProcessError): |
| 153 | cp_process, cp_out, cp_err = runtime_tools.run(cmd) |
| 154 | |
| 155 | @pytest.mark.parametrize("ust_label,tools_label,app_version,outcome", runtime_matrix_agent_interface) |
| 156 | def test_ust_java_agent_interface(tmpdir, ust_label, tools_label, app_version, outcome): |
| 157 | """ |
| 158 | Use the agent coming from ust_label, but run app under tools_version runtime using ust agent. |
| 159 | """ |
| 160 | |
| 161 | nb_iter = 100 |
| 162 | nb_events = 3 * nb_iter |
| 163 | |
| 164 | # Prepare environment |
| 165 | ust = ProjectFactory.get_precook(ust_label) |
| 166 | tools = ProjectFactory.get_precook(tools_label) |
| 167 | babeltrace = ProjectFactory.get_precook(Settings.default_babeltrace) |
| 168 | |
| 169 | tools_runtime_path = os.path.join(str(tmpdir), "tools") |
| 170 | ust_runtime_path = os.path.join(str(tmpdir), "ust") |
| 171 | app_path = os.path.join(str(tmpdir), "app") |
| 172 | |
| 173 | with Run.get_runtime(ust_runtime_path) as runtime_app, Run.get_runtime(tools_runtime_path) as runtime_tools: |
| 174 | runtime_tools.add_project(tools) |
| 175 | runtime_tools.add_project(babeltrace) |
| 176 | |
| 177 | runtime_app.add_project(ust) |
| 178 | runtime_app.lttng_home = runtime_tools.lttng_home |
| 179 | |
| 180 | trace_path = os.path.join(runtime_tools.lttng_home, 'trace') |
| 181 | |
| 182 | # Make application using the ust runtime |
| 183 | shutil.copytree(version_to_app[app_version], app_path) |
| 184 | runtime_app.run("javac App.java", cwd=app_path) |
| 185 | |
| 186 | # Start lttng-sessiond |
| 187 | sessiond = utils.sessiond_spawn(runtime_tools) |
| 188 | |
| 189 | # Create session using mi to get path and session name |
| 190 | runtime_tools.run('lttng create trace --output={}'.format(trace_path)) |
| 191 | |
| 192 | runtime_tools.run('lttng enable-event -j jello') |
| 193 | runtime_tools.run('lttng start') |
| 194 | |
| 195 | # Steal the classpath from ust project |
| 196 | ust_classpath = ust.special_env_variables['CLASSPATH'] |
| 197 | |
| 198 | # Run application with tools runtime |
| 199 | cmd = 'java App {}'.format(nb_iter) |
| 200 | runtime_tools.run(cmd, cwd=app_path, classpath=ust_classpath) |
| 201 | |
| 202 | # Stop tracing |
| 203 | runtime_tools.run('lttng stop') |
| 204 | runtime_tools.run('lttng destroy -a') |
| 205 | cp = runtime_tools.subprocess_terminate(sessiond) |
| 206 | if cp.returncode != 0: |
| 207 | pytest.fail("Sessiond return code") |
| 208 | |
| 209 | # Read trace with babeltrace and check for event count via number of line |
| 210 | cmd = 'babeltrace {}'.format(trace_path) |
| 211 | if outcome == "Success": |
| 212 | assert(utils.file_contains(runtime_tools.get_subprocess_stderr_path(sessiond),["New registration for pid"])) |
| 213 | cp_process, cp_out, cp_err = runtime_tools.run(cmd) |
| 214 | assert(utils.line_count(cp_out) == nb_events) |
| 215 | else: |
| 216 | if outcome == "Unsupported protocol": |
| 217 | assert(not(utils.file_contains(runtime_tools.get_subprocess_stderr_path(sessiond),["New registration for pid"]))) |
| 218 | cp_process, cp_out, cp_err = runtime_tools.run(cmd) |
| 219 | assert(utils.line_count(cp_out) == 0) |