| #!/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 sys |
| import unittest |
| from unittest import mock |
| |
| from aidegen import aidegen_main |
| from aidegen import constant |
| from aidegen import unittest_constants |
| from aidegen.lib import aidegen_metrics |
| from aidegen.lib import common_util |
| from aidegen.lib import eclipse_project_file_gen |
| from aidegen.lib import errors |
| from aidegen.lib import ide_util |
| from aidegen.lib import module_info |
| from aidegen.lib import native_util |
| from aidegen.lib import native_module_info |
| from aidegen.lib import native_project_info |
| from aidegen.lib import project_config |
| from aidegen.lib import project_file_gen |
| from aidegen.lib import project_info |
| from aidegen.vscode import vscode_workspace_file_gen |
| |
| |
| # pylint: disable=protected-access |
| # pylint: disable=invalid-name |
| # pylint: disable=too-many-arguments |
| # pylint: disable=too-many-function-args |
| # pylint: disable=too-many-statements |
| class AidegenMainUnittests(unittest.TestCase): |
| """Unit tests for aidegen_main.py""" |
| |
| def _init_project_config(self, args): |
| """Initialize project configurations.""" |
| self.assertIsNotNone(args) |
| config = project_config.ProjectConfig(args) |
| config.init_environment() |
| |
| 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.assertTrue(args.verbose) |
| args = aidegen_main._parse_args(['-v']) |
| self.assertTrue(args.verbose) |
| 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', unittest_constants.TEST_MODULE]) |
| self.assertEqual(args.ide_installed_path, |
| unittest_constants.TEST_MODULE) |
| args = aidegen_main._parse_args(['-n']) |
| self.assertTrue(args.no_launch) |
| args = aidegen_main._parse_args(['-r']) |
| self.assertTrue(args.config_reset) |
| args = aidegen_main._parse_args(['-s']) |
| self.assertTrue(args.skip_build) |
| self.assertEqual(args.exclude_paths, None) |
| excludes = 'path/to/a', 'path/to/b' |
| args = aidegen_main._parse_args(['-e', excludes]) |
| self.assertEqual(args.exclude_paths, [excludes]) |
| |
| @mock.patch.object(project_config.ProjectConfig, 'init_environment') |
| @mock.patch.object(project_config, 'ProjectConfig') |
| @mock.patch.object(project_file_gen.ProjectFileGenerator, |
| 'generate_ide_project_files') |
| @mock.patch.object(eclipse_project_file_gen.EclipseConf, |
| 'generate_ide_project_files') |
| def test_generate_project_files(self, mock_eclipse, mock_ide, mock_config, |
| mock_init): |
| """Test _generate_project_files with different conditions.""" |
| projects = ['module_a', 'module_v'] |
| args = aidegen_main._parse_args([projects, '-i', 'e']) |
| mock_init.return_value = None |
| self._init_project_config(args) |
| mock_config.ide_name = constant.IDE_ECLIPSE |
| aidegen_main._generate_project_files(projects) |
| self.assertTrue(mock_eclipse.called_with(projects)) |
| mock_config.ide_name = constant.IDE_ANDROID_STUDIO |
| aidegen_main._generate_project_files(projects) |
| self.assertTrue(mock_ide.called_with(projects)) |
| mock_config.ide_name = constant.IDE_INTELLIJ |
| aidegen_main._generate_project_files(projects) |
| self.assertTrue(mock_ide.called_with(projects)) |
| |
| @mock.patch.object(aidegen_main, 'main_with_message') |
| @mock.patch.object(aidegen_main, 'main_without_message') |
| def test_main(self, mock_without, mock_with): |
| """Test main with conditions.""" |
| aidegen_main.main(['-s']) |
| self.assertEqual(mock_without.call_count, 1) |
| aidegen_main.main(['']) |
| self.assertEqual(mock_with.call_count, 1) |
| |
| @mock.patch.object(aidegen_metrics, 'starts_asuite_metrics') |
| @mock.patch.object(project_config, 'is_whole_android_tree') |
| @mock.patch.object(aidegen_metrics, 'ends_asuite_metrics') |
| @mock.patch.object(aidegen_main, 'main_with_message') |
| def test_main_with_normal(self, mock_main, mock_ends_metrics, |
| mock_is_whole_tree, mock_starts_metrics): |
| """Test main with normal conditions.""" |
| aidegen_main.main(['-h']) |
| self.assertFalse(mock_main.called) |
| mock_is_whole_tree.return_value = True |
| aidegen_main.main(['']) |
| mock_starts_metrics.assert_called_with([constant.ANDROID_TREE]) |
| self.assertFalse(mock_ends_metrics.called) |
| aidegen_main.main(['-n']) |
| mock_ends_metrics.assert_called_with(constant.EXIT_CODE_NORMAL) |
| |
| @mock.patch.object(aidegen_metrics, 'ends_asuite_metrics') |
| @mock.patch.object(aidegen_main, 'main_with_message') |
| def test_main_with_build_fail_errors(self, mock_main, mock_ends_metrics): |
| """Test main with raising build failure error conditions.""" |
| mock_main.side_effect = errors.BuildFailureError |
| with self.assertRaises(errors.BuildFailureError): |
| aidegen_main.main(['']) |
| _, exc_value, exc_traceback = sys.exc_info() |
| msg = str(exc_value) |
| mock_ends_metrics.assert_called_with( |
| constant.EXIT_CODE_AIDEGEN_EXCEPTION, exc_traceback, msg) |
| |
| @mock.patch.object(aidegen_metrics, 'ends_asuite_metrics') |
| @mock.patch.object(aidegen_main, 'main_with_message') |
| def test_main_with_io_errors(self, mock_main, mock_ends_metrics): |
| """Test main with raising IO error conditions.""" |
| mock_main.side_effect = IOError |
| with self.assertRaises(IOError): |
| aidegen_main.main(['']) |
| _, exc_value, exc_traceback = sys.exc_info() |
| msg = str(exc_value) |
| mock_ends_metrics.assert_called_with(constant.EXIT_CODE_EXCEPTION, |
| exc_traceback, msg) |
| |
| @mock.patch.object(aidegen_main, '_launch_ide') |
| @mock.patch.object(ide_util, 'get_ide_util_instance') |
| def test_launch_native_projects_without_ide_object( |
| self, mock_get_ide, mock_launch_ide): |
| """Test _launch_native_projects function without ide object.""" |
| target = 'libui' |
| args = aidegen_main._parse_args([target, '-i', 'e']) |
| aidegen_main._launch_native_projects(None, args, []) |
| self.assertFalse(mock_get_ide.called) |
| self.assertFalse(mock_launch_ide.called) |
| |
| @mock.patch.object(aidegen_main, '_launch_ide') |
| @mock.patch.object(ide_util, 'get_ide_util_instance') |
| def test_launch_native_projects_with_ide_object( |
| self, mock_get_ide, mock_launch_ide): |
| """Test _launch_native_projects function without ide object.""" |
| target = 'libui' |
| args = aidegen_main._parse_args([target, '-i', 'e']) |
| ide_util_obj = 'some_obj' |
| mock_get_ide.return_value = None |
| aidegen_main._launch_native_projects(ide_util_obj, args, []) |
| self.assertTrue(mock_get_ide.called_with('c')) |
| self.assertFalse(mock_launch_ide.called) |
| mock_get_ide.reset_mock() |
| mock_launch_ide.reset_mock() |
| args.ide = ['j'] |
| aidegen_main._launch_native_projects(ide_util_obj, args, []) |
| self.assertTrue(mock_get_ide.called_with('c')) |
| self.assertFalse(mock_launch_ide.called) |
| mock_get_ide.reset_mock() |
| mock_launch_ide.reset_mock() |
| mock_get_ide.return_value = 'some_native_obj' |
| aidegen_main._launch_native_projects(ide_util_obj, args, []) |
| self.assertTrue(mock_get_ide.called_with('c')) |
| self.assertTrue(mock_launch_ide.called) |
| mock_get_ide.reset_mock() |
| mock_launch_ide.reset_mock() |
| args.ide = ['e'] |
| aidegen_main._launch_native_projects(ide_util_obj, args, []) |
| self.assertTrue(mock_get_ide.called_with('c')) |
| self.assertTrue(mock_launch_ide.called) |
| mock_get_ide.reset_mock() |
| mock_launch_ide.reset_mock() |
| args.ide = ['s'] |
| aidegen_main._launch_native_projects(ide_util_obj, args, []) |
| self.assertFalse(mock_get_ide.called) |
| self.assertTrue(mock_launch_ide.called) |
| |
| @mock.patch('builtins.print') |
| def test_launch_ide(self, mock_print): |
| """Test _launch_ide function with config parameter.""" |
| mock_ide_util = mock.MagicMock() |
| mock_ide_util.launch_ide.return_value = None |
| mock_ide_util.config_ide.return_value = None |
| launch_path = '/test/launch/ide/method' |
| aidegen_main._launch_ide(mock_ide_util, launch_path) |
| self.assertTrue(mock_ide_util.config_ide.called) |
| self.assertTrue(mock_ide_util.launch_ide.called) |
| mock_print.return_value = None |
| |
| @mock.patch('builtins.input') |
| def test_get_preferred_ide_from_user(self, mock_input): |
| """Test get_preferred_ide_from_user with different conditions.""" |
| test_data = [] |
| aidegen_main._get_preferred_ide_from_user(test_data) |
| self.assertFalse(mock_input.called) |
| mock_input.reset_mock() |
| |
| test_data = ['One', 'Two', 'Three'] |
| mock_input.return_value = '3' |
| self.assertEqual('Three', aidegen_main._get_preferred_ide_from_user( |
| test_data)) |
| self.assertEqual(1, mock_input.call_count) |
| mock_input.reset_mock() |
| |
| mock_input.side_effect = ['7', '5', '3'] |
| self.assertEqual('Three', aidegen_main._get_preferred_ide_from_user( |
| test_data)) |
| self.assertEqual(3, mock_input.call_count) |
| mock_input.reset_mock() |
| |
| mock_input.side_effect = ('.', '7', 't', '5', '1') |
| self.assertEqual('One', aidegen_main._get_preferred_ide_from_user( |
| test_data)) |
| self.assertEqual(5, mock_input.call_count) |
| mock_input.reset_mock() |
| |
| @mock.patch.object(project_config.ProjectConfig, 'init_environment') |
| @mock.patch('logging.warning') |
| @mock.patch.object(aidegen_main, '_launch_vscode') |
| @mock.patch.object(aidegen_main, '_launch_native_projects') |
| @mock.patch.object(native_util, 'generate_clion_projects') |
| @mock.patch.object(native_project_info.NativeProjectInfo, |
| 'generate_projects') |
| @mock.patch.object(aidegen_main, '_create_and_launch_java_projects') |
| @mock.patch.object(aidegen_main, '_get_preferred_ide_from_user') |
| def test_launch_ide_by_module_contents(self, mock_choice, mock_j, |
| mock_c_prj, mock_genc, mock_c, |
| mock_vs, mock_log, mock_init): |
| """Test _launch_ide_by_module_contents with different conditions.""" |
| args = aidegen_main._parse_args(['', '-i', 's']) |
| mock_init.return_value = None |
| self._init_project_config(args) |
| ide_obj = 'ide_obj' |
| test_both = False |
| aidegen_main._launch_ide_by_module_contents(args, ide_obj, None, |
| None, test_both) |
| self.assertFalse(mock_vs.called) |
| self.assertTrue(mock_log.called) |
| self.assertFalse(mock_choice.called) |
| self.assertFalse(mock_choice.called) |
| self.assertFalse(mock_j.called) |
| self.assertFalse(mock_c_prj.called) |
| self.assertFalse(mock_genc.called) |
| self.assertFalse(mock_c.called) |
| |
| test_both = True |
| aidegen_main._launch_ide_by_module_contents(args, ide_obj, None, |
| None, test_both) |
| self.assertTrue(mock_vs.called) |
| self.assertFalse(mock_j.called) |
| self.assertFalse(mock_genc.called) |
| self.assertFalse(mock_c.called) |
| mock_vs.reset_mock() |
| |
| test_j = ['a', 'b', 'c'] |
| test_c = ['1', '2', '3'] |
| mock_choice.return_value = constant.JAVA |
| aidegen_main._launch_ide_by_module_contents(args, ide_obj, test_j, |
| test_c) |
| self.assertFalse(mock_vs.called) |
| self.assertTrue(mock_j.called) |
| self.assertFalse(mock_genc.called) |
| self.assertFalse(mock_c.called) |
| |
| mock_vs.reset_mock() |
| mock_choice.reset_mock() |
| mock_c.reset_mock() |
| mock_genc.reset_mock() |
| mock_j.reset_mock() |
| mock_choice.return_value = constant.C_CPP |
| aidegen_main._launch_ide_by_module_contents(args, ide_obj, test_j, |
| test_c) |
| self.assertTrue(mock_c_prj.called) |
| self.assertFalse(mock_vs.called) |
| self.assertTrue(mock_genc.called) |
| self.assertTrue(mock_c.called) |
| self.assertFalse(mock_j.called) |
| |
| mock_vs.reset_mock() |
| mock_choice.reset_mock() |
| mock_c.reset_mock() |
| mock_genc.reset_mock() |
| mock_j.reset_mock() |
| test_none = None |
| aidegen_main._launch_ide_by_module_contents(args, ide_obj, test_none, |
| test_c) |
| self.assertFalse(mock_vs.called) |
| self.assertTrue(mock_genc.called) |
| self.assertTrue(mock_c.called) |
| self.assertFalse(mock_j.called) |
| |
| mock_vs.reset_mock() |
| mock_choice.reset_mock() |
| mock_c.reset_mock() |
| mock_genc.reset_mock() |
| mock_j.reset_mock() |
| aidegen_main._launch_ide_by_module_contents(args, ide_obj, test_j, |
| test_none) |
| self.assertFalse(mock_vs.called) |
| self.assertTrue(mock_j.called) |
| self.assertFalse(mock_c.called) |
| self.assertFalse(mock_genc.called) |
| |
| args = aidegen_main._parse_args(['frameworks/base', '-i', 'c']) |
| mock_vs.reset_mock() |
| mock_choice.reset_mock() |
| mock_c.reset_mock() |
| mock_genc.reset_mock() |
| mock_c_prj.reset_mock() |
| mock_j.reset_mock() |
| aidegen_main._launch_ide_by_module_contents(args, ide_obj, test_j, |
| test_c) |
| self.assertFalse(mock_vs.called) |
| self.assertFalse(mock_choice.called) |
| self.assertFalse(mock_j.called) |
| self.assertTrue(mock_c.called) |
| self.assertTrue(mock_c_prj.called) |
| self.assertTrue(mock_genc.called) |
| |
| args = aidegen_main._parse_args(['frameworks/base']) |
| mock_vs.reset_mock() |
| mock_choice.reset_mock() |
| mock_c.reset_mock() |
| mock_genc.reset_mock() |
| mock_c_prj.reset_mock() |
| mock_j.reset_mock() |
| os.environ[constant.AIDEGEN_TEST_MODE] = 'true' |
| aidegen_main._launch_ide_by_module_contents(args, None, test_j, test_c) |
| self.assertFalse(mock_vs.called) |
| self.assertFalse(mock_choice.called) |
| self.assertTrue(mock_j.called) |
| self.assertFalse(mock_c.called) |
| self.assertFalse(mock_c_prj.called) |
| self.assertFalse(mock_genc.called) |
| del os.environ[constant.AIDEGEN_TEST_MODE] |
| |
| @mock.patch.object(aidegen_main, '_launch_ide') |
| @mock.patch.object(aidegen_main, '_generate_project_files') |
| @mock.patch.object(project_info.ProjectInfo, 'multi_projects_locate_source') |
| @mock.patch.object(project_info.ProjectInfo, 'generate_projects') |
| def test_create_and_launch_java_projects(self, mock_prj, mock_compile, |
| mock_gen_file, mock_launch): |
| """Test _create_and_launch_java_projects.""" |
| ide_obj = 'test_ide' |
| target = ['a', 'b'] |
| mock_prj_list = mock.MagicMock() |
| mock_prj_list.project_absolute_path = 'test_path' |
| prj = [mock_prj_list] |
| mock_prj.return_value = prj |
| aidegen_main._create_and_launch_java_projects(ide_obj, target) |
| self.assertTrue(mock_prj.called_with(target)) |
| self.assertTrue(mock_compile.called_with(prj)) |
| self.assertTrue(mock_gen_file.called_with(prj)) |
| self.assertTrue(mock_launch.called_with(ide_obj)) |
| mock_launch.reset_mock() |
| |
| aidegen_main._create_and_launch_java_projects(None, target) |
| self.assertFalse(mock_launch.called) |
| |
| @mock.patch.object(aidegen_main, '_launch_ide') |
| @mock.patch.object(vscode_workspace_file_gen, |
| 'generate_code_workspace_file') |
| @mock.patch.object(common_util, 'get_related_paths') |
| def test_launch_vscode_without_ide_object( |
| self, mock_get_rel, mock_gen_code, mock_launch_ide): |
| """Test _launch_vscode function without ide object.""" |
| mock_get_rel.return_value = 'rel', 'abs' |
| aidegen_main._launch_vscode(None, mock.Mock(), ['Settings'], []) |
| self.assertTrue(mock_get_rel.called) |
| self.assertTrue(mock_gen_code.called) |
| self.assertFalse(mock_launch_ide.called) |
| |
| @mock.patch.object(aidegen_main, '_launch_ide') |
| @mock.patch.object(vscode_workspace_file_gen, |
| 'generate_code_workspace_file') |
| @mock.patch.object(common_util, 'get_related_paths') |
| def test_launch_vscode_with_ide_object( |
| self, mock_get_rel, mock_gen_code, mock_get_ide): |
| """Test _launch_vscode function with ide object.""" |
| mock_get_rel.return_value = 'rel', 'abs' |
| aidegen_main._launch_vscode(mock.Mock(), mock.Mock(), ['Settings'], []) |
| self.assertTrue(mock_get_rel.called) |
| self.assertTrue(mock_gen_code.called) |
| self.assertTrue(mock_get_ide.called) |
| |
| @mock.patch.object(aidegen_main, '_launch_vscode') |
| @mock.patch.object(aidegen_main, '_launch_ide') |
| @mock.patch.object(vscode_workspace_file_gen, |
| 'generate_code_workspace_file') |
| @mock.patch.object(common_util, 'get_related_paths') |
| def test_launch_vscode_with_both_languages( |
| self, mock_get_rel, mock_gen_code, mock_get_ide, mock_vscode): |
| """Test _launch_vscode function without ide object.""" |
| mock_get_rel.return_value = 'rel', 'abs' |
| aidegen_main._launch_vscode(None, mock.Mock(), ['Settings'], [], True) |
| self.assertFalse(mock_get_rel.called) |
| self.assertFalse(mock_gen_code.called) |
| self.assertFalse(mock_get_ide.called) |
| self.assertTrue(mock_vscode.called) |
| |
| @mock.patch.object(aidegen_main, '_launch_ide_by_module_contents') |
| @mock.patch.object(native_util, 'get_native_and_java_projects') |
| @mock.patch.object(native_module_info, 'NativeModuleInfo') |
| @mock.patch.object(module_info, 'AidegenModuleInfo') |
| @mock.patch.object(ide_util, 'get_ide_util_instance') |
| @mock.patch.object(project_config, 'ProjectConfig') |
| def test_aidegen_main(self, mock_config, mock_get_ide, mock_mod_info, |
| mock_native, mock_get_project, mock_launch_ide): |
| """Test aidegen_main function with conditions.""" |
| target = 'Settings' |
| args = aidegen_main._parse_args([target, '-i', 'v']) |
| config = mock.Mock() |
| config.targets = [target] |
| config.ide_name = constant.IDE_VSCODE |
| mock_config.return_value = config |
| ide = mock.Mock() |
| mock_get_ide.return_value = ide |
| mock_get_project.return_value = config.targets, [] |
| aidegen_main.aidegen_main(args) |
| self.assertTrue(mock_config.called) |
| self.assertTrue(mock_get_ide.called) |
| self.assertTrue(mock_mod_info.called) |
| self.assertTrue(mock_native.called) |
| self.assertTrue(mock_get_project.called) |
| mock_launch_ide.assert_called_with(args, ide, config.targets, [], True) |
| |
| mock_config.mock_reset() |
| mock_get_ide.mock_reset() |
| mock_mod_info.mock_reset() |
| mock_native.mock_reset() |
| mock_get_project.mock_reset() |
| mock_launch_ide.mock_reset() |
| |
| args = aidegen_main._parse_args([target]) |
| config.ide_name = constant.IDE_INTELLIJ |
| aidegen_main.aidegen_main(args) |
| self.assertTrue(mock_config.called) |
| self.assertTrue(mock_get_ide.called) |
| self.assertTrue(mock_mod_info.called) |
| self.assertTrue(mock_native.called) |
| self.assertTrue(mock_get_project.called) |
| mock_launch_ide.assert_called_with(args, ide, config.targets, [], False) |
| |
| |
| if __name__ == '__main__': |
| unittest.main() |