def mysql_operator_test(self): sql = """ CREATE TABLE IF NOT EXISTS test_airflow ( dummy VARCHAR(50) ); """ t = operators.MySqlOperator( task_id='basic_mysql', sql=sql, dag=self.dag) t.run(start_date=DEFAULT_DATE, end_date=DEFAULT_DATE, force=True)
def mysql_operator_test_multi(self): sql = [ "TRUNCATE TABLE test_airflow", "INSERT INTO test_airflow VALUES ('X')", ] t = operators.MySqlOperator(task_id='basic_mysql', sql=sql, dag=self.dag) t.run(start_date=DEFAULT_DATE, end_date=DEFAULT_DATE, force=True)