Add statedump regen and star globing for ust testing

[deliverable/lttng-ivc.git] / lttng_ivc / tests / ust_app_vs_ust_tools / test_ust_app_vs_ust_tools.py

import pytest
import os
import shutil
import signal
import subprocess

import lttng_ivc.utils.ProjectFactory as ProjectFactory
import lttng_ivc.utils.utils as utils
import lttng_ivc.utils.runtime as Run
import lttng_ivc.settings as Settings

"""

FC: Fully Compatible
BC: Feature of the smallest version number will works.
TU: Tracing unavailable

+------------------------------------------------------------------------------+
|              LTTng UST control protocol compatibility matrix                 |
|   (between applications and tools linked on LTTng UST and liblttng-ust-ctl)  |
+--------------------------+------------+------------+------------+------------+
| LTTng UST / LTTng Tools  | 2.7 (6.0)  | 2.8 (6.1)  | 2.9 (7.1)  | 2.10 (7.2) |
+--------------------------+------------+------------+------------+------------+
| 2.7 (6.0)                | FC         | BC         | TU         | TU         |
| 2.8 (6.1)                | BC         | FC         | TU         | TU         |
| 2.9 (7.1)                | TU         | TU         | FC         | BC         |
| 2.10 (7.2)               | TU         | TU         | BC         | FC         |
+--------------------------+------------+------------+------------+------------+

Version number of this API is defined in include/lttng/ust-abi.h of lttng-ust project

"""

"""
First tuple member: lttng-ust label
Second tuple member: lttng-tool label
Third tuple member: expected scenario
"""

test_matrix_tracing_available = [
    ("lttng-ust-2.7", "lttng-tools-2.7",   True),
    ("lttng-ust-2.7", "lttng-tools-2.8",   True),
    ("lttng-ust-2.7", "lttng-tools-2.9",   False),
    ("lttng-ust-2.7", "lttng-tools-2.10",  False),
    ("lttng-ust-2.8", "lttng-tools-2.7",   True),
    ("lttng-ust-2.8", "lttng-tools-2.8",   True),
    ("lttng-ust-2.8", "lttng-tools-2.9",   False),
    ("lttng-ust-2.8", "lttng-tools-2.10",  False),
    ("lttng-ust-2.9", "lttng-tools-2.7",   False),
    ("lttng-ust-2.9", "lttng-tools-2.8",   False),
    ("lttng-ust-2.9", "lttng-tools-2.9",   True),
    ("lttng-ust-2.9", "lttng-tools-2.10",  True),
    ("lttng-ust-2.10", "lttng-tools-2.7",  False),
    ("lttng-ust-2.10", "lttng-tools-2.8",  False),
    ("lttng-ust-2.10", "lttng-tools-2.9",  True),
    ("lttng-ust-2.10", "lttng-tools-2.10", True),
]

# Only consider test case for which tracing is valid
test_matrix_app_context = [
    ("lttng-ust-2.7", "lttng-tools-2.7",  ""),
    ("lttng-ust-2.7", "lttng-tools-2.8",  ""),
    ("lttng-ust-2.7", "lttng-tools-2.9",  ""),
    ("lttng-ust-2.7", "lttng-tools-2.10", ""),
    ("lttng-ust-2.8", "lttng-tools-2.7",  ""),
    ("lttng-ust-2.8", "lttng-tools-2.8",  ""),
    ("lttng-ust-2.8", "lttng-tools-2.9",  ""),
    ("lttng-ust-2.8", "lttng-tools-2.10", ""),
    ("lttng-ust-2.9", "lttng-tools-2.7",  ""),
    ("lttng-ust-2.9", "lttng-tools-2.8",  ""),
    ("lttng-ust-2.9", "lttng-tools-2.9",  ""),
    ("lttng-ust-2.9", "lttng-tools-2.10", ""),
    ("lttng-ust-2.10", "lttng-tools-2.7", ""),
    ("lttng-ust-2.10", "lttng-tools-2.8", ""),
    ("lttng-ust-2.10", "lttng-tools-2.9", ""),
    ("lttng-ust-2.10", "lttng-tools-2.10", ""),
]

"""
Statedump are supported starting at lttng-ust >= 2.9.
No need to test for prior lttng-ust givent that there is also a bump leading to
loss of tracing for application running under lttng-ust < 2.9.
"""
test_matrix_regen_statedump = [
        ("lttng-ust-2.9", "lttng-tools-2.9",   True),
        ("lttng-ust-2.9", "lttng-tools-2.10",  True),
        ("lttng-ust-2.10", "lttng-tools-2.9",  True),
        ("lttng-ust-2.10", "lttng-tools-2.10", True),
]

test_matrix_starglobing_enabler = [
        ("lttng-ust-2.9", "lttng-tools-2.9",   "Unsupported by tools"),
        ("lttng-ust-2.9", "lttng-tools-2.10",  "Unsupported by ust"),
        ("lttng-ust-2.10", "lttng-tools-2.9",  "Unsupported by tools"),
        ("lttng-ust-2.10", "lttng-tools-2.10", "Supported"),
]

runtime_matrix_tracing_available = []
runtime_matrix_regen_statedump = []
runtime_rix_starglobing_enabler = []

if not Settings.test_only:
    runtime_matrix_tracing_available = test_matrix_tracing_available
    runtime_matrix_regen_statedump = test_matrix_regen_statedump
    runtime_matrix_starglobing_enabler = test_matrix_starglobing_enabler
else:
    for tup in test_matrix_tracing_available:
        if (tup[0] in Settings.test_only or tup[1] in
                Settings.test_only):
            runtime_matrix_tracing_available.append(tup)
    for tup in test_matrix_regen_statedump:
        if (tup[0] in Settings.test_only or tup[1] in
                Settings.test_only):
            runtime_matrix_regen_statedump.append(tup)
    for tup in test_matrix_starglobing_enabler:
        if (tup[0] in Settings.test_only or tup[1] in
                Settings.test_only):
            runtime_matrix_starglobing_enabler.append(tup)


@pytest.mark.parametrize("ust_label,tools_label, should_trace", runtime_matrix_tracing_available)
def test_ust_app_tracing_available(tmpdir, ust_label, tools_label, should_trace):

    if ((ust_label == "lttng-ust-2.7" and tools_label == "lttng-tools-2.8") or
            (ust_label == "lttng-ust-2.8" and tools_label == "lttng-tools-2.7")):
        pytest.xfail("Failing but should work, problem regarding the size of fields structure")

    nb_events = 100

    # Prepare environment
    ust = ProjectFactory.get_precook(ust_label)
    tools = ProjectFactory.get_precook(tools_label)
    babeltrace = ProjectFactory.get_precook(Settings.default_babeltrace)

    tools_runtime_path = os.path.join(str(tmpdir), "tools")
    ust_runtime_path = os.path.join(str(tmpdir), "ust")
    app_path = os.path.join(str(tmpdir), "app")

    with Run.get_runtime(ust_runtime_path) as runtime_app, Run.get_runtime(tools_runtime_path) as runtime_tools:
        runtime_tools.add_project(tools)
        runtime_tools.add_project(babeltrace)

        runtime_app.add_project(ust)
        runtime_app.lttng_home = runtime_tools.lttng_home

        trace_path = os.path.join(runtime_tools.lttng_home, 'trace')

        # Make application using the ust runtime
        shutil.copytree(Settings.apps_gen_events_folder, app_path)
        runtime_app.run("make V=1", cwd=app_path)

        # Start lttng-sessiond
        sessiond = utils.sessiond_spawn(runtime_tools)

        # Create session using mi to get path and session name
        runtime_tools.run('lttng create trace --output={}'.format(trace_path))

        runtime_tools.run('lttng enable-event -u tp:tptest')
        runtime_tools.run('lttng start')

        # Run application
        cmd = './app {}'.format(nb_events)
        runtime_app.run(cmd, cwd=app_path)

        # Stop tracing
        runtime_tools.run('lttng destroy -a')
        cp = runtime_tools.subprocess_terminate(sessiond)
        if cp.returncode != 0:
            pytest.fail("Sessiond return code")

        try:
            # Read trace with babeltrace and check for event count via number of line
            cmd = 'babeltrace {}'.format(trace_path)
            cp_process, cp_out, cp_err = runtime_tools.run(cmd)
            assert(utils.line_count(cp_out) == nb_events)
        except subprocess.CalledProcessError as e:
            # Check if we expected a problem here
            if should_trace:
                # Something happened
                raise e


@pytest.mark.parametrize("ust_label,tools_label, success", runtime_matrix_regen_statedump)
def test_ust_app_regen_statedump(tmpdir, ust_label, tools_label, success):
    nb_events = 100

    if success:
        expected_events = 4
    else:
        expected_events = 2

    ust = ProjectFactory.get_precook(ust_label)
    tools = ProjectFactory.get_precook(tools_label)
    babeltrace = ProjectFactory.get_precook(Settings.default_babeltrace)

    tools_runtime_path = os.path.join(str(tmpdir), "tools")
    ust_runtime_path = os.path.join(str(tmpdir), "ust")
    app_path = os.path.join(str(tmpdir), "app")


    app_sync_start = os.path.join(app_path, 'sync_start')
    app_sync_end = os.path.join(app_path, 'sync_end')

    with Run.get_runtime(ust_runtime_path) as runtime_app, Run.get_runtime(tools_runtime_path) as runtime_tools:
        runtime_tools.add_project(tools)
        runtime_tools.add_project(babeltrace)

        runtime_app.add_project(ust)
        runtime_app.lttng_home = runtime_tools.lttng_home

        trace_path = os.path.join(runtime_tools.lttng_home, 'trace')

        # Make application using the ust runtime
        shutil.copytree(Settings.apps_gen_events_folder, app_path)
        runtime_app.run("make V=1", cwd=app_path)

        # Start lttng-sessiond
        sessiond = utils.sessiond_spawn(runtime_tools)

        # Create session using mi to get path and session name
        runtime_tools.run('lttng create trace --output={}'.format(trace_path))

        runtime_tools.run('lttng enable-event -u lttng_ust_statedump:start,lttng_ust_statedump:end')
        runtime_tools.run('lttng start')

        # Run application
        cmd = './app {} 0 {} {}'.format(nb_events, app_sync_start, app_sync_end)
        runtime_app.spawn_subprocess(cmd, cwd=app_path)

        utils.wait_for_file(app_sync_start)

        if not success:
            with pytest.raises(subprocess.CalledProcessError):
                runtime_tools.run('lttng regenerate statedump')
        else:
            runtime_tools.run('lttng regenerate statedump')

        utils.create_empty_file(app_sync_end)

        # Stop tracing
        runtime_tools.run('lttng destroy -a')
        cp = runtime_tools.subprocess_terminate(sessiond)
        if cp.returncode != 0:
            pytest.fail("Sessiond return code")

        # Read trace with babeltrace and check for event count via number of line
        cmd = 'babeltrace {}'.format(trace_path)
        cp_process, cp_out, cp_err = runtime_tools.run(cmd)
        assert(utils.line_count(cp_out) == expected_events)

@pytest.mark.parametrize("ust_label,tools_label, scenario", runtime_matrix_starglobing_enabler)
def test_ust_app_starglobing_enabler(tmpdir, ust_label, tools_label, scenario):

    nb_events = 100

    if scenario == "Unsupported by ust":
        expected_events = 0
    else:
        expected_events = nb_events

    # Prepare environment
    ust = ProjectFactory.get_precook(ust_label)
    tools = ProjectFactory.get_precook(tools_label)
    babeltrace = ProjectFactory.get_precook(Settings.default_babeltrace)

    tools_runtime_path = os.path.join(str(tmpdir), "tools")
    ust_runtime_path = os.path.join(str(tmpdir), "ust")
    app_path = os.path.join(str(tmpdir), "app")

    with Run.get_runtime(ust_runtime_path) as runtime_app, Run.get_runtime(tools_runtime_path) as runtime_tools:
        runtime_tools.add_project(tools)
        runtime_tools.add_project(babeltrace)

        runtime_app.add_project(ust)
        runtime_app.lttng_home = runtime_tools.lttng_home

        trace_path = os.path.join(runtime_tools.lttng_home, 'trace')

        # Make application using the ust runtime
        shutil.copytree(Settings.apps_gen_events_folder, app_path)
        runtime_app.run("make V=1", cwd=app_path)

        # Start lttng-sessiond
        sessiond = utils.sessiond_spawn(runtime_tools)

        # Create session using mi to get path and session name
        runtime_tools.run('lttng create trace --output={}'.format(trace_path))

        # If unsupported by tools simply finish early
        if scenario == "Unsupported by tools":
            with pytest.raises(subprocess.CalledProcessError):
                runtime_tools.run('lttng enable-event -u "tp:*te*"')

            # TODO move this to internal runtime...
            cp = runtime_tools.subprocess_terminate(sessiond)
            if cp.returncode != 0:
                pytest.fail("Sessiond return code")
            return
        else:
            runtime_tools.run('lttng enable-event -u "tp:*te*"')

        runtime_tools.run('lttng start')

        # Run application
        cmd = './app {}'.format(nb_events)
        runtime_app.run(cmd, cwd=app_path)

        # Stop tracing
        runtime_tools.run('lttng destroy -a')
        cp = runtime_tools.subprocess_terminate(sessiond)
        if cp.returncode != 0:
            pytest.fail("Sessiond return code")

            # Read trace with babeltrace and check for event count via number of line
        cmd = 'babeltrace {}'.format(trace_path)
        cp_process, cp_out, cp_err = runtime_tools.run(cmd)
        assert(utils.line_count(cp_out) == expected_events)