def test_mysql_operator_test(self): sql = """ CREATE TABLE IF NOT EXISTS test_airflow ( dummy VARCHAR(50) ); """ from airflow.operators.mysql_operator import MySqlOperator t = MySqlOperator(task_id='basic_mysql', sql=sql, dag=self.dag) t.run(start_date=DEFAULT_DATE, end_date=DEFAULT_DATE, ignore_ti_state=True)
def test_mysql_operator_test_multi(self): sql = [ "TRUNCATE TABLE test_airflow", "INSERT INTO test_airflow VALUES ('X')", ] from airflow.operators.mysql_operator import MySqlOperator t = MySqlOperator( task_id='mysql_operator_test_multi', mysql_conn_id='airflow_db', sql=sql, dag=self.dag) t.run(start_date=DEFAULT_DATE, end_date=DEFAULT_DATE, ignore_ti_state=True)
def test_mysql_operator_test(self): sql = """ CREATE TABLE IF NOT EXISTS test_airflow ( dummy VARCHAR(50) ); """ from airflow.operators.mysql_operator import MySqlOperator t = MySqlOperator( task_id='basic_mysql', sql=sql, dag=self.dag) t.run(start_date=DEFAULT_DATE, end_date=DEFAULT_DATE, ignore_ti_state=True)
def test_mysql_operator_test_multi(self): sql = [ "CREATE TABLE IF NOT EXISTS test_airflow (dummy VARCHAR(50))", "TRUNCATE TABLE test_airflow", "INSERT INTO test_airflow VALUES ('X')", ] from airflow.operators.mysql_operator import MySqlOperator op = MySqlOperator( task_id='mysql_operator_test_multi', sql=sql, dag=self.dag, ) op.run(start_date=DEFAULT_DATE, end_date=DEFAULT_DATE, ignore_ti_state=True)
def test_mysql_operator_test_multi(self): sql = [ "CREATE TABLE IF NOT EXISTS test_airflow (dummy VARCHAR(50))", "TRUNCATE TABLE test_airflow", "INSERT INTO test_airflow VALUES ('X')", ] from airflow.operators.mysql_operator import MySqlOperator t = MySqlOperator( task_id='mysql_operator_test_multi', sql=sql, dag=self.dag, ) t.run(start_date=DEFAULT_DATE, end_date=DEFAULT_DATE, ignore_ti_state=True)
def test_overwrite_schema(self): """ Verifies option to overwrite connection schema """ from airflow.operators.mysql_operator import MySqlOperator sql = "SELECT 1;" t = MySqlOperator( task_id='test_mysql_operator_test_schema_overwrite', sql=sql, dag=self.dag, database="foobar", ) from _mysql_exceptions import OperationalError try: t.run(start_date=DEFAULT_DATE, end_date=DEFAULT_DATE, ignore_ti_state=True) except OperationalError as e: assert "Unknown database 'foobar'" in str(e)