From 38af41e3bed4d03f3534abf3777a77fb9089589f Mon Sep 17 00:00:00 2001 From: -LAN- Date: Wed, 14 May 2025 15:35:02 +0800 Subject: [PATCH] test(test_sqlalchemy_repository): Refactors mock user creation in tests Updates the mock user setup in unit tests to use actual Account and Tenant instances instead of MagicMock. This change enhances realism in tests by reflecting actual class structures and relationships, aiding in more accurate test outcomes. Signed-off-by: -LAN- --- .../test_sqlalchemy_repository.py | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) 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 73809e22b5..cecdcfd20f 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 @@ -10,6 +10,7 @@ from sqlalchemy.orm import Session, sessionmaker from core.repositories import SQLAlchemyWorkflowNodeExecutionRepository from core.workflow.repository.workflow_node_execution_repository import OrderConfig +from models.account import Account, Tenant from models.workflow import WorkflowNodeExecution @@ -44,12 +45,16 @@ def session(): @pytest.fixture def mock_user(): - """Create a mock user for testing.""" - user = MagicMock() - user.tenant_id = "test-tenant" + """Create a user instance for testing.""" + user = Account() user.id = "test-user-id" - # Set up to be recognized as an Account - user.__class__.__name__ = "Account" + + tenant = Tenant() + tenant.id = "test-tenant" + tenant.name = "Test Workspace" + user._current_tenant = MagicMock() + user._current_tenant.id = "test-tenant" + return user