test(providers/databrick): fix execute_complete test cases

apache · Jul 20, 2023 · f5f15d8 · f5f15d8
1 parent 72e7046
commit f5f15d8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/providers/databricks/operators/test_databricks.py b/tests/providers/databricks/operators/test_databricks.py
@@ -517,7 +517,7 @@ def test_execute_complete_failure(self, db_mock_class):
             op.execute_complete(context=None, event=event)
 
     def test_execute_complete_incorrect_event_validation_failure(self):
-        event = {}
+        event = {"event_id": "no such column"}
         op = DatabricksSubmitRunDeferrableOperator(task_id=TASK_ID)
         with pytest.raises(AirflowException):
             op.execute_complete(context=None, event=event)
@@ -951,7 +951,7 @@ def test_execute_complete_failure(self, db_mock_class):
             op.execute_complete(context=None, event=event)
 
     def test_execute_complete_incorrect_event_validation_failure(self):
-        event = {}
+        event = {"event_id": "no such column"}
         op = DatabricksRunNowDeferrableOperator(task_id=TASK_ID, job_id=JOB_ID)
         with pytest.raises(AirflowException):
             op.execute_complete(context=None, event=event)