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- <laipz8200@outlook.com>
pull/19430/head
-LAN- 1 year ago
parent efcad48e91
commit 38af41e3be
No known key found for this signature in database
GPG Key ID: 6BA0D108DED011FF

@ -10,6 +10,7 @@ from sqlalchemy.orm import Session, sessionmaker
from core.repositories import SQLAlchemyWorkflowNodeExecutionRepository from core.repositories import SQLAlchemyWorkflowNodeExecutionRepository
from core.workflow.repository.workflow_node_execution_repository import OrderConfig from core.workflow.repository.workflow_node_execution_repository import OrderConfig
from models.account import Account, Tenant
from models.workflow import WorkflowNodeExecution from models.workflow import WorkflowNodeExecution
@ -44,12 +45,16 @@ def session():
@pytest.fixture @pytest.fixture
def mock_user(): def mock_user():
"""Create a mock user for testing.""" """Create a user instance for testing."""
user = MagicMock() user = Account()
user.tenant_id = "test-tenant"
user.id = "test-user-id" 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 return user

Loading…
Cancel
Save