diff --git a/api/tests/unit_tests/core/workflow/graph_engine/event_management/test_event_handlers.py b/api/tests/unit_tests/core/workflow/graph_engine/event_management/test_event_handlers.py index 2b8f04979d..5d17b7a243 100644 --- a/api/tests/unit_tests/core/workflow/graph_engine/event_management/test_event_handlers.py +++ b/api/tests/unit_tests/core/workflow/graph_engine/event_management/test_event_handlers.py @@ -2,8 +2,6 @@ from __future__ import annotations -from datetime import datetime - from core.workflow.enums import NodeExecutionType, NodeState, NodeType, WorkflowNodeExecutionStatus from core.workflow.graph import Graph from core.workflow.graph_engine.domain.graph_execution import GraphExecution @@ -16,6 +14,7 @@ from core.workflow.graph_events import NodeRunRetryEvent, NodeRunStartedEvent from core.workflow.node_events import NodeRunResult from core.workflow.nodes.base.entities import RetryConfig from core.workflow.runtime import GraphRuntimeState, VariablePool +from libs.datetime_utils import naive_utc_now class _StubEdgeProcessor: @@ -75,7 +74,7 @@ def test_retry_does_not_emit_additional_start_event() -> None: execution_id = "exec-1" node_type = NodeType.CODE - start_time = datetime.utcnow() + start_time = naive_utc_now() start_event = NodeRunStartedEvent( id=execution_id, diff --git a/api/tests/unit_tests/core/workflow/graph_engine/orchestration/test_dispatcher.py b/api/tests/unit_tests/core/workflow/graph_engine/orchestration/test_dispatcher.py index e6d4508fdf..c1fc4acd73 100644 --- a/api/tests/unit_tests/core/workflow/graph_engine/orchestration/test_dispatcher.py +++ b/api/tests/unit_tests/core/workflow/graph_engine/orchestration/test_dispatcher.py @@ -3,7 +3,6 @@ from __future__ import annotations import queue -from datetime import datetime from unittest import mock from core.workflow.entities.pause_reason import SchedulingPause @@ -18,6 +17,7 @@ from core.workflow.graph_events import ( NodeRunSucceededEvent, ) from core.workflow.node_events import NodeRunResult +from libs.datetime_utils import naive_utc_now def test_dispatcher_should_consume_remains_events_after_pause(): @@ -109,7 +109,7 @@ def _make_started_event() -> NodeRunStartedEvent: node_id="node-1", node_type=NodeType.CODE, node_title="Test Node", - start_at=datetime.utcnow(), + start_at=naive_utc_now(), ) @@ -119,7 +119,7 @@ def _make_succeeded_event() -> NodeRunSucceededEvent: node_id="node-1", node_type=NodeType.CODE, node_title="Test Node", - start_at=datetime.utcnow(), + start_at=naive_utc_now(), node_run_result=NodeRunResult(status=WorkflowNodeExecutionStatus.SUCCEEDED), ) @@ -153,7 +153,7 @@ def test_dispatcher_drain_event_queue(): node_id="node-1", node_type=NodeType.CODE, node_title="Code", - start_at=datetime.utcnow(), + start_at=naive_utc_now(), ), NodeRunPauseRequestedEvent( id="pause-event", @@ -165,7 +165,7 @@ def test_dispatcher_drain_event_queue(): id="success-event", node_id="node-1", node_type=NodeType.CODE, - start_at=datetime.utcnow(), + start_at=naive_utc_now(), node_run_result=NodeRunResult(status=WorkflowNodeExecutionStatus.SUCCEEDED), ), ]