#!/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_data.""" import os import unittest from unittest import mock from aidegen import unittest_constants from aidegen.lib import common_util from aidegen.lib import module_info from aidegen.lib import project_config from aidegen.lib import source_locator from atest import module_info as amodule_info # pylint: disable=too-many-arguments # pylint: disable=protected-access # pylint: disable=invalid-name class ModuleDataUnittests(unittest.TestCase): """Unit tests for module_data.py""" @mock.patch('os.path.dirname') @mock.patch('logging.debug') @mock.patch.object(source_locator.ModuleData, '_get_source_folder') @mock.patch.object(source_locator.ModuleData, '_check_key') @mock.patch.object(common_util, 'is_target') @mock.patch.object(common_util, 'get_android_root_dir') def test_collect_srcs_paths(self, mock_android_root_dir, mock_is_target, mock_check_key, mock_get_src, mock_log, mock_dirname): """Test _collect_srcs_paths create the source path list.""" module = source_locator.ModuleData( unittest_constants.TEST_MODULE, unittest_constants.MODULE_INFO, 0) mock_check_key.return_value = False module._collect_srcs_paths() self.assertFalse(mock_dirname.called) mock_check_key.return_value = True mock_is_target.return_value = True mock_android_root_dir.return_value = unittest_constants.TEST_DATA_PATH module._collect_srcs_paths() self.assertTrue(mock_is_target.called) self.assertTrue(mock_get_src.called) mock_is_target.return_value = False module._collect_srcs_paths() self.assertTrue(mock_log.called) def test_get_package_name(self): """test get the package name from a java file.""" result_package_name = 'com.android' test_java = os.path.join(unittest_constants.TEST_DATA_PATH, unittest_constants.MODULE_PATH, 'src/main/java/com/android/java.java') package_name = source_locator.ModuleData._get_package_name(test_java) self.assertEqual(package_name, result_package_name) # Test on java file with no package name. result_package_name = None test_java = os.path.join(unittest_constants.TEST_DATA_PATH, unittest_constants.MODULE_PATH, 'src/main/java/com/android/no_package.java') package_name = source_locator.ModuleData._get_package_name(test_java) self.assertEqual(package_name, result_package_name) @mock.patch('aidegen.lib.common_util.get_android_root_dir') def test_get_source_folder(self, mock_android_root_dir): """Test _get_source_folder process.""" # Test for getting the source path by parse package name from a java. test_java = 'packages/apps/test/src/main/java/com/android/java.java' result_source = 'packages/apps/test/src/main/java' mock_android_root_dir.return_value = unittest_constants.TEST_DATA_PATH module_data = source_locator.ModuleData( unittest_constants.TEST_MODULE, unittest_constants.MODULE_INFO, 0) src_path = module_data._get_source_folder(test_java) self.assertEqual(src_path, result_source) # Return path is None if the java file doesn't exist. test_java = 'file_not_exist.java' src_path = module_data._get_source_folder(test_java) self.assertEqual(src_path, None) # Return path is None on the java file without package name. test_java = ('packages/apps/test/src/main/java/com/android/' 'no_package.java') src_path = module_data._get_source_folder(test_java) self.assertEqual(src_path, None) def test_get_r_dir(self): """Test get_r_dir.""" module_data = source_locator.ModuleData( unittest_constants.TEST_MODULE, unittest_constants.MODULE_INFO, 0) # Test for aapt2.srcjar test_aapt2_srcjar = 'a/aapt2.srcjar' expect_result = 'a/aapt2' r_dir = module_data._get_r_dir(test_aapt2_srcjar) self.assertEqual(r_dir, expect_result) # Test for R.srcjar test_r_jar = 'b/android/R.srcjar' expect_result = 'b/aapt2/R' r_dir = module_data._get_r_dir(test_r_jar) self.assertEqual(r_dir, expect_result) # Test the R.srcjar is not under the android folder. test_wrong_r_jar = 'b/test/R.srcjar' expect_result = None r_dir = module_data._get_r_dir(test_wrong_r_jar) self.assertEqual(r_dir, expect_result) # Test for the target file is not aapt2.srcjar or R.srcjar test_unknown_target = 'c/proto.srcjar' expect_result = None r_dir = module_data._get_r_dir(test_unknown_target) self.assertEqual(r_dir, expect_result) @mock.patch('os.path.exists') @mock.patch('aidegen.lib.common_util.get_android_root_dir') def test_collect_r_src_path(self, mock_android_root_dir, mock_exists): """Test collect_r_src_path.""" mock_exists.return_value = True # Test on target srcjar exists in srcjars. test_module = dict(unittest_constants.MODULE_INFO) test_module['srcs'] = [] mock_android_root_dir.return_value = unittest_constants.TEST_DATA_PATH module_data = source_locator.ModuleData(unittest_constants.TEST_MODULE, test_module, 0) # Test the module is not APPS. module_data._collect_r_srcs_paths() expect_result = [] self.assertEqual(module_data.r_java_paths, expect_result) # Test the module is not a target module. test_module['depth'] = 1 module_data = source_locator.ModuleData(unittest_constants.TEST_MODULE, test_module, 1) module_data._collect_r_srcs_paths() expect_result = [] self.assertEqual(module_data.r_java_paths, expect_result) # Test the srcjar target doesn't exist. test_module['class'] = ['APPS'] test_module['srcjars'] = [] module_data = source_locator.ModuleData(unittest_constants.TEST_MODULE, test_module, 0) module_data._collect_r_srcs_paths() expect_result = [] self.assertEqual(module_data.r_java_paths, expect_result) # Test the srcjar target exists. test_module['srcjars'] = [('out/soong/.intermediates/packages/apps/' 'test_aapt2/aapt2.srcjar')] module_data = source_locator.ModuleData(unittest_constants.TEST_MODULE, test_module, 0) module_data._collect_r_srcs_paths() expect_result = [ 'out/soong/.intermediates/packages/apps/test_aapt2/aapt2' ] self.assertEqual(module_data.r_java_paths, expect_result) mock_exists.return_value = False module_data._collect_r_srcs_paths() expect_result = set([('out/soong/.intermediates/packages/apps/' 'test_aapt2/aapt2.srcjar')]) self.assertEqual(module_data.build_targets, expect_result) def test_parse_source_path(self): """Test _parse_source_path.""" # The package name of e.java is c.d. test_java = 'a/b/c/d/e.java' package_name = 'c.d' expect_result = 'a/b' src_path = source_locator.ModuleData._parse_source_path( test_java, package_name) self.assertEqual(src_path, expect_result) # The package name of e.java is c.d. test_java = 'a/b/c.d/e.java' package_name = 'c.d' expect_result = 'a/b' src_path = source_locator.ModuleData._parse_source_path( test_java, package_name) self.assertEqual(src_path, expect_result) # The package name of e.java is x.y. test_java = 'a/b/c/d/e.java' package_name = 'x.y' expect_result = 'a/b/c/d' src_path = source_locator.ModuleData._parse_source_path( test_java, package_name) self.assertEqual(src_path, expect_result) # The package name of f.java is c.d. test_java = 'a/b/c.d/e/c/d/f.java' package_name = 'c.d' expect_result = 'a/b/c.d/e' src_path = source_locator.ModuleData._parse_source_path( test_java, package_name) self.assertEqual(src_path, expect_result) # The package name of f.java is c.d.e. test_java = 'a/b/c.d/e/c.d/e/f.java' package_name = 'c.d.e' expect_result = 'a/b/c.d/e' src_path = source_locator.ModuleData._parse_source_path( test_java, package_name) self.assertEqual(src_path, expect_result) @mock.patch('aidegen.lib.common_util.get_android_root_dir') def test_append_jar_file(self, mock_android_root_dir): """Test _append_jar_file process.""" # Append an existing jar file path to module_data.jar_files. test_jar_file = os.path.join(unittest_constants.MODULE_PATH, 'test.jar') result_jar_list = [test_jar_file] mock_android_root_dir.return_value = unittest_constants.TEST_DATA_PATH module_data = source_locator.ModuleData( unittest_constants.TEST_MODULE, unittest_constants.MODULE_INFO, 0) module_data._append_jar_file(test_jar_file) module_data._append_jar_file(test_jar_file) self.assertEqual(module_data.jar_files, result_jar_list) # Skip if the jar file doesn't exist. test_jar_file = os.path.join(unittest_constants.MODULE_PATH, 'jar_not_exist.jar') module_data.jar_files = [] module_data._append_jar_file(test_jar_file) self.assertEqual(module_data.jar_files, []) # Skip if it's not a jar file. test_jar_file = os.path.join(unittest_constants.MODULE_PATH, 'test.java') module_data.jar_files = [] module_data._append_jar_file(test_jar_file) self.assertEqual(module_data.jar_files, []) @mock.patch.object(source_locator.ModuleData, '_check_key') @mock.patch('aidegen.lib.common_util.get_android_root_dir') def test_append_jar_from_installed(self, mock_android_root_dir, mock_check_key): """Test _append_jar_from_installed handling.""" mock_check_key.return_value = True # Test appends the first jar file of 'installed'. mod_info = dict(unittest_constants.MODULE_INFO) mod_info['installed'] = [ os.path.join(unittest_constants.MODULE_PATH, 'test.aar'), os.path.join(unittest_constants.MODULE_PATH, 'test.jar'), os.path.join(unittest_constants.MODULE_PATH, 'tests/test_second.jar') ] result_jar_list = [ os.path.join(unittest_constants.MODULE_PATH, 'test.jar') ] mock_android_root_dir.return_value = unittest_constants.TEST_DATA_PATH module_data = source_locator.ModuleData(unittest_constants.TEST_MODULE, mod_info, 0) module_data._append_jar_from_installed() self.assertEqual(module_data.jar_files, result_jar_list) # Test on the jar file path matches the path prefix. module_data.jar_files = [] result_jar_list = [ os.path.join(unittest_constants.MODULE_PATH, 'tests/test_second.jar') ] module_data._append_jar_from_installed( os.path.join(unittest_constants.MODULE_PATH, 'tests/')) self.assertEqual(module_data.jar_files, result_jar_list) mock_check_key.return_value = False module_data.jar_files = [] module_data._append_jar_from_installed( os.path.join(unittest_constants.MODULE_PATH, 'tests/')) self.assertEqual(module_data.jar_files, []) @mock.patch('aidegen.lib.common_util.get_android_root_dir') def test_set_jars_jarfile(self, mock_android_root_dir): """Test _set_jars_jarfile handling.""" # Combine the module path with jar file name in 'jars' and then append # it to module_data.jar_files. mod_info = dict(unittest_constants.MODULE_INFO) mod_info['jars'] = [ 'test.jar', 'src/test.jar', # This jar file doesn't exist. 'tests/test_second.jar' ] result_jar_list = [ os.path.join(unittest_constants.MODULE_PATH, 'test.jar'), os.path.join(unittest_constants.MODULE_PATH, 'tests/test_second.jar') ] result_missing_jars = set() mock_android_root_dir.return_value = unittest_constants.TEST_DATA_PATH module_data = source_locator.ModuleData(unittest_constants.TEST_MODULE, mod_info, 0) module_data._set_jars_jarfile() self.assertEqual(module_data.jar_files, result_jar_list) self.assertEqual(module_data.missing_jars, result_missing_jars) @mock.patch('aidegen.lib.common_util.get_android_root_dir') def test_locate_sources_path(self, mock_android_root_dir): """Test locate_sources_path handling.""" # Test collect source path. mod_info = dict(unittest_constants.MODULE_INFO) result_src_list = ['packages/apps/test/src/main/java'] result_test_list = ['packages/apps/test/tests'] result_jar_list = [] result_r_path = [] mock_android_root_dir.return_value = unittest_constants.TEST_DATA_PATH module_data = source_locator.ModuleData(unittest_constants.TEST_MODULE, mod_info, 0) module_data.locate_sources_path() self.assertEqual(module_data.src_dirs, result_src_list) self.assertEqual(module_data.test_dirs, result_test_list) self.assertEqual(module_data.jar_files, result_jar_list) self.assertEqual(module_data.r_java_paths, result_r_path) # Test find jar files. jar_file = ('out/soong/.intermediates/packages/apps/test/test/' 'android_common/test.jar') mod_info['jarjar_rules'] = ['jarjar-rules.txt'] mod_info['installed'] = [jar_file] result_jar_list = [jar_file] module_data = source_locator.ModuleData(unittest_constants.TEST_MODULE, mod_info, 0) module_data.locate_sources_path() self.assertEqual(module_data.jar_files, result_jar_list) @mock.patch('aidegen.lib.common_util.get_android_root_dir') def test_collect_jar_by_depth_value(self, mock_android_root_dir): """Test parameter --depth handling.""" # Test find jar by module's depth greater than the --depth value from # command line. depth_by_source = 2 mod_info = dict(unittest_constants.MODULE_INFO) mod_info['depth'] = 3 mod_info['installed'] = [ ('out/soong/.intermediates/packages/apps/test/test/android_common/' 'test.jar') ] result_src_list = [] result_jar_list = [ ('out/soong/.intermediates/packages/apps/test/test/' 'android_common/test.jar') ] mock_android_root_dir.return_value = unittest_constants.TEST_DATA_PATH module_data = source_locator.ModuleData(unittest_constants.TEST_MODULE, mod_info, depth_by_source) module_data.locate_sources_path() self.assertEqual(module_data.src_dirs, result_src_list) self.assertEqual(module_data.jar_files, result_jar_list) # Test find source folder when module's depth equal to the --depth value # from command line. depth_by_source = 2 mod_info = dict(unittest_constants.MODULE_INFO) mod_info['depth'] = 2 result_src_list = ['packages/apps/test/src/main/java'] result_test_list = ['packages/apps/test/tests'] result_jar_list = [] result_r_path = [] module_data = source_locator.ModuleData(unittest_constants.TEST_MODULE, mod_info, depth_by_source) module_data.locate_sources_path() self.assertEqual(module_data.src_dirs, result_src_list) self.assertEqual(module_data.test_dirs, result_test_list) self.assertEqual(module_data.jar_files, result_jar_list) self.assertEqual(module_data.r_java_paths, result_r_path) # Test find source folder when module's depth smaller than the --depth # value from command line. depth_by_source = 3 mod_info = dict(unittest_constants.MODULE_INFO) mod_info['depth'] = 2 result_src_list = ['packages/apps/test/src/main/java'] result_test_list = ['packages/apps/test/tests'] result_jar_list = [] result_r_path = [] module_data = source_locator.ModuleData(unittest_constants.TEST_MODULE, mod_info, depth_by_source) module_data.locate_sources_path() self.assertEqual(module_data.src_dirs, result_src_list) self.assertEqual(module_data.test_dirs, result_test_list) self.assertEqual(module_data.jar_files, result_jar_list) self.assertEqual(module_data.r_java_paths, result_r_path) def test_collect_srcjar_path(self): """Test collect srcjar path.""" srcjar_path = 'a/b/aapt2.srcjar' test_module = dict(unittest_constants.MODULE_INFO) test_module['srcjars'] = [srcjar_path] expacted_result = [srcjar_path] module_data = source_locator.ModuleData(unittest_constants.TEST_MODULE, test_module, 0) module_data._collect_srcjar_path('R.java') self.assertEqual(module_data.srcjar_paths, []) module_data._collect_srcjar_path(srcjar_path) self.assertEqual(module_data.srcjar_paths, expacted_result) @mock.patch('os.path.exists') def test_collect_all_srcjar_path(self, mock_exists): """Test collect all srcjar paths as source root folders.""" mock_exists.return_value = True test_module = dict(unittest_constants.MODULE_INFO) test_module['srcjars'] = [ 'a/b/aidl0.srcjar', 'a/b/aidl2.srcjar', 'a/b/aidl1.srcjar', 'a/b/aidl2.srcjar' ] expacted_result = [ 'a/b/aidl0.srcjar', 'a/b/aidl2.srcjar', 'a/b/aidl1.srcjar' ] module_data = source_locator.ModuleData(unittest_constants.TEST_MODULE, test_module, 0) module_data._collect_all_srcjar_paths() self.assertEqual(module_data.srcjar_paths, expacted_result) mock_exists.return_value = False test_module['srcjars'] = ['a/b/aidl0.srcjar'] expacted_result = set(['a/b/aidl0.srcjar']) module_data = source_locator.ModuleData(unittest_constants.TEST_MODULE, test_module, 0) module_data._collect_all_srcjar_paths() self.assertEqual(module_data.build_targets, expacted_result) def test_collect_missing_jars(self): """Test _collect_missing_jars.""" mod_name = 'test' mod_info = {'name': 'test'} test_path = 'a/b/c' mod_data = source_locator.EclipseModuleData(mod_name, mod_info, test_path) mod_data.missing_jars = set('a') mod_data.referenced_by_jar = False mod_data._collect_missing_jars() self.assertEqual(mod_data.build_targets, set()) mod_data.referenced_by_jar = True mod_data._collect_missing_jars() self.assertEqual(mod_data.build_targets, {'a'}) @mock.patch.object(source_locator.ModuleData, '_check_key') def test_check_classes_jar_exist(self, mock_check_key): """Test _check_classes_jar_exist.""" mod_data = source_locator.ModuleData( unittest_constants.TEST_MODULE, unittest_constants.MODULE_INFO, 0) mod_data._check_classes_jar_exist() self.assertTrue(mock_check_key.called) @mock.patch('os.path.exists') @mock.patch.object(common_util, 'get_android_root_dir') def test_switch_repackaged(self, mock_android_root_dir, mock_exist): """Test _switch_repackaged.""" mock_android_root_dir.return_value = '/a' mock_exist.return_value = False mod_data = source_locator.ModuleData( unittest_constants.TEST_MODULE, unittest_constants.MODULE_INFO, 0) self.assertEqual(mod_data._switch_repackaged('b/c'), 'b/c') mock_exist.return_value = True self.assertEqual(mod_data._switch_repackaged('b/c'), 'b/repackaged/c') def test_add_to_source_or_test_dirs(self): """Test _add_to_source_or_test_dirs.""" mod_name = 'test' mod_info = {'name': 'test'} mod_data = source_locator.ModuleData(mod_name, mod_info, 0) mod_data._add_to_source_or_test_dirs('libcore/ojluni/src/lambda/java') self.assertEqual(mod_data.src_dirs, []) mod_data._add_to_source_or_test_dirs('a') self.assertEqual(mod_data.src_dirs, ['a']) mod_data._add_to_source_or_test_dirs('b') self.assertEqual(mod_data.src_dirs, ['a', 'b']) mod_data._add_to_source_or_test_dirs('a') self.assertEqual(mod_data.src_dirs, ['a', 'b']) mod_data._add_to_source_or_test_dirs('tests/src') self.assertEqual(mod_data.test_dirs, ['tests/src']) @mock.patch.object(source_locator.ModuleData, '_append_jar_file') def test_append_classes_jar(self, mock_append_jar): """Test _append_classes_jar.""" mod_name = 'test' mod_info = {'name': 'test'} mod_data = source_locator.ModuleData(mod_name, mod_info, 0) mod_data.module_data['classes_jar'] = ['a.jar'] mod_data._append_classes_jar() self.assertTrue(mock_append_jar.called) mock_append_jar.return_value = False mod_data._append_classes_jar() self.assertEqual(mod_data.jar_files, []) @mock.patch.object(amodule_info, 'ModuleInfo') @mock.patch.object(amodule_info.ModuleInfo, 'get_paths') @mock.patch.object(project_config.ProjectConfig, 'get_instance') def test_collect_dep_paths(self, mock_config, mock_get_paths, mock_atest_module_info): """Test _collect_dep_paths.""" mod_name = 'test' mod_info = { 'name': 'test', 'path': ['frameworks/base'], 'dependencies': ['test_module'] } mod_data = source_locator.ModuleData(mod_name, mod_info, 0) mock_instance = mock_config.return_value mock_instance.atest_module_info = mock_atest_module_info mock_instance.atest_module_info.get_paths = mock_get_paths mock_get_paths.return_value = [] expected = [ 'frameworks/base/framework_srcjars', 'libcore', ] mod_data._collect_dep_paths() self.assertEqual(mod_data.dep_paths, expected) mod_info['path'] = ['libcore'] mod_data = source_locator.ModuleData(mod_name, mod_info, 0) expected = [ 'frameworks/base', 'frameworks/base/framework_srcjars', ] mod_data._collect_dep_paths() self.assertEqual(mod_data.dep_paths, expected) mock_get_paths.return_value = ['test'] mod_info['path'] = ['test'] mod_data = source_locator.ModuleData(mod_name, mod_info, 0) expected = [ 'frameworks/base', 'frameworks/base/framework_srcjars', 'libcore', ] mod_data._collect_dep_paths() self.assertEqual(mod_data.dep_paths, expected) mock_get_paths.return_value = ['dep/path'] expected.append('dep/path') mod_data._collect_dep_paths() self.assertEqual(mod_data.dep_paths, expected) class EclipseModuleDataUnittests(unittest.TestCase): """Unit tests for the EclipseModuleData in module_data.py""" @mock.patch.object(module_info.AidegenModuleInfo, 'is_project_path_relative_module') @mock.patch.object(source_locator.ModuleData, '__init__') def test___init__(self, mock_base_init, mock_method): """Test the implement of __init__().""" mod_name = 'test' mod_info = {'name': 'test'} test_path = 'a/b/c' source_locator.EclipseModuleData(mod_name, mod_info, test_path) self.assertTrue(mock_base_init.called) self.assertTrue(mock_method.called) @mock.patch.object(source_locator.ModuleData, '_collect_missing_jars') @mock.patch.object(source_locator.ModuleData, '_collect_classes_jars') @mock.patch.object(source_locator.EclipseModuleData, '_locate_jar_path') @mock.patch.object(source_locator.EclipseModuleData, '_locate_project_source_path') def test_locate_sources_path(self, mock_src, mock_jar, mock_class_jar, mock_missing_jar): """Test locate_sources_path.""" mod_name = 'test' mod_info = {'name': 'test'} test_path = 'a/b/c' mod_data = source_locator.EclipseModuleData(mod_name, mod_info, test_path) mod_data.is_project = True mod_data.locate_sources_path() self.assertTrue(mock_src.called) self.assertTrue(mock_class_jar.called) self.assertTrue(mock_missing_jar.called) mock_src.reset() mock_jar.reset() mod_data.is_project = False mod_data.locate_sources_path() self.assertTrue(mock_jar.called) @mock.patch.object(source_locator.ModuleData, '_collect_srcs_paths') @mock.patch.object(source_locator.ModuleData, '_collect_r_srcs_paths') def test_locate_project_source_path(self, mock_src, mock_r): """Test _locate_project_source_path.""" mod_name = 'test' mod_info = {'name': 'test'} test_path = 'a/b/c' mod_data = source_locator.EclipseModuleData(mod_name, mod_info, test_path) mod_data._locate_project_source_path() self.assertTrue(mock_src.called) self.assertTrue(mock_r.called) @mock.patch.object(source_locator.ModuleData, '_append_classes_jar') @mock.patch.object(source_locator.ModuleData, '_check_key') @mock.patch.object(source_locator.ModuleData, '_set_jars_jarfile') @mock.patch.object(source_locator.ModuleData, '_check_jars_exist') @mock.patch.object(source_locator.ModuleData, '_append_jar_from_installed') @mock.patch.object(source_locator.ModuleData, '_check_jarjar_rules_exist') def test_locate_jar_path(self, mock_jarjar, mock_append_jar, mock_check_jar, mock_set_jar, mock_check_key, mock_append_class): """Test _locate_jar_path.""" mod_name = 'test' mod_info = {'name': 'test', 'path': 'x/y'} test_path = 'a/b/c' mod_data = source_locator.EclipseModuleData(mod_name, mod_info, test_path) mock_jarjar.return_value = False mock_check_jar.return_value = False mock_check_key.return_value = False mod_data._locate_jar_path() self.assertTrue(mock_append_jar.called) self.assertFalse(mock_set_jar.called) self.assertFalse(mock_append_class.called) mock_append_jar.reset_mock() mock_jarjar.return_value = False mock_check_jar.return_value = False mock_check_key.return_value = True mod_data._locate_jar_path() self.assertFalse(mock_append_jar.called) self.assertFalse(mock_set_jar.called) self.assertTrue(mock_append_class.called) mock_check_key.reset_mock() mock_append_class.reset_mock() mock_append_jar.reset_mock() mock_jarjar.return_value = False mock_check_jar.return_value = True mod_data._locate_jar_path() self.assertFalse(mock_append_jar.called) self.assertTrue(mock_set_jar.called) self.assertFalse(mock_check_key.called) self.assertFalse(mock_append_class.called) mock_append_jar.reset_mock() mock_set_jar.reset_mock() mock_check_jar.reset_mock() mock_check_key.reset_mock() mock_append_class.reset_mock() mock_jarjar.return_value = True mod_data._locate_jar_path() self.assertTrue(mock_append_jar.called) self.assertFalse(mock_check_jar.called) self.assertFalse(mock_set_jar.called) self.assertFalse(mock_check_key.called) self.assertFalse(mock_append_class.called) def test_add_to_source_or_test_dirs(self): """Test _add_to_source_or_test_dirs.""" mod_name = 'test' mod_info = {'name': 'test'} test_path = 'a/b/c' mod_data = source_locator.EclipseModuleData(mod_name, mod_info, test_path) mod_data._add_to_source_or_test_dirs('libcore/ojluni/src/lambda/java') self.assertEqual(mod_data.src_dirs, []) mod_data._add_to_source_or_test_dirs('a') self.assertEqual(mod_data.src_dirs, ['a']) if __name__ == '__main__': unittest.main()