diff options
Diffstat (limited to 'simpleperf/scripts/test/pprof_proto_generator_test.py')
| -rw-r--r-- | simpleperf/scripts/test/pprof_proto_generator_test.py | 171 |
1 files changed, 2 insertions, 169 deletions
diff --git a/simpleperf/scripts/test/pprof_proto_generator_test.py b/simpleperf/scripts/test/pprof_proto_generator_test.py index cbeb8d6a..dcdf2ca9 100644 --- a/simpleperf/scripts/test/pprof_proto_generator_test.py +++ b/simpleperf/scripts/test/pprof_proto_generator_test.py @@ -14,17 +14,12 @@ # See the License for the specific language governing permissions and # limitations under the License. -from collections import namedtuple import google.protobuf -import os -import re -import tempfile -from typing import List, Optional, Set +from typing import List, Optional from binary_cache_builder import BinaryCacheBuilder -from pprof_proto_generator import load_pprof_profile, PprofProfileGenerator +from pprof_proto_generator import load_pprof_profile from . test_utils import TestBase, TestHelper -from simpleperf_utils import ReportLibOptions class TestPprofProtoGenerator(TestBase): @@ -54,16 +49,6 @@ class TestPprofProtoGenerator(TestBase): self.assertIn(key, self.run_generator(['--pid', '10419', '10416'])) self.assertNotIn(key, self.run_generator(['--pid', '10416'])) - def test_thread_labels(self): - output = self.run_generator() - self.assertIn('label[0] = thread:Binder:10419_1', output) - self.assertIn('label[0] = thread:Binder:10419_2', output) - self.assertIn('label[0] = thread:Binder:10419_3', output) - self.assertIn('label[0] = thread:Binder:10419_4', output) - self.assertIn('label[1] = threadpool:Binder:%d_%d', output) - self.assertIn('label[2] = pid:10419', output) - self.assertIn('label[3] = tid:10459', output) - def test_tid_filter(self): key1 = 'art::ProfileSaver::Run()' # function in thread 10459 key2 = 'PlayScene::DoFrame()' # function in thread 10463 @@ -96,19 +81,6 @@ class TestPprofProtoGenerator(TestBase): """ Test the build ids generated are not padded with zeros. """ self.assertIn('build_id: e3e938cc9e40de2cfe1a5ac7595897de(', self.run_generator()) - def test_build_id_with_binary_cache(self): - """ Test the build ids for elf files in binary_cache are not padded with zero. """ - # Test with binary_cache. - testdata_file = TestHelper.testdata_path('runtest_two_functions_arm64_perf.data') - - # Build binary_cache. - binary_cache_builder = BinaryCacheBuilder(TestHelper.ndk_path, False) - binary_cache_builder.build_binary_cache(testdata_file, [TestHelper.testdata_dir]) - - # Generate profile. - output = self.run_generator(testdata_file=testdata_file) - self.assertIn('build_id: b4f1b49b0fe9e34e78fb14e5374c930c(', output) - def test_location_address(self): """ Test if the address of a location is within the memory range of the corresponding mapping. @@ -120,12 +92,6 @@ class TestPprofProtoGenerator(TestBase): self.assertLessEqual(mapping.memory_start, location.address) self.assertGreaterEqual(mapping.memory_limit, location.address) - def test_sample_type(self): - """Test sample types have the right units.""" - output = self.run_generator() - self.assertIn('type=cpu-cycles_samples, unit=samples', output) - self.assertIn('type=cpu-cycles, unit=cpu-cycles', output) - def test_multiple_perf_data(self): """ Test reporting multiple recording file. """ profile1 = self.generate_profile(None, ['aggregatable_perf1.data']) @@ -158,136 +124,3 @@ class TestPprofProtoGenerator(TestBase): output = self.run_generator(testdata_file=testdata_file) self.assertIn('simpleperf_runtest_two_functions_arm64', output) self.assertIn('two_functions.cpp', output) - - def test_line_info(self): - """ Check line numbers generated in profile. """ - testdata_file = TestHelper.testdata_path('runtest_two_functions_arm64_perf.data') - - # Build binary_cache. - binary_cache_builder = BinaryCacheBuilder(TestHelper.ndk_path, False) - binary_cache_builder.build_binary_cache(testdata_file, [TestHelper.testdata_dir]) - - # Generate profile. - profile = self.generate_profile(None, [testdata_file]) - - CheckItem = namedtuple( - 'CheckItem', ['addr', 'source_file', 'source_line', 'func_name', 'func_start_line']) - - check_items = [ - CheckItem(0x113c, 'two_functions.cpp', 22, 'main', 20), - CheckItem(0x1140, 'two_functions.cpp', 23, 'main', 20), - CheckItem(0x1094, 'two_functions.cpp', 9, 'Function1', 6), - CheckItem(0x1104, 'two_functions.cpp', 16, 'Function2', 13), - ] - mapping = None - for mapping in profile.mapping: - binary_path = profile.string_table[mapping.filename] - if 'runtest_two_functions_arm64' in binary_path: - self.assertTrue(mapping.has_line_numbers) - mapping = mapping - break - self.assertIsNotNone(mapping) - - for check_item in check_items: - found = False - for location in profile.location: - if location.mapping_id != mapping.id: - continue - addr = location.address - mapping.memory_start + mapping.file_offset - if addr == check_item.addr: - found = True - self.assertEqual(len(location.line), 1) - line = location.line[0] - function = profile.function[line.function_id - 1] - self.assertIn(check_item.source_file, profile.string_table[function.filename]) - self.assertEqual(line.line, check_item.source_line) - self.assertIn(check_item.func_name, profile.string_table[function.name]) - self.assertEqual(function.start_line, check_item.func_start_line) - break - self.assertTrue(found, check_item) - - def test_function_name_not_changed_by_line_info(self): - """ Adding line info shouldn't override function names from report library, which are more - accurate when proguard mapping file is given. - """ - testdata_file = TestHelper.testdata_path('runtest_two_functions_arm64_perf.data') - - # Build binary_cache. - binary_cache_builder = BinaryCacheBuilder(TestHelper.ndk_path, False) - binary_cache_builder.build_binary_cache(testdata_file, [TestHelper.testdata_dir]) - - # Read recording file. - config = {'ndk_path': None, 'max_chain_length': 1000000, - 'report_lib_options': ReportLibOptions(False, '', None, None)} - generator = PprofProfileGenerator(config) - generator.load_record_file(testdata_file) - - # Change function name. - sample = generator.sample_list[0] - self.assertGreaterEqual(len(sample.location_ids), 1) - location = generator.location_list[sample.location_ids[0] - 1] - self.assertGreaterEqual(len(location.lines), 1) - function = generator.get_function(location.lines[0].function_id) - function_name = generator.get_string(function.name_id) - self.assertEqual(function_name, 'Function1()') - location.lines[0].function_id = generator.get_function_id( - 'NewFunction1()', generator.get_string(function.dso_name_id), function.vaddr_in_dso) - - # Add line info. - generator.gen_source_lines(1) - - # Check function name and line info. - sample = generator.sample_list[0] - self.assertGreaterEqual(len(sample.location_ids), 1) - location = generator.location_list[sample.location_ids[0] - 1] - self.assertGreaterEqual(len(location.lines), 1) - function = generator.get_function(location.lines[0].function_id) - function_name = generator.get_string(function.name_id) - self.assertEqual(function_name, 'NewFunction1()') - self.assertNotEqual(function.source_filename_id, 0) - source_filename = generator.get_string(function.source_filename_id) - self.assertIn('two_functions.cpp', source_filename) - - def test_comments(self): - profile = self.generate_profile(None, ['perf_with_interpreter_frames.data']) - comments = "\n".join([profile.string_table[i] for i in profile.comment]) - comments = comments.replace('\\', '/') - self.assertIn('Simpleperf Record Command:\n/data/data/com.google.sample.tunnel/simpleperf record --in-app --tracepoint-events /data/local/tmp/tracepoint_events --app com.google.sample.tunnel -g --no-post-unwind --duration 30', comments) - self.assertIn('Converted to pprof with:', comments) - # The full path changes per-machine, so only assert on a subset of the - # path. - self.assertIn('testdata/perf_with_interpreter_frames.data', comments) - self.assertIn('Architecture:\naarch64', comments) - - def test_sample_filters(self): - def get_threads_for_filter(filter: str) -> Set[int]: - report = self.run_generator(filter.split(), testdata_file='perf_display_bitmaps.data') - threads = set() - pattern = re.compile(r'\s+tid:(\d+)') - threads = set() - for m in re.finditer(pattern, report): - threads.add(int(m.group(1))) - return threads - - self.assertNotIn(31850, get_threads_for_filter('--exclude-pid 31850')) - self.assertIn(31850, get_threads_for_filter('--include-pid 31850')) - self.assertIn(31850, get_threads_for_filter('--pid 31850')) - self.assertNotIn(31881, get_threads_for_filter('--exclude-tid 31881')) - self.assertIn(31881, get_threads_for_filter('--include-tid 31881')) - self.assertIn(31881, get_threads_for_filter('--tid 31881')) - self.assertNotIn(31881, get_threads_for_filter( - '--exclude-process-name com.example.android.displayingbitmaps')) - self.assertIn(31881, get_threads_for_filter( - '--include-process-name com.example.android.displayingbitmaps')) - self.assertNotIn(31850, get_threads_for_filter( - '--exclude-thread-name com.example.android.displayingbitmaps')) - self.assertIn(31850, get_threads_for_filter( - '--include-thread-name com.example.android.displayingbitmaps')) - - with tempfile.NamedTemporaryFile('w', delete=False) as filter_file: - filter_file.write('GLOBAL_BEGIN 684943449406175\nGLOBAL_END 684943449406176') - filter_file.flush() - threads = get_threads_for_filter('--filter-file ' + filter_file.name) - self.assertIn(31881, threads) - self.assertNotIn(31850, threads) - os.unlink(filter_file.name) |