Extract common test utils

tracetools_test/test/test_publisher.py

@@ -1,18 +1,8 @@
-import time
-import shutil
-from launch import LaunchDescription
-from launch import LaunchIntrospector
-from launch import LaunchService
-
-from launch_ros import get_default_launch_description
-import launch_ros.actions
 import unittest
-from tracetools_analysis.test.utils import get_trace_event_names
-from tracetools_trace.tools.lttng import (
-    lttng_setup,
-    lttng_start,
-    lttng_stop,
-    lttng_destroy,
+from tracetools_test.utils import (
+    get_trace_event_names,
+    run_and_trace,
+    cleanup_trace,
 )
 
 PKG = 'tracetools_test'
@@ -24,32 +14,18 @@ publisher_creation_events = [
 class TestPublisher(unittest.TestCase):
 
     def test_creation(self):
-        session_name = f'session-test-publisher-creation-{time.strftime("%Y%m%d%H%M%S")}'
-        path = '/tmp/' + session_name
-        print(f'trace directory: {path}')
+        session_name_prefix = 'session-test-publisher-creation'
+        base_path = '/tmp'
+        test_node = 'test_publisher'
 
-        lttng_setup(session_name, path, ros_events=publisher_creation_events, kernel_events=None)
-        lttng_start(session_name)
-
-        ld = LaunchDescription([
-            launch_ros.actions.Node(
-                package=PKG, node_executable='test_publisher', output='screen'),
-        ])
-        ls = LaunchService()
-        ls.include_launch_description(get_default_launch_description())
-        ls.include_launch_description(ld)
-
-        exit_code = ls.run()
+        exit_code, full_path = run_and_trace(session_name_prefix, base_path, publisher_creation_events, None, PKG, test_node)
         self.assertEqual(exit_code, 0)
 
-        lttng_stop(session_name)
-        lttng_destroy(session_name)
-
-        trace_events = get_trace_event_names(path)
+        trace_events = get_trace_event_names(full_path)
         print(f'trace_events: {trace_events}')
-        self.assertListEqual(publisher_creation_events, list(trace_events))
+        self.assertSetEqual(set(publisher_creation_events), trace_events)
 
-        shutil.rmtree(path)
+        cleanup_trace(full_path)
 
 
 if __name__ == '__main__':