From 3eb4c58f17aeeb24a26ca6027a26f136d10a5643 Mon Sep 17 00:00:00 2001 From: Jens Scheffler <95105677+jscheffl@users.noreply.github.com> Date: Tue, 10 Dec 2024 23:25:36 +0100 Subject: [PATCH] Fix Trino Provider intergation test after removal of deprecations (#44829) --- providers/tests/integration/trino/hooks/test_trino.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/providers/tests/integration/trino/hooks/test_trino.py b/providers/tests/integration/trino/hooks/test_trino.py index 1faf183117b76..d5a8b6d938a65 100644 --- a/providers/tests/integration/trino/hooks/test_trino.py +++ b/providers/tests/integration/trino/hooks/test_trino.py @@ -50,7 +50,9 @@ def test_should_record_records_with_kerberos_auth(self): @mock.patch.dict("os.environ", AIRFLOW_CONN_TRINO_DEFAULT="trino://airflow@trino:8080/") def test_openlineage_methods(self): - op = SQLExecuteQueryOperator(task_id="trino_test", sql="SELECT name FROM tpch.sf1.customer LIMIT 3") + op = SQLExecuteQueryOperator( + task_id="trino_test", sql="SELECT name FROM tpch.sf1.customer LIMIT 3", conn_id="trino_default" + ) op.execute({}) lineage = op.get_openlineage_facets_on_start() assert lineage.inputs[0].namespace == "trino://trino:8080"