1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
#!/usr/bin/env python3
#
# Copyright (C) 2021 The Android Open Source Project
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import collections
import json
from typing import Any, Dict, List
from . test_utils import TestBase, TestHelper
class TestReportHtml(TestBase):
def test_long_callchain(self):
self.run_cmd(['report_html.py', '-i',
TestHelper.testdata_path('perf_with_long_callchain.data')])
def test_aggregated_by_thread_name(self):
# Calculate event_count for each thread name before aggregation.
event_count_for_thread_name = collections.defaultdict(lambda: 0)
# use "--min_func_percent 0" to avoid cutting any thread.
record_data = self.get_record_data(['--min_func_percent', '0', '-i',
TestHelper.testdata_path('aggregatable_perf1.data'),
TestHelper.testdata_path('aggregatable_perf2.data')])
event = record_data['sampleInfo'][0]
for process in event['processes']:
for thread in process['threads']:
thread_name = record_data['threadNames'][str(thread['tid'])]
event_count_for_thread_name[thread_name] += thread['eventCount']
# Check event count for each thread after aggregation.
record_data = self.get_record_data(['--aggregate-by-thread-name',
'--min_func_percent', '0', '-i',
TestHelper.testdata_path('aggregatable_perf1.data'),
TestHelper.testdata_path('aggregatable_perf2.data')])
event = record_data['sampleInfo'][0]
hit_count = 0
for process in event['processes']:
for thread in process['threads']:
thread_name = record_data['threadNames'][str(thread['tid'])]
self.assertEqual(thread['eventCount'],
event_count_for_thread_name[thread_name])
hit_count += 1
self.assertEqual(hit_count, len(event_count_for_thread_name))
def test_no_empty_process(self):
""" Test not showing a process having no threads. """
perf_data = TestHelper.testdata_path('two_process_perf.data')
record_data = self.get_record_data(['-i', perf_data])
processes = record_data['sampleInfo'][0]['processes']
self.assertEqual(len(processes), 2)
# One process is removed because all its threads are removed for not
# reaching the min_func_percent limit.
record_data = self.get_record_data(['-i', perf_data, '--min_func_percent', '20'])
processes = record_data['sampleInfo'][0]['processes']
self.assertEqual(len(processes), 1)
def test_proguard_mapping_file(self):
""" Test --proguard-mapping-file option. """
testdata_file = TestHelper.testdata_path('perf_need_proguard_mapping.data')
proguard_mapping_file = TestHelper.testdata_path('proguard_mapping.txt')
original_methodname = 'androidx.fragment.app.FragmentActivity.startActivityForResult'
# Can't show original method name without proguard mapping file.
record_data = self.get_record_data(['-i', testdata_file])
self.assertNotIn(original_methodname, json.dumps(record_data))
# Show original method name with proguard mapping file.
record_data = self.get_record_data(
['-i', testdata_file, '--proguard-mapping-file', proguard_mapping_file])
self.assertIn(original_methodname, json.dumps(record_data))
def get_record_data(self, options: List[str]) -> Dict[str, Any]:
self.run_cmd(['report_html.py'] + options)
with open('report.html', 'r') as fh:
data = fh.read()
start_str = 'type="application/json"'
end_str = '</script>'
start_pos = data.find(start_str)
self.assertNotEqual(start_pos, -1)
start_pos = data.find('>', start_pos)
self.assertNotEqual(start_pos, -1)
start_pos += 1
end_pos = data.find(end_str, start_pos)
self.assertNotEqual(end_pos, -1)
json_data = data[start_pos:end_pos]
return json.loads(json_data)
|
