def test_hive_airflow_default_config_queue_override(self): specific_mapred_queue = 'default' op = HiveOperator(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 = HiveOperator(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)