Fix the tests

astronomer · Jun 6, 2024 · 1c538e6 · 1c538e6
1 parent 0d45a54
commit 1c538e6
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/tests/sensors/test_time_sensor.py b/tests/sensors/test_time_sensor.py
@@ -63,7 +63,6 @@ def test_task_is_deferred(self):
 
         assert isinstance(exc_info.value.trigger, DateTimeTrigger)
         assert exc_info.value.trigger.moment == timezone.datetime(2020, 7, 7, 10)
-        assert exc_info.value.method_name == "execute_complete"
         assert exc_info.value.kwargs is None
 
     def test_target_time_aware(self):

diff --git a/tests/triggers/test_temporal.py b/tests/triggers/test_temporal.py
@@ -56,7 +56,7 @@ def test_datetime_trigger_serialization():
     trigger = DateTimeTrigger(moment)
     classpath, kwargs = trigger.serialize()
     assert classpath == "airflow.triggers.temporal.DateTimeTrigger"
-    assert kwargs == {"moment": moment}
+    assert kwargs == {"moment": moment, "end_task": False}
 
 
 def test_timedelta_trigger_serialization():