#!/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()
|
