This is an automated email from the ASF dual-hosted git repository.

potiuk pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git


The following commit(s) were added to refs/heads/main by this push:
     new 62444d8cff Resolve openlineage provider deprecations in tests (#40133)
62444d8cff is described below

commit 62444d8cff8f91e77da1363d38e88f6ed2418a9a
Author: Gopal Dirisala <[email protected]>
AuthorDate: Sat Jun 8 15:35:58 2024 +0530

    Resolve openlineage provider deprecations in tests (#40133)
---
 tests/deprecations_ignore.yml                        | 3 ---
 tests/providers/openlineage/plugins/test_listener.py | 2 +-
 tests/providers/openlineage/plugins/test_utils.py    | 2 +-
 3 files changed, 2 insertions(+), 5 deletions(-)

diff --git a/tests/deprecations_ignore.yml b/tests/deprecations_ignore.yml
index 57dbb7a605..4eb460674c 100644
--- a/tests/deprecations_ignore.yml
+++ b/tests/deprecations_ignore.yml
@@ -367,9 +367,6 @@
 - 
tests/providers/mysql/operators/test_mysql.py::TestMySql::test_mysql_operator_test_multi
 - 
tests/providers/mysql/operators/test_mysql.py::TestMySql::test_overwrite_schema
 - 
tests/providers/mysql/operators/test_mysql.py::test_execute_openlineage_events
-- 
tests/providers/openlineage/plugins/test_listener.py::test_listener_does_not_change_task_instance
-- tests/providers/openlineage/plugins/test_utils.py::test_get_dagrun_start_end
-- tests/providers/openlineage/utils/test_utils.py::test_get_custom_facets
 - 
tests/providers/oracle/hooks/test_oracle.py::TestOracleHookConn::test_get_conn_events
 - 
tests/providers/oracle/hooks/test_oracle.py::TestOracleHookConn::test_get_conn_host
 - 
tests/providers/oracle/hooks/test_oracle.py::TestOracleHookConn::test_get_conn_host_alternative_port
diff --git a/tests/providers/openlineage/plugins/test_listener.py 
b/tests/providers/openlineage/plugins/test_listener.py
index 15d928ec03..572a6877f3 100644
--- a/tests/providers/openlineage/plugins/test_listener.py
+++ b/tests/providers/openlineage/plugins/test_listener.py
@@ -71,7 +71,7 @@ def test_listener_does_not_change_task_instance(render_mock, 
xcom_push_mock):
         "test",
         start_date=dt.datetime(2022, 1, 1),
         user_defined_macros={"render_df": render_df},
-        params={"df": render_df()},
+        params={"df": {"col": [1, 2]}},
     )
     t = TemplateOperator(task_id="template_op", dag=dag, do_xcom_push=True, 
df=dag.param("df"))
     run_id = str(uuid.uuid1())
diff --git a/tests/providers/openlineage/plugins/test_utils.py 
b/tests/providers/openlineage/plugins/test_utils.py
index 6062e87b8d..d3e54504ed 100644
--- a/tests/providers/openlineage/plugins/test_utils.py
+++ b/tests/providers/openlineage/plugins/test_utils.py
@@ -58,7 +58,7 @@ class SafeStrDict(dict):
 def test_get_dagrun_start_end():
     start_date = datetime.datetime(2022, 1, 1)
     end_date = datetime.datetime(2022, 1, 1, hour=2)
-    dag = AIRFLOW_DAG("test", start_date=start_date, end_date=end_date, 
schedule_interval="@once")
+    dag = AIRFLOW_DAG("test", start_date=start_date, end_date=end_date, 
schedule="@once")
     AIRFLOW_DAG.bulk_write_to_db([dag])
     dag_model = DagModel.get_dagmodel(dag.dag_id)
     run_id = str(uuid.uuid1())

Reply via email to