def test_invalid_cmd_type(self): """ Tests to verify if the cmd_type is not import or export, an exception is raised. """ operator = SqoopOperator(task_id='sqoop_job', dag=self.dag, cmd_type='invalid') with self.assertRaises(AirflowException): operator.execute({})
def test_invalid_import_options(self): """ Tests to verify if a user passes both a query and a table then an exception is raised. """ import_query_and_table_configs = self._config.copy() import_query_and_table_configs['cmd_type'] = 'import' operator = SqoopOperator(task_id='sqoop_job', dag=self.dag, **import_query_and_table_configs) with self.assertRaises(AirflowException): operator.execute({})