| #!/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 aidegen_main.""" |
| |
| from __future__ import print_function |
| |
| import os |
| import unittest |
| from unittest import mock |
| |
| import aidegen.unittest_constants as uc |
| from aidegen import aidegen_main |
| from aidegen.lib import aidegen_metrics |
| from aidegen import constant |
| from aidegen.lib import common_util |
| from aidegen.lib.common_util import COLORED_INFO |
| from aidegen.lib.errors import IDENotExistError |
| from aidegen.lib.errors import ProjectPathNotExistError |
| from aidegen.lib.ide_util import IdeUtil |
| from atest import module_info |
| |
| |
| # pylint: disable=protected-access |
| # pylint: disable=invalid-name |
| class AidegenMainUnittests(unittest.TestCase): |
| """Unit tests for aidegen_main.py""" |
| |
| def test_parse_args(self): |
| """Test _parse_args with different conditions.""" |
| args = aidegen_main._parse_args([]) |
| self.assertEqual(args.targets, ['']) |
| self.assertEqual(args.ide[0], 'j') |
| target = 'tradefed' |
| args = aidegen_main._parse_args([target]) |
| self.assertEqual(args.targets, [target]) |
| depth = '2' |
| args = aidegen_main._parse_args(['-d', depth]) |
| self.assertEqual(args.depth, int(depth)) |
| args = aidegen_main._parse_args(['-v']) |
| self.assertEqual(args.verbose, True) |
| args = aidegen_main._parse_args(['-v']) |
| self.assertEqual(args.verbose, True) |
| args = aidegen_main._parse_args(['-i', 's']) |
| self.assertEqual(args.ide[0], 's') |
| args = aidegen_main._parse_args(['-i', 'e']) |
| self.assertEqual(args.ide[0], 'e') |
| args = aidegen_main._parse_args(['-p', uc.TEST_MODULE]) |
| self.assertEqual(args.ide_installed_path, uc.TEST_MODULE) |
| args = aidegen_main._parse_args(['-n']) |
| self.assertEqual(args.no_launch, True) |
| args = aidegen_main._parse_args(['-r']) |
| self.assertEqual(args.config_reset, True) |
| args = aidegen_main._parse_args(['-s']) |
| self.assertEqual(args.skip_build, True) |
| |
| @mock.patch('aidegen_main.logging.basicConfig') |
| def test_configure_logging(self, mock_log_config): |
| """Test _configure_logging with different arguments.""" |
| aidegen_main._configure_logging(True) |
| log_format = aidegen_main._LOG_FORMAT |
| datefmt = aidegen_main._DATE_FORMAT |
| level = aidegen_main.logging.DEBUG |
| self.assertTrue( |
| mock_log_config.called_with( |
| level=level, format=log_format, datefmt=datefmt)) |
| aidegen_main._configure_logging(False) |
| level = aidegen_main.logging.INFO |
| self.assertTrue( |
| mock_log_config.called_with( |
| level=level, format=log_format, datefmt=datefmt)) |
| |
| @mock.patch.object(IdeUtil, 'is_ide_installed') |
| def test_get_ide_util_instance(self, mock_installed): |
| """Test _get_ide_util_instance with different conditions.""" |
| target = 'tradefed' |
| args = aidegen_main._parse_args([target, '-n']) |
| self.assertEqual(aidegen_main._get_ide_util_instance(args), None) |
| args = aidegen_main._parse_args([target]) |
| self.assertIsInstance( |
| aidegen_main._get_ide_util_instance(args), IdeUtil) |
| mock_installed.return_value = False |
| with self.assertRaises(IDENotExistError): |
| aidegen_main._get_ide_util_instance(args) |
| |
| @mock.patch('builtins.print') |
| def test_check_skip_build(self, mock_print): |
| """Test _check_skip_build with different conditions.""" |
| target = 'tradefed' |
| args = aidegen_main._parse_args([target, '-s']) |
| aidegen_main._check_skip_build(args) |
| self.assertFalse(mock_print.called) |
| args = aidegen_main._parse_args([target]) |
| aidegen_main._check_skip_build(args) |
| msg = aidegen_main._SKIP_BUILD_INFO.format( |
| COLORED_INFO( |
| aidegen_main._SKIP_BUILD_CMD.format(' '.join(args.targets)))) |
| info = '\n{} {}\n'.format(aidegen_main._INFO, msg) |
| self.assertTrue(mock_print.called_with(info)) |
| |
| @mock.patch.object(aidegen_main, 'generate_ide_project_files') |
| @mock.patch.object(aidegen_main, 'generate_eclipse_project_files') |
| def test_generate_project_files(self, mock_eclipse, mock_ide): |
| """Test _generate_project_files with different conditions.""" |
| projects = ['module_a', 'module_v'] |
| aidegen_main._generate_project_files('e', projects) |
| self.assertTrue(mock_eclipse.called_with(projects)) |
| aidegen_main._generate_project_files('s', projects) |
| self.assertTrue(mock_ide.called_with(projects)) |
| aidegen_main._generate_project_files('j', projects) |
| self.assertTrue(mock_ide.called_with(projects)) |
| |
| @mock.patch.object(common_util, 'get_atest_module_info') |
| @mock.patch.object(aidegen_metrics, 'starts_asuite_metrics') |
| def test_show_collect_data_notice(self, mock_metrics, mock_get): |
| """Test main process always run through the target test function.""" |
| target = 'nothing' |
| args = aidegen_main._parse_args([target, '-s', '-n']) |
| with self.assertRaises(ProjectPathNotExistError): |
| err = common_util.PATH_NOT_EXISTS_ERROR.format(target) |
| mock_get.side_effect = ProjectPathNotExistError(err) |
| aidegen_main.main_without_message(args) |
| self.assertTrue(mock_metrics.called) |
| |
| @mock.patch.object(common_util, 'get_related_paths') |
| def test_compile_targets_for_whole_android_tree(self, mock_get): |
| """Test _add_whole_android_tree_project with different conditions.""" |
| mod_info = module_info.ModuleInfo() |
| targets = [''] |
| cwd = constant.ANDROID_ROOT_PATH |
| self.assertEqual( |
| targets, |
| aidegen_main._compile_targets_for_whole_android_tree( |
| mod_info, targets, cwd)) |
| base_dir = 'frameworks/base' |
| expected_targets = ['', base_dir] |
| cwd = os.path.join(constant.ANDROID_ROOT_PATH, base_dir) |
| mock_get.return_value = None, cwd |
| self.assertEqual( |
| expected_targets, |
| aidegen_main._compile_targets_for_whole_android_tree( |
| mod_info, targets, cwd)) |
| targets = [base_dir] |
| cwd = constant.ANDROID_ROOT_PATH |
| self.assertEqual( |
| expected_targets, |
| aidegen_main._compile_targets_for_whole_android_tree( |
| mod_info, targets, cwd)) |
| |
| |
| if __name__ == '__main__': |
| unittest.main() |