| #!/usr/bin/env vpython3 |
| # Copyright 2025 The Chromium Authors |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| import argparse |
| import pathlib |
| import tempfile |
| import unittest |
| import unittest.mock |
| |
| import generate_orderfile |
| |
| |
| class GenerateOrderfileTest(unittest.TestCase): |
| |
| def setUp(self): |
| self.temp_dir = tempfile.TemporaryDirectory() |
| self.mock_out_dir = pathlib.Path(self.temp_dir.name) / 'out' |
| self.mock_out_dir.mkdir() |
| |
| self.mock_args = argparse.Namespace(android_browser='test-browser', |
| out_dir=self.mock_out_dir, |
| arch='arm64', |
| profile_webview=False, |
| streamline_for_debugging=False, |
| save_profile_data=False, |
| verbosity=0, |
| isolated_script_test_output=None) |
| |
| self.profile_tool_patch = unittest.mock.patch( |
| 'android_profile_tool.AndroidProfileTool', autospec=True) |
| self.mock_profile_tool_cls = self.profile_tool_patch.start() |
| self.mock_profile_tool = self.mock_profile_tool_cls.return_value |
| |
| self.get_libchrome_patch = unittest.mock.patch( |
| 'orderfile_shared.GetLibchromeSoPath', |
| return_value=str(self.mock_out_dir / 'lib.unstripped/libchrome.so')) |
| self.mock_get_libchrome = self.get_libchrome_patch.start() |
| |
| self.collect_profiles_patch = unittest.mock.patch( |
| 'orderfile_shared.CollectProfiles', return_value=['profile1.prof']) |
| self.mock_collect_profiles = self.collect_profiles_patch.start() |
| |
| self.process_profiles_patch = unittest.mock.patch( |
| 'orderfile_shared.ProcessProfiles', |
| return_value=(['symbol1', 'symbol2'], 100)) |
| self.mock_process_profiles = self.process_profiles_patch.start() |
| |
| self.add_dummy_funcs_patch = unittest.mock.patch( |
| 'orderfile_shared.AddDummyFunctions') |
| self.mock_add_dummy_funcs = self.add_dummy_funcs_patch.start() |
| |
| def tearDown(self): |
| self.temp_dir.cleanup() |
| self.profile_tool_patch.stop() |
| self.get_libchrome_patch.stop() |
| self.collect_profiles_patch.stop() |
| self.process_profiles_patch.stop() |
| self.add_dummy_funcs_patch.stop() |
| |
| def test_GenerateOrderfile(self): |
| generate_orderfile.GenerateOrderfile(self.mock_args, 'device1') |
| |
| self.mock_profile_tool_cls.assert_called_once_with(str(self.mock_out_dir / |
| 'profile_data'), |
| 'device1', |
| debug=False, |
| verbosity=0) |
| |
| self.mock_get_libchrome.assert_called_once_with(self.mock_out_dir, |
| 'arm64', False) |
| self.mock_collect_profiles.assert_called_once_with(self.mock_profile_tool, |
| False, 'arm64', |
| 'test-browser', |
| str(self.mock_out_dir), |
| None) |
| self.mock_process_profiles.assert_called_once_with( |
| ['profile1.prof'], |
| str(self.mock_out_dir / 'lib.unstripped/libchrome.so')) |
| |
| unpatched_orderfile = (self.mock_out_dir / |
| 'orderfiles/unpatched_orderfile.arm64') |
| self.assertTrue(unpatched_orderfile.exists()) |
| with open(unpatched_orderfile) as f: |
| self.assertEqual(f.read(), 'symbol1\nsymbol2') |
| |
| self.mock_profile_tool.Cleanup.assert_called_once() |
| |
| patched_orderfile = self.mock_out_dir / 'orderfiles/orderfile.arm64.out' |
| self.mock_add_dummy_funcs.assert_called_once_with(str(unpatched_orderfile), |
| str(patched_orderfile)) |
| |
| def test_GetOrderfilesDir(self): |
| orderfiles_dir = generate_orderfile._GetOrderfilesDir(self.mock_args) |
| self.assertEqual(orderfiles_dir, self.mock_out_dir / 'orderfiles') |
| self.assertTrue(orderfiles_dir.exists()) |
| |
| def test_GetOrderfileFilename(self): |
| filename = generate_orderfile._GetOrderfileFilename(self.mock_args) |
| expected_path = self.mock_out_dir / 'orderfiles/orderfile.arm64.out' |
| self.assertEqual(filename, str(expected_path)) |
| |
| def test_GetUnpatchedOrderfileFilename(self): |
| filename = generate_orderfile._GetUnpatchedOrderfileFilename(self.mock_args) |
| expected_path = self.mock_out_dir / 'orderfiles/unpatched_orderfile.arm64' |
| self.assertEqual(filename, str(expected_path)) |
| |
| |
| if __name__ == '__main__': |
| unittest.main() |
