def test_hive_airflow_default_config_queue_override(self):
specific_mapred_queue = 'default'
op = MockHiveOperator(
task_id='test_default_config_queue',
hql=self.hql,
mapred_queue=specific_mapred_queue,
mapred_queue_priority='HIGH',
mapred_job_name='airflow.test_default_config_queue',
dag=self.dag)
self.assertEqual(op.get_hook().mapred_queue, specific_mapred_queue)
def test_hive_airflow_default_config_queue(self):
op = MockHiveOperator(
task_id='test_default_config_queue',
hql=self.hql,
mapred_queue_priority='HIGH',
mapred_job_name='airflow.test_default_config_queue',
dag=self.dag,
)
# just check that the correct default value in test_default.cfg is used
test_config_hive_mapred_queue = conf.get('hive', 'default_hive_mapred_queue')
self.assertEqual(op.get_hook().mapred_queue, test_config_hive_mapred_queue)