Update Trace action test

This commit is contained in:
Christophe Bedard 2019-12-27 13:23:43 -05:00
parent 99a656a5c6
commit be9e54b629

View file

@ -15,6 +15,7 @@
import unittest import unittest
from tracetools_launch.action import Trace from tracetools_launch.action import Trace
from tracetools_launch.actions.ld_preload import LdPreload
class TestTraceAction(unittest.TestCase): class TestTraceAction(unittest.TestCase):
@ -48,9 +49,34 @@ class TestTraceAction(unittest.TestCase):
for events in events_lists_no_match: for events in events_lists_no_match:
self.assertFalse(Trace.has_profiling_events(events)) self.assertFalse(Trace.has_profiling_events(events))
def test_has_ust_memory_events(self) -> None:
events_lists_match = [
[
'hashtag:yopo',
'lttng_ust_libc:malloc',
'lttng_ust_libc:realloc',
],
[
'lttng_ust_libc:still_a_match',
],
]
events_lists_no_match = [
[],
[
'my_random:event',
'lttng_ust_whatever'
]
]
for events in events_lists_match:
self.assertTrue(Trace.has_ust_memory_events(events))
for events in events_lists_no_match:
self.assertFalse(Trace.has_ust_memory_events(events))
def test_get_shared_lib_path(self) -> None: def test_get_shared_lib_path(self) -> None:
# Only test not finding a lib for now # Only test not finding a lib for now
self.assertIsNone(Trace.get_shared_lib_path('random_lib_that_does_not_exist_I_hope.so')) self.assertIsNone(
LdPreload.get_shared_lib_path('random_lib_that_does_not_exist_I_hope.so')
)
if __name__ == '__main__': if __name__ == '__main__':