niklas/tracetools_analysis
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix fake addr field in ProfileHandler test

Browse source
This commit is contained in:
Christophe Bedard 2019-08-09 13:26:12 +02:00
parent 6b6ef4d2cb
commit 498b9f4d15
1 changed files with 14 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

18
tracetools_analysis/test/tracetools_analysis/test_profile_handler.py
View file

@ -270,6 +270,13 @@ expected = [
] ]
address_to_func = {
'0xfA': '0xfA',
'0xfAA': '0xfAA',
'0xfB': '0xfB',
}
class TestProfileHandler(unittest.TestCase): class TestProfileHandler(unittest.TestCase):
def __init__(self, *args) -> None: def __init__(self, *args) -> None:
@ -292,16 +299,19 @@ class TestProfileHandler(unittest.TestCase):
return expected_df.append(expected_data, ignore_index=True) return expected_df.append(expected_data, ignore_index=True)
@staticmethod @staticmethod
def add_fake_fields(events: List[DictEvent]) -> None: def transform_fake_fields(events: List[DictEvent]) -> None:
# Actual value does not matter here; it just needs to be there
for event in events: for event in events:
# Actual value does not matter here; it just needs to be there
event['cpu_id'] = 69 event['cpu_id'] = 69
if event['_name'] == 'lttng_ust_cyg_profile_fast:func_entry':
# The 'addr' field is supposed to be an int
event['addr'] = ProfileHandler.addr_to_int(event['addr'])
@classmethod @classmethod
def setUpClass(cls): def setUpClass(cls):
cls.add_fake_fields(input_events) cls.transform_fake_fields(input_events)
cls.expected = cls.build_expected_df(expected) cls.expected = cls.build_expected_df(expected)
cls.handler = ProfileHandler() cls.handler = ProfileHandler(address_to_func=address_to_func)
cls.processor = Processor(cls.handler) cls.processor = Processor(cls.handler)
cls.processor.process(input_events) cls.processor.process(input_events)