ros2_tracing/tracetools_test/test/test_subscription.py
2019-06-03 09:45:36 +02:00

33 lines
902 B
Python

import unittest
from tracetools_test.utils import (
get_trace_event_names,
run_and_trace,
cleanup_trace,
)
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_prefix = 'session-test-subscription-creation'
base_path = '/tmp'
test_node = 'test_subscription'
exit_code, full_path = run_and_trace(session_name_prefix, base_path, subscription_creation_events, None, PKG, test_node)
self.assertEqual(exit_code, 0)
trace_events = get_trace_event_names(full_path)
print(f'trace_events: {trace_events}')
self.assertSetEqual(set(subscription_creation_events), trace_events)
cleanup_trace(full_path)
if __name__ == '__main__':
unittest.main()