| #!/usr/bin/env python3 |
| # |
| # Copyright 2019, 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. |
| |
| """Unittests for native_util.""" |
| |
| import os |
| import unittest |
| from unittest import mock |
| |
| from aidegen import unittest_constants |
| from aidegen.lib import common_util |
| from aidegen.lib import native_module_info |
| from aidegen.lib import native_util |
| from aidegen.lib import project_info |
| |
| |
| # pylint: disable=protected-access |
| # pylint: disable=too-many-arguments |
| class AidegenNativeUtilUnittests(unittest.TestCase): |
| """Unit tests for native_util.py""" |
| |
| @mock.patch.object(native_util, '_check_native_project_exists') |
| @mock.patch.object(common_util, 'check_java_or_kotlin_file_exists') |
| @mock.patch.object(common_util, 'get_related_paths') |
| def test_analyze_native_and_java_projects( |
| self, mock_get_related, mock_check_java, mock_check_native): |
| """Test analyze_native_and_java_projects function.""" |
| mock_get_related.return_value = None, None |
| mock_check_java.return_value = True |
| mock_check_native.return_value = True |
| targets = ['a'] |
| self.assertEqual((targets, targets), |
| native_util._analyze_native_and_java_projects( |
| None, None, targets)) |
| mock_check_native.return_value = False |
| self.assertEqual((targets, []), |
| native_util._analyze_native_and_java_projects( |
| None, None, targets)) |
| mock_check_java.return_value = False |
| mock_check_native.return_value = True |
| self.assertEqual(([], targets), |
| native_util._analyze_native_and_java_projects( |
| None, None, targets)) |
| |
| def test_check_native_project_exists(self): |
| """Test _check_native_project_exists function.""" |
| rel_path = 'a/b' |
| path_to_module_info = {'a/b/c': {}} |
| self.assertTrue( |
| native_util._check_native_project_exists(path_to_module_info, |
| rel_path)) |
| rel_path = 'a/b/c/d' |
| self.assertFalse( |
| native_util._check_native_project_exists(path_to_module_info, |
| rel_path)) |
| |
| def test_find_parent(self): |
| """Test _find_parent function with conditions.""" |
| current_parent = None |
| abs_path = 'a/b/c/d' |
| expected = abs_path |
| result = native_util._find_parent(abs_path, current_parent) |
| self.assertEqual(result, expected) |
| current_parent = 'a/b/c/d/e' |
| result = native_util._find_parent(abs_path, current_parent) |
| self.assertEqual(result, expected) |
| current_parent = 'a/b/c' |
| expected = current_parent |
| result = native_util._find_parent(abs_path, current_parent) |
| self.assertEqual(result, expected) |
| current_parent = 'a/b/f' |
| expected = 'a/b' |
| result = native_util._find_parent(abs_path, current_parent) |
| self.assertEqual(result, expected) |
| |
| @mock.patch.object(native_module_info.NativeModuleInfo, |
| '_load_module_info_file') |
| @mock.patch.object(native_util, '_find_parent') |
| @mock.patch.object(common_util, 'get_related_paths') |
| def test_get_merged_native_target_is_module( |
| self, mock_get_related, mock_find_parent, mock_load_info): |
| """Test _get_merged_native_target function if the target is a module.""" |
| mock_get_related.return_value = 'c/d', 'a/b/c/d' |
| parent = 'a/b' |
| mock_find_parent.return_value = parent |
| targets = ['multiarch'] |
| expected = (parent, targets) |
| mock_load_info.return_value = ( |
| None, unittest_constants.CC_NAME_TO_MODULE_INFO) |
| cc_mod_info = native_module_info.NativeModuleInfo() |
| new_parent, new_targets = native_util._get_merged_native_target( |
| cc_mod_info, targets) |
| result = (new_parent, new_targets) |
| self.assertEqual(result, expected) |
| |
| @mock.patch.object(native_module_info.NativeModuleInfo, |
| '_load_module_info_file') |
| @mock.patch.object(native_util, '_find_parent') |
| @mock.patch.object(common_util, 'get_related_paths') |
| def test_get_merged_native_target_is_path(self, mock_get_related, |
| mock_find_parent, mock_load_info): |
| """Test _get_merged_native_target function if the target is a path.""" |
| parent = 'a/b' |
| rel_path = 'shared/path/to/be/used2' |
| mock_get_related.return_value = rel_path, os.path.join(parent, rel_path) |
| mock_find_parent.return_value = parent |
| mock_load_info.return_value = ( |
| None, unittest_constants.CC_NAME_TO_MODULE_INFO) |
| targets = [rel_path] |
| result_targets = unittest_constants.TESTABLE_MODULES_WITH_SHARED_PATH |
| expected = (parent, result_targets) |
| cc_mod_info = native_module_info.NativeModuleInfo() |
| new_parent, new_targets = native_util._get_merged_native_target( |
| cc_mod_info, targets) |
| result = (new_parent, new_targets) |
| self.assertEqual(result, expected) |
| |
| def test_filter_out_modules(self): |
| """Test _filter_out_modules with conditions.""" |
| targets = ['shared/path/to/be/used2'] |
| result = ([], targets) |
| self.assertEqual( |
| result, native_util._filter_out_modules(targets, lambda x: False)) |
| targets = ['multiarch'] |
| result = (targets, []) |
| self.assertEqual( |
| result, native_util._filter_out_modules(targets, lambda x: True)) |
| |
| @mock.patch.object(native_util, '_filter_out_rust_projects') |
| @mock.patch.object(native_util, '_analyze_native_and_java_projects') |
| @mock.patch.object(native_util, '_filter_out_modules') |
| def test_get_java_cc_and_rust_projects(self, mock_fil, mock_ana, |
| mock_fil_rust): |
| """Test get_java_cc_and_rust_projects handling.""" |
| targets = ['multiarch'] |
| mock_fil_rust.return_value = [] |
| mock_fil.return_value = [], targets |
| cc_mod_info = mock.Mock() |
| cc_mod_info.is_module = mock.Mock() |
| cc_mod_info.is_module.return_value = True |
| at_mod_info = mock.Mock() |
| at_mod_info.is_module = mock.Mock() |
| at_mod_info.is_module.return_value = True |
| mock_ana.return_value = [], targets |
| native_util.get_java_cc_and_rust_projects( |
| at_mod_info, cc_mod_info, targets) |
| self.assertEqual(mock_fil.call_count, 2) |
| self.assertEqual(mock_ana.call_count, 1) |
| |
| @mock.patch.object(native_util, '_get_rust_targets') |
| @mock.patch.object(common_util, 'get_json_dict') |
| @mock.patch('builtins.print') |
| @mock.patch('os.path.isfile') |
| @mock.patch('os.path.join') |
| @mock.patch.object(common_util, 'get_blueprint_json_path') |
| @mock.patch.object(common_util, 'get_android_root_dir') |
| def test_filter_out_rust_projects(self, mock_get_root, mock_get_json, |
| mock_join, mock_is_file, mock_print, |
| mock_get_dict, mock_get_rust): |
| """Test _filter_out_rust_projects with conditions.""" |
| mock_is_file.return_value = False |
| native_util._filter_out_rust_projects(['a/b/rust']) |
| self.assertTrue(mock_get_root.called) |
| self.assertTrue(mock_get_json.called) |
| self.assertTrue(mock_join.called) |
| self.assertTrue(mock_print.called) |
| self.assertFalse(mock_get_dict.called) |
| self.assertFalse(mock_get_rust.called) |
| |
| mock_get_root.mock_reset() |
| mock_get_json.mock_reset() |
| mock_join.mock_reset() |
| mock_print.mock_reset() |
| mock_get_dict.mock_reset() |
| mock_get_rust.mock_reset() |
| mock_is_file.return_value = True |
| mock_get_dict.return_value = {} |
| native_util._filter_out_rust_projects(['a/b/rust']) |
| self.assertTrue(mock_get_root.called) |
| self.assertTrue(mock_get_json.called) |
| self.assertTrue(mock_join.called) |
| self.assertTrue(mock_print.called) |
| self.assertFalse(mock_get_rust.called) |
| |
| mock_get_root.mock_reset() |
| mock_get_json.mock_reset() |
| mock_join.mock_reset() |
| mock_print.mock_reset() |
| mock_get_rust.mock_reset() |
| mock_is_file.return_value = True |
| crates = [{native_util._ROOT_MODULE: 'a/b/rust/src'}] |
| mock_get_dict.return_value = {native_util._CRATES_KEY: crates} |
| mock_get_root.return_value = 'a/b' |
| native_util._filter_out_rust_projects(['a/b/rust']) |
| self.assertTrue(mock_get_json.called) |
| self.assertTrue(mock_join.called) |
| self.assertTrue(mock_get_rust.called) |
| mock_get_rust.assert_called_with(['a/b/rust'], crates, 'a/b') |
| |
| @mock.patch.object(project_info, 'batch_build_dependencies') |
| @mock.patch.object(common_util, 'is_source_under_relative_path') |
| @mock.patch('os.path.isdir') |
| def test_get_rust_targets(self, mock_is_dir, mock_is_under, mock_rebuilds): |
| """Test _get_rust_targets with conditions.""" |
| mock_is_dir.return_value = True |
| mock_is_under.return_value = True |
| display_name = 'rust_module' |
| mod_info = [ |
| { |
| native_util._DISPLAY_NAME: display_name, |
| native_util._ROOT_MODULE: 'a/b/rust/src' |
| } |
| ] |
| targets = ['a/b/rust'] |
| self.assertEqual( |
| targets, |
| native_util._get_rust_targets(targets, mod_info, 'a/b')) |
| mock_rebuilds.assert_called_with({display_name}) |
| |
| def test_get_relative_path(self): |
| """Test _get_relative_path with conditions.""" |
| root = common_util.get_android_root_dir() |
| cwd = os.getcwd() |
| rel_target = os.path.relpath(cwd, root) |
| self.assertEqual(rel_target, native_util._get_relative_path('.', root)) |
| |
| root = 'a/b' |
| target = 'a/b/rust' |
| rel_target = 'rust' |
| self.assertEqual( |
| rel_target, native_util._get_relative_path(target, root)) |
| |
| def test_is_target_relative_module(self): |
| """Test _is_target_relative_module with conditions.""" |
| path = 'a/b' |
| target = 'a/b' |
| self.assertTrue( |
| native_util._is_target_relative_module(path, target)) |
| |
| path = 'a/b/c' |
| self.assertTrue( |
| native_util._is_target_relative_module(path, target)) |
| |
| path = 'out/a/b/c' |
| self.assertTrue( |
| native_util._is_target_relative_module(path, target)) |
| |
| target = 'a/bc' |
| self.assertFalse( |
| native_util._is_target_relative_module(path, target)) |
| |
| |
| if __name__ == '__main__': |
| unittest.main() |
