From 01518e83121cbd3a6a601ab9819a9514670f6faa Mon Sep 17 00:00:00 2001 From: -LAN- Date: Fri, 9 May 2025 13:14:55 +0800 Subject: [PATCH] Update api/tests/unit_tests/repositories/workflow_node_execution/test_sqlalchemy_repository.py Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com> --- .../workflow_node_execution/test_sqlalchemy_repository.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/api/tests/unit_tests/repositories/workflow_node_execution/test_sqlalchemy_repository.py b/api/tests/unit_tests/repositories/workflow_node_execution/test_sqlalchemy_repository.py index 20dd29bc22..608cc5aa25 100644 --- a/api/tests/unit_tests/repositories/workflow_node_execution/test_sqlalchemy_repository.py +++ b/api/tests/unit_tests/repositories/workflow_node_execution/test_sqlalchemy_repository.py @@ -210,7 +210,8 @@ def test_get_db_models_by_workflow_run(repository, session, mocker: MockerFixtur assert result[0] is mock_execution # Verify that _to_domain_model was NOT called (since we're returning raw DB models) - assert not hasattr(repository, "_to_domain_model.assert_not_called") or not repository._to_domain_model.called + if hasattr(repository, "_to_domain_model"): + repository._to_domain_model.assert_not_called() def test_get_running_executions(repository, session, mocker: MockerFixture):