From 1c538e6cab9051c20abf71d5ebfe935c7b2a2c91 Mon Sep 17 00:00:00 2001
From: Ankit Chaurasia <8670962+sunank200@users.noreply.github.com>
Date: Wed, 5 Jun 2024 16:55:26 +0545
Subject: [PATCH] Fix the tests

---
 tests/sensors/test_time_sensor.py | 1 -
 tests/triggers/test_temporal.py   | 2 +-
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/tests/sensors/test_time_sensor.py b/tests/sensors/test_time_sensor.py
index 54a0212a247a9..bcf4700742a14 100644
--- 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
index 6e8d32c467e63..d0edc5e8d1e5a 100644
--- 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():