style: normalize python formatting gates
This commit is contained in:
Xiaoxia AI
@@ -1,44 +1,43 @@
|
||||
"""
|
||||
移除成员 Use Case 测试
|
||||
"""
|
||||
import pytest
|
||||
|
||||
from unittest.mock import Mock
|
||||
|
||||
import pytest
|
||||
|
||||
from packages.application.workspace.remove_member_use_case import (
|
||||
RemoveMemberUseCase,
|
||||
RemoveMemberRequest,
|
||||
LeaveWorkspaceUseCase,
|
||||
LeaveWorkspaceRequest,
|
||||
LeaveWorkspaceUseCase,
|
||||
RemoveMemberRequest,
|
||||
RemoveMemberUseCase,
|
||||
)
|
||||
from packages.domain.entities import (
|
||||
Workspace,
|
||||
WorkspaceMember,
|
||||
WorkspaceMemberRole,
|
||||
)
|
||||
from packages.domain.entities import Workspace, WorkspaceMember, WorkspaceMemberRole
|
||||
|
||||
|
||||
class TestRemoveMemberUseCase:
|
||||
"""移除成员测试"""
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def mock_workspace_repo(self):
|
||||
repo = Mock()
|
||||
repo.find_by_id = Mock(return_value=None)
|
||||
return repo
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def mock_member_repo(self):
|
||||
repo = Mock()
|
||||
repo.find_by_workspace_and_user = Mock(return_value=None)
|
||||
repo.delete = Mock(return_value=True)
|
||||
return repo
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def use_case(self, mock_workspace_repo, mock_member_repo):
|
||||
return RemoveMemberUseCase(
|
||||
workspace_repository=mock_workspace_repo,
|
||||
workspace_member_repository=mock_member_repo,
|
||||
)
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def test_workspace(self):
|
||||
return Workspace(
|
||||
@@ -46,7 +45,7 @@ class TestRemoveMemberUseCase:
|
||||
name="Test Workspace",
|
||||
owner_user_id="owner-id",
|
||||
)
|
||||
|
||||
|
||||
def test_remove_member_success_by_owner(
|
||||
self,
|
||||
use_case,
|
||||
@@ -56,35 +55,38 @@ class TestRemoveMemberUseCase:
|
||||
):
|
||||
"""测试 Owner 移除成员成功"""
|
||||
mock_workspace_repo.find_by_id.return_value = test_workspace
|
||||
|
||||
|
||||
owner_member = WorkspaceMember(
|
||||
id="member-1",
|
||||
workspace_id="workspace-123",
|
||||
user_id="owner-id",
|
||||
role=WorkspaceMemberRole.OWNER,
|
||||
)
|
||||
|
||||
|
||||
target_member = WorkspaceMember(
|
||||
id="member-2",
|
||||
workspace_id="workspace-123",
|
||||
user_id="target-id",
|
||||
role=WorkspaceMemberRole.MEMBER,
|
||||
)
|
||||
|
||||
mock_member_repo.find_by_workspace_and_user.side_effect = [owner_member, target_member]
|
||||
|
||||
|
||||
mock_member_repo.find_by_workspace_and_user.side_effect = [
|
||||
owner_member,
|
||||
target_member,
|
||||
]
|
||||
|
||||
request = RemoveMemberRequest(
|
||||
workspace_id="workspace-123",
|
||||
requester_user_id="owner-id",
|
||||
target_user_id="target-id",
|
||||
)
|
||||
|
||||
|
||||
success, error = use_case.execute(request)
|
||||
|
||||
|
||||
assert success is True
|
||||
assert error is None
|
||||
mock_member_repo.delete.assert_called_once_with("member-2")
|
||||
|
||||
|
||||
def test_remove_member_success_by_admin(
|
||||
self,
|
||||
use_case,
|
||||
@@ -94,34 +96,37 @@ class TestRemoveMemberUseCase:
|
||||
):
|
||||
"""测试 Admin 移除普通成员成功"""
|
||||
mock_workspace_repo.find_by_id.return_value = test_workspace
|
||||
|
||||
|
||||
admin_member = WorkspaceMember(
|
||||
id="member-1",
|
||||
workspace_id="workspace-123",
|
||||
user_id="admin-id",
|
||||
role=WorkspaceMemberRole.ADMIN,
|
||||
)
|
||||
|
||||
|
||||
target_member = WorkspaceMember(
|
||||
id="member-2",
|
||||
workspace_id="workspace-123",
|
||||
user_id="target-id",
|
||||
role=WorkspaceMemberRole.VIEWER,
|
||||
)
|
||||
|
||||
mock_member_repo.find_by_workspace_and_user.side_effect = [admin_member, target_member]
|
||||
|
||||
|
||||
mock_member_repo.find_by_workspace_and_user.side_effect = [
|
||||
admin_member,
|
||||
target_member,
|
||||
]
|
||||
|
||||
request = RemoveMemberRequest(
|
||||
workspace_id="workspace-123",
|
||||
requester_user_id="admin-id",
|
||||
target_user_id="target-id",
|
||||
)
|
||||
|
||||
|
||||
success, error = use_case.execute(request)
|
||||
|
||||
|
||||
assert success is True
|
||||
assert error is None
|
||||
|
||||
|
||||
def test_remove_member_cannot_remove_owner(
|
||||
self,
|
||||
use_case,
|
||||
@@ -131,34 +136,37 @@ class TestRemoveMemberUseCase:
|
||||
):
|
||||
"""测试不能移除 Owner"""
|
||||
mock_workspace_repo.find_by_id.return_value = test_workspace
|
||||
|
||||
|
||||
admin_member = WorkspaceMember(
|
||||
id="member-1",
|
||||
workspace_id="workspace-123",
|
||||
user_id="admin-id",
|
||||
role=WorkspaceMemberRole.ADMIN,
|
||||
)
|
||||
|
||||
|
||||
owner_member = WorkspaceMember(
|
||||
id="member-2",
|
||||
workspace_id="workspace-123",
|
||||
user_id="owner-id",
|
||||
role=WorkspaceMemberRole.OWNER,
|
||||
)
|
||||
|
||||
mock_member_repo.find_by_workspace_and_user.side_effect = [admin_member, owner_member]
|
||||
|
||||
|
||||
mock_member_repo.find_by_workspace_and_user.side_effect = [
|
||||
admin_member,
|
||||
owner_member,
|
||||
]
|
||||
|
||||
request = RemoveMemberRequest(
|
||||
workspace_id="workspace-123",
|
||||
requester_user_id="admin-id",
|
||||
target_user_id="owner-id",
|
||||
)
|
||||
|
||||
|
||||
success, error = use_case.execute(request)
|
||||
|
||||
|
||||
assert success is False
|
||||
assert error == "Cannot remove the workspace owner"
|
||||
|
||||
|
||||
def test_remove_member_admin_cannot_remove_admin(
|
||||
self,
|
||||
use_case,
|
||||
@@ -168,34 +176,37 @@ class TestRemoveMemberUseCase:
|
||||
):
|
||||
"""测试 Admin 不能移除另一个 Admin"""
|
||||
mock_workspace_repo.find_by_id.return_value = test_workspace
|
||||
|
||||
|
||||
admin_member1 = WorkspaceMember(
|
||||
id="member-1",
|
||||
workspace_id="workspace-123",
|
||||
user_id="admin-id-1",
|
||||
role=WorkspaceMemberRole.ADMIN,
|
||||
)
|
||||
|
||||
|
||||
admin_member2 = WorkspaceMember(
|
||||
id="member-2",
|
||||
workspace_id="workspace-123",
|
||||
user_id="admin-id-2",
|
||||
role=WorkspaceMemberRole.ADMIN,
|
||||
)
|
||||
|
||||
mock_member_repo.find_by_workspace_and_user.side_effect = [admin_member1, admin_member2]
|
||||
|
||||
|
||||
mock_member_repo.find_by_workspace_and_user.side_effect = [
|
||||
admin_member1,
|
||||
admin_member2,
|
||||
]
|
||||
|
||||
request = RemoveMemberRequest(
|
||||
workspace_id="workspace-123",
|
||||
requester_user_id="admin-id-1",
|
||||
target_user_id="admin-id-2",
|
||||
)
|
||||
|
||||
|
||||
success, error = use_case.execute(request)
|
||||
|
||||
|
||||
assert success is False
|
||||
assert error == "Admins cannot remove other admins"
|
||||
|
||||
|
||||
def test_remove_member_cannot_remove_self(
|
||||
self,
|
||||
use_case,
|
||||
@@ -205,27 +216,27 @@ class TestRemoveMemberUseCase:
|
||||
):
|
||||
"""测试不能移除自己"""
|
||||
mock_workspace_repo.find_by_id.return_value = test_workspace
|
||||
|
||||
|
||||
admin_member = WorkspaceMember(
|
||||
id="member-1",
|
||||
workspace_id="workspace-123",
|
||||
user_id="admin-id",
|
||||
role=WorkspaceMemberRole.ADMIN,
|
||||
)
|
||||
|
||||
|
||||
mock_member_repo.find_by_workspace_and_user.return_value = admin_member
|
||||
|
||||
|
||||
request = RemoveMemberRequest(
|
||||
workspace_id="workspace-123",
|
||||
requester_user_id="admin-id",
|
||||
target_user_id="admin-id",
|
||||
)
|
||||
|
||||
|
||||
success, error = use_case.execute(request)
|
||||
|
||||
|
||||
assert success is False
|
||||
assert "Cannot remove yourself" in error
|
||||
|
||||
|
||||
def test_remove_member_no_permission(
|
||||
self,
|
||||
use_case,
|
||||
@@ -235,51 +246,51 @@ class TestRemoveMemberUseCase:
|
||||
):
|
||||
"""测试普通成员没有权限移除"""
|
||||
mock_workspace_repo.find_by_id.return_value = test_workspace
|
||||
|
||||
|
||||
regular_member = WorkspaceMember(
|
||||
id="member-1",
|
||||
workspace_id="workspace-123",
|
||||
user_id="user-id",
|
||||
role=WorkspaceMemberRole.MEMBER,
|
||||
)
|
||||
|
||||
|
||||
mock_member_repo.find_by_workspace_and_user.return_value = regular_member
|
||||
|
||||
|
||||
request = RemoveMemberRequest(
|
||||
workspace_id="workspace-123",
|
||||
requester_user_id="user-id",
|
||||
target_user_id="target-id",
|
||||
)
|
||||
|
||||
|
||||
success, error = use_case.execute(request)
|
||||
|
||||
|
||||
assert success is False
|
||||
assert error == "Only owners and admins can remove members"
|
||||
|
||||
|
||||
class TestLeaveWorkspaceUseCase:
|
||||
"""离开 Workspace 测试"""
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def mock_workspace_repo(self):
|
||||
repo = Mock()
|
||||
repo.find_by_id = Mock(return_value=None)
|
||||
return repo
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def mock_member_repo(self):
|
||||
repo = Mock()
|
||||
repo.find_by_workspace_and_user = Mock(return_value=None)
|
||||
repo.delete = Mock(return_value=True)
|
||||
return repo
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def use_case(self, mock_workspace_repo, mock_member_repo):
|
||||
return LeaveWorkspaceUseCase(
|
||||
workspace_repository=mock_workspace_repo,
|
||||
workspace_member_repository=mock_member_repo,
|
||||
)
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def test_workspace(self):
|
||||
return Workspace(
|
||||
@@ -287,7 +298,7 @@ class TestLeaveWorkspaceUseCase:
|
||||
name="Test Workspace",
|
||||
owner_user_id="owner-id",
|
||||
)
|
||||
|
||||
|
||||
def test_leave_workspace_success(
|
||||
self,
|
||||
use_case,
|
||||
@@ -297,7 +308,7 @@ class TestLeaveWorkspaceUseCase:
|
||||
):
|
||||
"""测试离开工作空间成功"""
|
||||
mock_workspace_repo.find_by_id.return_value = test_workspace
|
||||
|
||||
|
||||
member = WorkspaceMember(
|
||||
id="member-1",
|
||||
workspace_id="workspace-123",
|
||||
@@ -305,18 +316,18 @@ class TestLeaveWorkspaceUseCase:
|
||||
role=WorkspaceMemberRole.MEMBER,
|
||||
)
|
||||
mock_member_repo.find_by_workspace_and_user.return_value = member
|
||||
|
||||
|
||||
request = LeaveWorkspaceRequest(
|
||||
workspace_id="workspace-123",
|
||||
user_id="user-id",
|
||||
)
|
||||
|
||||
|
||||
success, error = use_case.execute(request)
|
||||
|
||||
|
||||
assert success is True
|
||||
assert error is None
|
||||
mock_member_repo.delete.assert_called_once_with("member-1")
|
||||
|
||||
|
||||
def test_leave_workspace_owner_cannot_leave(
|
||||
self,
|
||||
use_case,
|
||||
@@ -326,7 +337,7 @@ class TestLeaveWorkspaceUseCase:
|
||||
):
|
||||
"""测试 Owner 不能离开"""
|
||||
mock_workspace_repo.find_by_id.return_value = test_workspace
|
||||
|
||||
|
||||
owner_member = WorkspaceMember(
|
||||
id="member-1",
|
||||
workspace_id="workspace-123",
|
||||
@@ -334,17 +345,17 @@ class TestLeaveWorkspaceUseCase:
|
||||
role=WorkspaceMemberRole.OWNER,
|
||||
)
|
||||
mock_member_repo.find_by_workspace_and_user.return_value = owner_member
|
||||
|
||||
|
||||
request = LeaveWorkspaceRequest(
|
||||
workspace_id="workspace-123",
|
||||
user_id="owner-id",
|
||||
)
|
||||
|
||||
|
||||
success, error = use_case.execute(request)
|
||||
|
||||
|
||||
assert success is False
|
||||
assert "Owner cannot leave workspace" in error
|
||||
|
||||
|
||||
def test_leave_workspace_not_member(
|
||||
self,
|
||||
use_case,
|
||||
@@ -355,13 +366,13 @@ class TestLeaveWorkspaceUseCase:
|
||||
"""测试不是成员"""
|
||||
mock_workspace_repo.find_by_id.return_value = test_workspace
|
||||
mock_member_repo.find_by_workspace_and_user.return_value = None
|
||||
|
||||
|
||||
request = LeaveWorkspaceRequest(
|
||||
workspace_id="workspace-123",
|
||||
user_id="user-id",
|
||||
)
|
||||
|
||||
|
||||
success, error = use_case.execute(request)
|
||||
|
||||
|
||||
assert success is False
|
||||
assert error == "You are not a member of this workspace"
|
||||
|
||||
Reference in New Issue
Block a user