Fix Trino Provider intergation test after removal of deprecations (#4…

…4829)
apache · Dec 10, 2024 · 3eb4c58 · 3eb4c58
1 parent 102af55
commit 3eb4c58
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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"