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)