| #!/usr/bin/env python3 |
| # |
| # Copyright 2018, 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 module_info_utils.""" |
| |
| import copy |
| import os |
| import subprocess |
| import unittest |
| from unittest import mock |
| |
| import aidegen.unittest_constants as uc |
| from aidegen.lib import errors |
| from aidegen.lib import metrics |
| from aidegen.lib import module_info_util |
| from atest import module_info |
| |
| |
| _TEST_CLASS_DICT = {'class': ['JAVA_LIBRARIES']} |
| _TEST_SRCS_BAR_DICT = {'srcs': ['Bar']} |
| _TEST_SRCS_BAZ_DICT = {'srcs': ['Baz']} |
| _TEST_DEP_FOO_DIST = {'dependencies': ['Foo']} |
| _TEST_DEP_SRC_DICT = {'dependencies': ['Foo'], 'srcs': ['Bar']} |
| _TEST_DEP_SRC_MORE_DICT = {'dependencies': ['Foo'], 'srcs': ['Baz', 'Bar']} |
| _TEST_MODULE_A_DICT = { |
| 'module_a': { |
| 'class': ['JAVA_LIBRARIES'], |
| 'path': ['path_a'], |
| 'installed': ['out/path_a/a.jar'], |
| 'dependencies': ['Foo'], |
| 'srcs': ['Bar'] |
| } |
| } |
| _TEST_MODULE_A_DICT_HAS_NONEED_ITEMS = { |
| 'module_a': { |
| 'class': ['JAVA_LIBRARIES'], |
| 'path': ['path_a'], |
| 'installed': ['out/path_a/a.jar'], |
| 'dependencies': ['Foo'], |
| 'srcs': ['Bar'], |
| 'compatibility_suites': ['null-suite'], |
| 'module_name': ['ltp_fstat03_64'] |
| } |
| } |
| _TEST_MODULE_A_JOIN_PATH_DICT = { |
| 'module_a': { |
| 'class': ['JAVA_LIBRARIES'], |
| 'path': ['path_a'], |
| 'installed': ['out/path_a/a.jar'], |
| 'dependencies': ['Foo'], |
| 'srcs': ['path_a/Bar'] |
| } |
| } |
| |
| |
| # pylint: disable=protected-access |
| # pylint: disable=invalid-name |
| class AidegenModuleInfoUtilUnittests(unittest.TestCase): |
| """Unit tests for moduole_info_utils.py""" |
| |
| def test_merge_module_keys_with_empty_dict(self): |
| """Test _merge_module_keys with an empty dictionary.""" |
| test_b_dict = {} |
| test_m_dict = copy.deepcopy(_TEST_DEP_SRC_DICT) |
| module_info_util._merge_module_keys(test_m_dict, test_b_dict) |
| self.assertEqual(_TEST_DEP_SRC_DICT, test_m_dict) |
| |
| def test_merge_module_keys_with_key_not_in_orginial_dict(self): |
| """Test _merge_module_keys with the key does not exist in the dictionary |
| to be merged into. |
| """ |
| test_b_dict = _TEST_SRCS_BAR_DICT |
| test_m_dict = copy.deepcopy(_TEST_DEP_FOO_DIST) |
| module_info_util._merge_module_keys(test_m_dict, test_b_dict) |
| self.assertEqual(_TEST_DEP_SRC_DICT, test_m_dict) |
| |
| def test_merge_module_keys_with_key_in_orginial_dict(self): |
| """Test _merge_module_keys with with the key exists in the dictionary |
| to be merged into. |
| """ |
| test_b_dict = _TEST_SRCS_BAZ_DICT |
| test_m_dict = copy.deepcopy(_TEST_DEP_SRC_DICT) |
| module_info_util._merge_module_keys(test_m_dict, test_b_dict) |
| self.assertEqual( |
| set(_TEST_DEP_SRC_MORE_DICT['srcs']), set(test_m_dict['srcs'])) |
| self.assertEqual( |
| set(_TEST_DEP_SRC_MORE_DICT['dependencies']), |
| set(test_m_dict['dependencies'])) |
| |
| def test_merge_module_keys_with_duplicated_item_dict(self): |
| """Test _merge_module_keys with with the key exists in the dictionary |
| to be merged into. |
| """ |
| test_b_dict = _TEST_CLASS_DICT |
| test_m_dict = copy.deepcopy(_TEST_CLASS_DICT) |
| module_info_util._merge_module_keys(test_m_dict, test_b_dict) |
| self.assertEqual(_TEST_CLASS_DICT, test_m_dict) |
| |
| def test_copy_needed_items_from_empty_dict(self): |
| """Test _copy_needed_items_from an empty dictionary.""" |
| test_mk_dict = {} |
| want_dict = {} |
| self.assertEqual(want_dict, |
| module_info_util._copy_needed_items_from(test_mk_dict)) |
| |
| def test_copy_needed_items_from_all_needed_items_dict(self): |
| """Test _copy_needed_items_from a dictionary with all needed items.""" |
| self.assertEqual( |
| _TEST_MODULE_A_DICT, |
| module_info_util._copy_needed_items_from(_TEST_MODULE_A_DICT)) |
| |
| def test_copy_needed_items_from_some_needed_items_dict(self): |
| """Test _copy_needed_items_from a dictionary with some needed items.""" |
| self.assertEqual( |
| _TEST_MODULE_A_DICT, |
| module_info_util._copy_needed_items_from( |
| _TEST_MODULE_A_DICT_HAS_NONEED_ITEMS)) |
| |
| @mock.patch('subprocess.check_call') |
| @mock.patch('os.environ.copy') |
| def test_build_target_normal(self, mock_copy, mock_check_call): |
| """Test _build_target with verbose true and false.""" |
| mock_copy.return_value = '' |
| amodule_info = module_info.ModuleInfo() |
| cmd = [module_info_util._GENERATE_JSON_COMMAND] |
| module_info_util._build_target(cmd, uc.TEST_MODULE, amodule_info, True) |
| self.assertTrue(mock_copy.called) |
| self.assertTrue(mock_check_call.called) |
| mock_check_call.assert_called_with( |
| cmd, |
| stderr=subprocess.STDOUT, |
| env=mock_copy.return_value, |
| shell=True) |
| module_info_util._build_target(cmd, uc.TEST_MODULE, amodule_info, False) |
| self.assertTrue(mock_check_call.called) |
| mock_check_call.assert_called_with(cmd, shell=True) |
| |
| @mock.patch('os.path.getmtime') |
| @mock.patch('os.path.isfile') |
| def test_is_new_json_file_generated(self, mock_isfile, mock_getmtime): |
| """Test _is_new_json_file_generated with different situations.""" |
| jfile = 'path/test.json' |
| mock_isfile.return_value = True |
| self.assertEqual( |
| mock_isfile.return_value, |
| module_info_util._is_new_json_file_generated(jfile, None)) |
| mock_isfile.return_value = False |
| self.assertEqual( |
| mock_isfile.return_value, |
| module_info_util._is_new_json_file_generated(jfile, None)) |
| original_file_mtime = 1000 |
| mock_getmtime.return_value = original_file_mtime |
| self.assertEqual( |
| False, |
| module_info_util._is_new_json_file_generated( |
| jfile, original_file_mtime)) |
| mock_getmtime.return_value = 1001 |
| self.assertEqual( |
| True, |
| module_info_util._is_new_json_file_generated( |
| jfile, original_file_mtime)) |
| |
| @mock.patch.object(metrics, 'ends_asuite_metrics') |
| @mock.patch('builtins.input') |
| @mock.patch('glob.glob') |
| def test_build_failed_handle(self, mock_glob, mock_input, _send_exit): |
| """Test _build_failed_handle with different situations.""" |
| mock_glob.return_value = ['project/file.iml'] |
| mock_input.return_value = 'N' |
| with self.assertRaises(SystemExit) as cm: |
| module_info_util._build_failed_handle(uc.TEST_MODULE) |
| self.assertEqual(cm.exception.code, 1) |
| mock_glob.return_value = [] |
| with self.assertRaises(errors.BuildFailureError): |
| module_info_util._build_failed_handle(uc.TEST_MODULE) |
| |
| @mock.patch('builtins.open') |
| def test_get_soong_build_json_dict_failed(self, mock_open): |
| """Test _get_soong_build_json_dict failure and raise error.""" |
| mock_open.side_effect = IOError |
| with self.assertRaises(errors.JsonFileNotExistError): |
| module_info_util._get_soong_build_json_dict() |
| |
| @mock.patch('aidegen.lib.module_info_util._build_failed_handle') |
| @mock.patch('aidegen.lib.module_info_util._is_new_json_file_generated') |
| @mock.patch('subprocess.check_call') |
| def test_build_target(self, mock_call, mock_new, mock_handle): |
| """Test _build_target with different arguments.""" |
| cmd = [module_info_util._GENERATE_JSON_COMMAND] |
| main_project = '' |
| amodule_info = {} |
| verbose = False |
| module_info_util._build_target(cmd, main_project, amodule_info, verbose) |
| mock_call.assert_called_with(cmd, shell=True) |
| verbose = True |
| full_env_vars = os.environ.copy() |
| module_info_util._build_target(cmd, main_project, amodule_info, verbose) |
| mock_call.assert_called_with(cmd, stderr=subprocess.STDOUT, |
| env=full_env_vars, shell=True) |
| mock_call.side_effect = subprocess.CalledProcessError(1, '') |
| mock_new.return_value = False |
| module_info_util._build_target(cmd, main_project, amodule_info, verbose) |
| self.assertTrue(mock_new.called) |
| self.assertFalse(mock_handle.called) |
| mock_new.return_value = True |
| module_info_util._build_target(cmd, main_project, amodule_info, verbose) |
| self.assertTrue(mock_new.called) |
| self.assertTrue(mock_handle.called) |
| |
| |
| if __name__ == '__main__': |
| unittest.main() |