Mark db tests for tests introduced in PR apache#40724

astronomer · Jul 18, 2024 · 8966635 · 8966635
1 parent 3083147
commit 8966635
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/providers/databricks/plugins/test_databricks_workflow.py b/tests/providers/databricks/plugins/test_databricks_workflow.py
@@ -126,6 +126,7 @@ def app():
         yield app
 
 
+@pytest.mark.db_test
 def test_get_task_instance(app):
     with app.app_context():
         operator = Mock()
@@ -143,6 +144,7 @@ def test_get_task_instance(app):
             assert result == dag_run
 
 
+@pytest.mark.db_test
 def test_workflow_job_run_link(app):
     with app.app_context():
         link = WorkflowJobRunLink()
@@ -177,6 +179,7 @@ def test_workflow_job_run_link(app):
                         assert "https://mockhost/#job/1/run/1" in result
 
 
+@pytest.mark.db_test
 def test_workflow_job_repair_single_failed_link(app):
     with app.app_context():
         link = WorkflowJobRepairSingleTaskLink()