Add subscription creation test
This commit is contained in:
parent
6caaa51764
commit
cc8187e55b
5 changed files with 111 additions and 2 deletions
57
tracetools_test/test/test_subscription.py
Normal file
57
tracetools_test/test/test_subscription.py
Normal file
|
@ -0,0 +1,57 @@
|
|||
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,
|
||||
)
|
||||
|
||||
PKG = 'tracetools_test'
|
||||
|
||||
subscription_creation_events = [
|
||||
'ros2:rcl_subscription_init',
|
||||
'ros2:rclcpp_subscription_callback_added',
|
||||
]
|
||||
|
||||
class TestSubscription(unittest.TestCase):
|
||||
|
||||
def test_creation(self):
|
||||
session_name = f'session-test-subscription-creation-{time.strftime("%Y%m%d%H%M%S")}'
|
||||
path = '/tmp/' + session_name
|
||||
print(f'trace directory: {path}')
|
||||
|
||||
lttng_setup(session_name, path, ros_events=subscription_creation_events, kernel_events=None)
|
||||
lttng_start(session_name)
|
||||
|
||||
ld = LaunchDescription([
|
||||
launch_ros.actions.Node(
|
||||
package=PKG, node_executable='test_subscription', output='screen'),
|
||||
])
|
||||
ls = LaunchService()
|
||||
ls.include_launch_description(get_default_launch_description())
|
||||
ls.include_launch_description(ld)
|
||||
|
||||
exit_code = ls.run()
|
||||
self.assertEqual(exit_code, 0)
|
||||
|
||||
lttng_stop(session_name)
|
||||
lttng_destroy(session_name)
|
||||
|
||||
trace_events = get_trace_event_names(path)
|
||||
print(f'trace_events: {trace_events}')
|
||||
self.assertListEqual(subscription_creation_events, list(trace_events))
|
||||
|
||||
shutil.rmtree(path)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
Loading…
Add table
Add a link
Reference in a new issue