def test_get_path_to_module_info(self, mock_load_module): """Test that we correctly create the path to module info dict.""" mod_one = 'mod1' mod_two = 'mod2' mod_path_one = '/path/to/mod1' mod_path_two = '/path/to/mod2' mod_info_dict = { mod_one: { constants.MODULE_PATH: [mod_path_one] }, mod_two: { constants.MODULE_PATH: [mod_path_two] } } mock_load_module.return_value = ('mod_target', mod_info_dict) path_to_mod_info = { mod_path_one: [{ constants.MODULE_NAME: mod_one, constants.MODULE_PATH: [mod_path_one] }], mod_path_two: [{ constants.MODULE_NAME: mod_two, constants.MODULE_PATH: [mod_path_two] }] } mod_info = module_info.ModuleInfo() self.assertDictEqual(path_to_mod_info, mod_info._get_path_to_module_info(mod_info_dict))
def test_get_path(self): """Test that we get the module path when it's properly loaded.""" # Load up the test json file and check that module is in it mod_info = module_info.ModuleInfo(module_file=JSON_FILE_PATH) self.assertEqual(mod_info.get_paths(EXPECTED_MOD_TARGET), EXPECTED_MOD_TARGET_PATH) self.assertEqual(mod_info.get_paths(MOD_NO_PATH), [])
def main(argv): """Entry point of atest script. Args: argv: A list of arguments. Returns: Exit code. """ args = _parse_args(argv) _configure_logging(args.verbose) if _missing_environment_variables(): return constants.EXIT_CODE_ENV_NOT_SETUP if args.generate_baseline and args.generate_new_metrics: logging.error( 'Cannot collect both baseline and new metrics at the same time.') return constants.EXIT_CODE_ERROR if not _has_valid_regression_detection_args(args): return constants.EXIT_CODE_ERROR results_dir = make_test_run_dir() mod_info = module_info.ModuleInfo(force_build=args.rebuild_module_info) translator = cli_translator.CLITranslator(module_info=mod_info) build_targets = set() test_infos = set() if _will_run_tests(args): try: build_targets, test_infos = translator.translate(args.tests) except atest_error.TestDiscoveryException: logging.exception('Error occured in test discovery:') logging.info( 'This can happen after a repo sync or if the test is ' 'new. Running: with "%s" may resolve the issue.', REBUILD_MODULE_INFO_FLAG) return constants.EXIT_CODE_TEST_NOT_FOUND build_targets |= test_runner_handler.get_test_runner_reqs( mod_info, test_infos) extra_args = get_extra_args(args) if args.detect_regression: build_targets |= (regression_test_runner.RegressionTestRunner( '').get_test_runner_build_reqs()) # args.steps will be None if none of -bit set, else list of params set. steps = args.steps if args.steps else ALL_STEPS if build_targets and BUILD_STEP in steps: # Add module-info.json target to the list of build targets to keep the # file up to date. build_targets.add(mod_info.module_info_target) success = atest_utils.build(build_targets, args.verbose) if not success: return constants.EXIT_CODE_BUILD_FAILURE elif TEST_STEP not in steps: logging.warn('Install step without test step currently not ' 'supported, installing AND testing instead.') steps.append(TEST_STEP) if TEST_STEP in steps: test_runner_handler.run_all_tests(results_dir, test_infos, extra_args) if args.detect_regression: regression_args = _get_regression_detection_args(args, results_dir) regression_test_runner.RegressionTestRunner('').run_tests( None, regression_args) return constants.EXIT_CODE_SUCCESS
def test_load_mode_info_file_out_dir_handling(self, _isfile, _open, _json): """Test _load_module_info_file out dir handling.""" # Test out default out dir is used. build_top = '/path/to/top' default_out_dir = os.path.join(build_top, 'out/dir/here') os_environ_mock = { constants.ANDROID_OUT: default_out_dir, 'ANDROID_PRODUCT_OUT': default_out_dir, constants.ANDROID_BUILD_TOP: build_top } default_out_dir_mod_targ = 'out/dir/here/module-info.json' # Make sure module_info_target is what we think it is. with mock.patch.dict('os.environ', os_environ_mock, clear=True): mod_info = module_info.ModuleInfo() self.assertEqual(default_out_dir_mod_targ, mod_info.module_info_target) # Test out custom out dir is used (OUT_DIR=dir2). custom_out_dir = os.path.join(build_top, 'out2/dir/here') os_environ_mock = { constants.ANDROID_OUT: custom_out_dir, 'OUT_DIR': 'out2', 'ANDROID_PRODUCT_OUT': custom_out_dir, constants.ANDROID_BUILD_TOP: build_top } custom_out_dir_mod_targ = 'out2/dir/here/module-info.json' # Make sure module_info_target is what we think it is. with mock.patch.dict('os.environ', os_environ_mock, clear=True): mod_info = module_info.ModuleInfo() self.assertEqual(custom_out_dir_mod_targ, mod_info.module_info_target) # Test out custom abs out dir is used (OUT_DIR=/tmp/out/dir2). abs_custom_out_dir = '/tmp/out/dir' os_environ_mock = { constants.ANDROID_OUT: abs_custom_out_dir, 'OUT_DIR': abs_custom_out_dir, 'ANDROID_PRODUCT_OUT': abs_custom_out_dir, constants.ANDROID_BUILD_TOP: build_top } custom_abs_out_dir_mod_targ = '/tmp/out/dir/module-info.json' # Make sure module_info_target is what we think it is. with mock.patch.dict('os.environ', os_environ_mock, clear=True): mod_info = module_info.ModuleInfo() self.assertEqual(custom_abs_out_dir_mod_targ, mod_info.module_info_target)
def test_get_module_names(self): """test that we get the module name properly.""" mod_info = module_info.ModuleInfo(module_file=JSON_FILE_PATH) self.assertEqual( mod_info.get_module_names(EXPECTED_MOD_TARGET_PATH[0]), [EXPECTED_MOD_TARGET]) self.assertEqual(mod_info.get_module_names(PATH_TO_MULT_MODULES), MULT_MOODULES_WITH_SHARED_PATH)
def test_is_module(self): """Test that we get the module when it's properly loaded.""" # Load up the test json file and check that module is in it mod_info = module_info.ModuleInfo(module_file=JSON_FILE_PATH) self.assertTrue(mod_info.is_module(EXPECTED_MOD_TARGET)) self.assertFalse(mod_info.is_module(UNEXPECTED_MOD_TARGET))