diff --git a/providers/edge3/tests/unit/edge3/executors/test_edge_executor.py b/providers/edge3/tests/unit/edge3/executors/test_edge_executor.py index 947264ea39503..7c86b39f12dd4 100644 --- a/providers/edge3/tests/unit/edge3/executors/test_edge_executor.py +++ b/providers/edge3/tests/unit/edge3/executors/test_edge_executor.py @@ -21,6 +21,7 @@ from unittest.mock import MagicMock, patch import pytest +import time_machine from airflow.configuration import conf from airflow.models.taskinstancekey import TaskInstanceKey @@ -275,9 +276,7 @@ def test_sync_active_worker(self): ) session.commit() - with patch( - "airflow.utils.timezone.utcnow", return_value=datetime(2023, 1, 1, 1, 0, 0, tzinfo=timezone.utc) - ): + with time_machine.travel(datetime(2023, 1, 1, 1, 0, 0, tzinfo=timezone.utc), tick=False): with conf_vars({("edge", "heartbeat_interval"): "10"}): executor.sync()