def test_get_data_sources(self, ds): job, job_exec = _create_all_stack(edp.JOB_TYPE_PIG) job_exec.input_id = 's1' job_exec.output_id = 's2' ds.side_effect = _conductor_data_source_get input_source, output_source = ( job_manager._get_data_sources(job_exec, job)) self.assertEqual('obj_s1', input_source) self.assertEqual('obj_s2', output_source)
def test_get_data_sources_java(self): configs = {sw.HADOOP_SWIFT_USERNAME: '******', sw.HADOOP_SWIFT_PASSWORD: '******'} configs = { 'configs': configs, 'args': ['swift://ex/i', 'output_path'] } job, job_exec = _create_all_stack(edp.JOB_TYPE_JAVA, configs) input_source, output_source = ( job_manager._get_data_sources(job_exec, job)) self.assertEqual(None, input_source) self.assertEqual(None, output_source)