def test_get_possible_mapreduce_config_from(self, load_json_file, load_hadoop_json_for_tag, get_poss_mr_configs): expected_config = {'configs': []} actual_config = ch_helper.get_possible_mapreduce_config_from( 'sample-file-name.json') load_hadoop_json_for_tag.assert_called_once_with( 'sample-file-name.json', 'mapred-site.xml') get_poss_mr_configs.assert_called_once_with() self.assertEqual(expected_config, actual_config)
def test_get_possible_mapreduce_config_from(self, load_json_file, load_hadoop_json_for_tag, get_poss_mr_configs): expected_config = { 'configs': [] } actual_config = ch_helper.get_possible_mapreduce_config_from( 'sample-file-name.json') load_hadoop_json_for_tag.assert_called_once_with( 'sample-file-name.json', 'mapred-site.xml') get_poss_mr_configs.assert_called_once_with() self.assertEqual(expected_config, actual_config)
def get_possible_job_config(job_type): if edp.compare_job_type(job_type, edp.JOB_TYPE_HIVE): return {'job_config': ch_helper.get_possible_hive_config_from( 'plugins/hdp/versions/version_1_3_2/resources/' 'ambari-config-resource.json')} if edp.compare_job_type(job_type, edp.JOB_TYPE_MAPREDUCE, edp.JOB_TYPE_MAPREDUCE_STREAMING): return {'job_config': ch_helper.get_possible_mapreduce_config_from( 'plugins/hdp/versions/version_1_3_2/resources/' 'ambari-config-resource.json')} if edp.compare_job_type(job_type, edp.JOB_TYPE_PIG): return {'job_config': ch_helper.get_possible_pig_config_from( 'plugins/hdp/versions/version_1_3_2/resources/' 'ambari-config-resource.json')} return edp_engine.EdpOozieEngine.get_possible_job_config(job_type)