Fix failing tests from #9124 (#9356)

apache · Jun 17, 2020 · 281095e · 281095e
1 parent 66e7382
commit 281095e
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 5 deletions.
diff --git a/airflow/operators/check_operator.py b/airflow/operators/check_operator.py
@@ -24,6 +24,11 @@
     SQLCheckOperator, SQLIntervalCheckOperator, SQLThresholdCheckOperator, SQLValueCheckOperator,
 )
 
+warnings.warn(
+    "This module is deprecated. Please use `airflow.operators.sql`.",
+    DeprecationWarning, stacklevel=2
+)
+
 
 class CheckOperator(SQLCheckOperator):
     """

diff --git a/airflow/operators/sql_branch_operator.py b/airflow/operators/sql_branch_operator.py
@@ -19,6 +19,11 @@
 
 from airflow.operators.sql import BranchSQLOperator
 
+warnings.warn(
+    """This module is deprecated. Please use `airflow.operators.sql`.""",
+    DeprecationWarning, stacklevel=2
+)
+
 
 class BranchSqlOperator(BranchSQLOperator):
     """

diff --git a/tests/test_core_to_contrib.py b/tests/test_core_to_contrib.py
@@ -1170,24 +1170,24 @@
         'airflow.operators.presto_check_operator.PrestoValueCheckOperator',
     ),
     (
-        'airflow.operators.check_operator.CheckOperator',
         'airflow.operators.sql.SQLCheckOperator',
+        'airflow.operators.check_operator.CheckOperator',
     ),
     (
-        'airflow.operators.check_operator.IntervalCheckOperator',
         'airflow.operators.sql.SQLIntervalCheckOperator',
+        'airflow.operators.check_operator.IntervalCheckOperator',
     ),
     (
-        'airflow.operators.check_operator.ValueCheckOperator',
         'airflow.operators.sql.SQLValueCheckOperator',
+        'airflow.operators.check_operator.ValueCheckOperator',
     ),
     (
-        'airflow.operators.check_operator.ThresholdCheckOperator',
         'airflow.operators.sql.SQLThresholdCheckOperator',
+        'airflow.operators.check_operator.ThresholdCheckOperator',
     ),
     (
-        'airflow.operators.sql_branch_operator.BranchSqlOperator',
         'airflow.operators.sql.BranchSQLOperator',
+        'airflow.operators.sql_branch_operator.BranchSqlOperator',
     ),
     (
         'airflow.operators.python.BranchPythonOperator',