qzl/social-app
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: 重构 Agent 模块为 AgentScope，删除旧版 CrewAI/LiteLLM 实现

Browse Source
This commit is contained in:
qzl
2026-03-11 20:51:56 +08:00
parent 177ed616bf
commit 145e3dc615
149 changed files with 5120 additions and 11356 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/tests/unit/v1/schedule_items/test_service.py
+68 -8
View File
@@ -97,6 +97,35 @@ class FakeRepo:
del data
return MagicMock()
async def list_subscribed_items_by_date_range(
self,
subscriber_id: UUID,
start_at: datetime,
end_at: datetime,
):
del subscriber_id, start_at, end_at
return []
async def get_user_subscriptions(self, subscriber_id: UUID):
del subscriber_id
return []
async def get_subscriptions_by_item_id(self, item_id: UUID):
del item_id
return []
async def get_subscription(self, item_id: UUID, subscriber_id: UUID):
del item_id, subscriber_id
return None
async def update_subscription_status(
self, item_id: UUID, subscriber_id: UUID, status
):
del item_id, subscriber_id, status
async def delete_subscriptions_by_item_id(self, item_id: UUID):
del item_id
@pytest.fixture
def mock_session() -> AsyncMock:
@@ -106,8 +135,15 @@ def mock_session() -> AsyncMock:
return session
@pytest.fixture
def mock_inbox_repository() -> MagicMock:
return MagicMock()
@pytest.mark.asyncio
async def test_create_success(mock_session: AsyncMock) -> None:
async def test_create_success(
mock_session: AsyncMock, mock_inbox_repository: MagicMock
) -> None:
user_id = UUID("00000000-0000-0000-0000-000000000001")
request = ScheduleItemCreateRequest(
title="Test Event",
@@ -117,6 +153,7 @@ async def test_create_success(mock_session: AsyncMock) -> None:
repository=FakeRepo(None),
session=mock_session,
current_user=CurrentUser(id=user_id),
inbox_repository=mock_inbox_repository,
)
result = await service.create(request)
@@ -126,7 +163,9 @@ async def test_create_success(mock_session: AsyncMock) -> None:
@pytest.mark.asyncio
async def test_create_invalid_end_at(mock_session: AsyncMock) -> None:
async def test_create_invalid_end_at(
mock_session: AsyncMock, mock_inbox_repository: MagicMock
) -> None:
user_id = UUID("00000000-0000-0000-0000-000000000001")
request = ScheduleItemCreateRequest(
title="Test Event",
@@ -137,6 +176,7 @@ async def test_create_invalid_end_at(mock_session: AsyncMock) -> None:
repository=FakeRepo(None),
session=mock_session,
current_user=CurrentUser(id=user_id),
inbox_repository=mock_inbox_repository,
)
with pytest.raises(HTTPException) as exc_info:
@@ -146,13 +186,16 @@ async def test_create_invalid_end_at(mock_session: AsyncMock) -> None:
@pytest.mark.asyncio
async def test_get_by_id_success(mock_session: AsyncMock) -> None:
async def test_get_by_id_success(
mock_session: AsyncMock, mock_inbox_repository: MagicMock
) -> None:
user_id = UUID("00000000-0000-0000-0000-000000000001")
item = _create_mock_schedule_item()
service = ScheduleItemService(
repository=FakeRepo(item),
session=mock_session,
current_user=CurrentUser(id=user_id),
inbox_repository=mock_inbox_repository,
)
result = await service.get_by_id(item.id)
@@ -161,12 +204,15 @@ async def test_get_by_id_success(mock_session: AsyncMock) -> None:
@pytest.mark.asyncio
async def test_get_by_id_not_found(mock_session: AsyncMock) -> None:
async def test_get_by_id_not_found(
mock_session: AsyncMock, mock_inbox_repository: MagicMock
) -> None:
user_id = UUID("00000000-0000-0000-0000-000000000001")
service = ScheduleItemService(
repository=FakeRepo(None),
session=mock_session,
current_user=CurrentUser(id=user_id),
inbox_repository=mock_inbox_repository,
)
with pytest.raises(HTTPException) as exc_info:
@@ -176,13 +222,16 @@ async def test_get_by_id_not_found(mock_session: AsyncMock) -> None:
@pytest.mark.asyncio
async def test_update_success(mock_session: AsyncMock) -> None:
async def test_update_success(
mock_session: AsyncMock, mock_inbox_repository: MagicMock
) -> None:
user_id = UUID("00000000-0000-0000-0000-000000000001")
item = _create_mock_schedule_item()
service = ScheduleItemService(
repository=FakeRepo(item),
session=mock_session,
current_user=CurrentUser(id=user_id),
inbox_repository=mock_inbox_repository,
)
result = await service.update(item.id, ScheduleItemUpdateRequest(title="Updated"))
@@ -191,13 +240,16 @@ async def test_update_success(mock_session: AsyncMock) -> None:
@pytest.mark.asyncio
async def test_delete_success(mock_session: AsyncMock) -> None:
async def test_delete_success(
mock_session: AsyncMock, mock_inbox_repository: MagicMock
) -> None:
user_id = UUID("00000000-0000-0000-0000-000000000001")
item = _create_mock_schedule_item()
service = ScheduleItemService(
repository=FakeRepo(item),
session=mock_session,
current_user=CurrentUser(id=user_id),
inbox_repository=mock_inbox_repository,
)
await service.delete(item.id)
@@ -206,7 +258,9 @@ async def test_delete_success(mock_session: AsyncMock) -> None:
@pytest.mark.asyncio
async def test_create_maps_metadata_to_extra_metadata(mock_session: AsyncMock) -> None:
async def test_create_maps_metadata_to_extra_metadata(
mock_session: AsyncMock, mock_inbox_repository: MagicMock
) -> None:
user_id = UUID("00000000-0000-0000-0000-000000000001")
captured: dict | None = None
@@ -232,6 +286,7 @@ async def test_create_maps_metadata_to_extra_metadata(mock_session: AsyncMock) -
repository=CaptureRepo(None),
session=mock_session,
current_user=CurrentUser(id=user_id),
inbox_repository=mock_inbox_repository,
)
await service.create(request)
@@ -244,7 +299,9 @@ async def test_create_maps_metadata_to_extra_metadata(mock_session: AsyncMock) -
@pytest.mark.asyncio
async def test_update_maps_metadata_to_extra_metadata(mock_session: AsyncMock) -> None:
async def test_update_maps_metadata_to_extra_metadata(
mock_session: AsyncMock, mock_inbox_repository: MagicMock
) -> None:
user_id = UUID("00000000-0000-0000-0000-000000000001")
item = _create_mock_schedule_item()
captured: dict | None = None
@@ -261,6 +318,7 @@ async def test_update_maps_metadata_to_extra_metadata(mock_session: AsyncMock) -
repository=CaptureRepo(item),
session=mock_session,
current_user=CurrentUser(id=user_id),
inbox_repository=mock_inbox_repository,
)
await service.update(
@@ -285,6 +343,7 @@ async def test_update_maps_metadata_to_extra_metadata(mock_session: AsyncMock) -
@pytest.mark.asyncio
async def test_update_maps_null_metadata_to_extra_metadata_null(
mock_session: AsyncMock,
mock_inbox_repository: MagicMock,
) -> None:
user_id = UUID("00000000-0000-0000-0000-000000000001")
item = _create_mock_schedule_item()
@@ -302,6 +361,7 @@ async def test_update_maps_null_metadata_to_extra_metadata_null(
repository=CaptureRepo(item),
session=mock_session,
current_user=CurrentUser(id=user_id),
inbox_repository=mock_inbox_repository,
)
await service.update(
backend/tests/unit/v1/schedule_items/test_share.py
+50 -1
View File
@@ -63,6 +63,12 @@ class ShareRepo:
return self._item
return None
async def get_subscription(self, item_id: UUID, subscriber_id: UUID) -> None:
return None
async def create_subscription(self, data: dict[str, object]) -> None:
return None
class AuthGatewayStub:
async def get_user_by_email(self, email: str) -> UserByEmailResponse:
@@ -74,6 +80,44 @@ class AuthGatewayStub:
)
class InboxRepoStub:
async def create(self, data: dict[str, object]) -> InboxMessage:
return InboxMessage(
id=uuid4(),
recipient_id=UUID("00000000-0000-0000-0000-000000000222"),
sender_id=UUID("00000000-0000-0000-0000-000000000001"),
message_type=InboxMessageType.CALENDAR,
schedule_item_id=uuid4(),
content='{"type": "invite", "permission": 1, "action": "pending"}',
created_by=UUID("00000000-0000-0000-0000-000000000001"),
)
async def get_by_id(
self, message_id: UUID, recipient_id: UUID
) -> InboxMessage | None:
return None
async def list_by_recipient(
self, recipient_id: UUID, is_read: bool | None = None
) -> list[InboxMessage]:
return []
async def mark_as_read(
self, message_id: UUID, recipient_id: UUID
) -> InboxMessage | None:
return None
async def get_pending_calendar_invite(
self, schedule_item_id: UUID, recipient_id: UUID
) -> InboxMessage | None:
return None
async def get_calendar_invite(
self, schedule_item_id: UUID, recipient_id: UUID
) -> InboxMessage | None:
return None
class AuthGatewayInvalidIdStub:
async def get_user_by_email(self, email: str) -> UserByEmailResponse:
return UserByEmailResponse(
@@ -97,6 +141,7 @@ async def test_share_forbidden_when_not_owner() -> None:
session=AsyncMock(),
current_user=CurrentUser(id=requester_id),
auth_gateway=AuthGatewayStub(),
inbox_repository=InboxRepoStub(),
)
with pytest.raises(HTTPException) as exc_info:
@@ -127,6 +172,7 @@ async def test_share_success_creates_calendar_invitation_message() -> None:
session=session,
current_user=CurrentUser(id=owner_id),
auth_gateway=AuthGatewayStub(),
inbox_repository=InboxRepoStub(),
)
result = await service.share(
@@ -146,7 +192,7 @@ async def test_share_success_creates_calendar_invitation_message() -> None:
assert message.sender_id == owner_id
assert message.schedule_item_id == item_id
assert message.message_type == InboxMessageType.CALENDAR
assert message.content == '{"permission": 5}'
assert message.content == '{"type": "invite", "permission": 5, "action": "pending"}'
session.commit.assert_awaited_once()
@@ -158,6 +204,7 @@ async def test_share_returns_not_found_when_item_missing() -> None:
session=AsyncMock(),
current_user=CurrentUser(id=requester_id),
auth_gateway=AuthGatewayStub(),
inbox_repository=InboxRepoStub(),
)
with pytest.raises(HTTPException) as exc_info:
@@ -187,6 +234,7 @@ async def test_share_invalid_auth_user_id_returns_503() -> None:
session=session,
current_user=CurrentUser(id=owner_id),
auth_gateway=AuthGatewayInvalidIdStub(),
inbox_repository=InboxRepoStub(),
)
with pytest.raises(HTTPException) as exc_info:
@@ -219,6 +267,7 @@ async def test_share_sqlalchemy_error_rolls_back() -> None:
session=session,
current_user=CurrentUser(id=owner_id),
auth_gateway=AuthGatewayStub(),
inbox_repository=InboxRepoStub(),
)
with pytest.raises(HTTPException) as exc_info:
backend/tests/unit/v1/schedule_items/test_subscription.py
+220
View File
@@ -0,0 +1,220 @@
import json
from datetime import datetime, timezone
from unittest.mock import AsyncMock, MagicMock
from uuid import UUID, uuid4
import pytest
from fastapi import HTTPException
from core.auth.models import CurrentUser
from models.inbox_messages import InboxMessage, InboxMessageStatus
from models.schedule_items import (
ScheduleItem,
ScheduleItemSourceType,
ScheduleItemStatus,
)
from models.schedule_subscriptions import ScheduleSubscription
from v1.schedule_items.service import ScheduleItemService
def _create_mock_schedule_item(
item_id: UUID = uuid4(),
owner_id: UUID = UUID("00000000-0000-0000-0000-000000000001"),
title: str = "Test Event",
) -> ScheduleItem:
item = MagicMock(spec=ScheduleItem)
item.id = item_id
item.owner_id = owner_id
item.title = title
item.description = None
item.start_at = datetime(2026, 2, 28, 16, 0, 0, tzinfo=timezone.utc)
item.end_at = datetime(2026, 2, 28, 17, 0, 0, tzinfo=timezone.utc)
item.timezone = "UTC"
item.extra_metadata = {}
item.source_type = ScheduleItemSourceType.MANUAL
item.status = ScheduleItemStatus.ACTIVE
item.created_at = datetime(2026, 2, 27, 10, 0, 0, tzinfo=timezone.utc)
item.updated_at = datetime(2026, 2, 27, 10, 0, 0, tzinfo=timezone.utc)
item.deleted_at = None
return item
class FakeInboxRepo:
def __init__(self, inbox_message: InboxMessage | None = None) -> None:
self._inbox = inbox_message
async def get_pending_calendar_invite(
self, schedule_item_id: UUID, recipient_id: UUID
) -> InboxMessage | None:
if self._inbox:
return self._inbox
return None
async def create(self, data: dict) -> InboxMessage:
return MagicMock()
async def get_by_id(
self, message_id: UUID, recipient_id: UUID
) -> InboxMessage | None:
return None
async def list_by_recipient(
self, recipient_id: UUID, is_read: bool | None = None
) -> list[InboxMessage]:
return []
async def mark_as_read(
self, message_id: UUID, recipient_id: UUID
) -> InboxMessage | None:
return None
@pytest.fixture
def mock_session() -> AsyncMock:
session = AsyncMock()
session.commit = AsyncMock()
session.rollback = AsyncMock()
return session
@pytest.fixture
def mock_repo() -> MagicMock:
repo = MagicMock()
repo.create_subscription = AsyncMock(return_value=MagicMock())
return repo
@pytest.mark.asyncio
async def test_accept_subscription_success(
mock_session: AsyncMock, mock_repo: MagicMock
) -> None:
user_id = UUID("00000000-0000-0000-0000-000000000001")
sender_id = UUID("00000000-0000-0000-0000-000000000002")
item_id = uuid4()
inbox_message = MagicMock(spec=InboxMessage)
inbox_message.id = uuid4()
inbox_message.sender_id = sender_id
inbox_message.content = json.dumps({"type": "invite", "permission": 1})
inbox_message.status = InboxMessageStatus.PENDING
service = ScheduleItemService(
repository=mock_repo,
session=mock_session,
current_user=CurrentUser(id=user_id),
inbox_repository=FakeInboxRepo(inbox_message),
)
result = await service.accept_subscription(item_id)
assert result == {"message": "Subscription accepted"}
mock_session.commit.assert_awaited_once()
@pytest.mark.asyncio
async def test_accept_subscription_not_found(
mock_session: AsyncMock, mock_repo: MagicMock
) -> None:
user_id = UUID("00000000-0000-0000-0000-000000000001")
item_id = uuid4()
service = ScheduleItemService(
repository=mock_repo,
session=mock_session,
current_user=CurrentUser(id=user_id),
inbox_repository=FakeInboxRepo(None),
)
with pytest.raises(HTTPException) as exc_info:
await service.accept_subscription(item_id)
assert exc_info.value.status_code == 404
assert "No pending invitation found" in exc_info.value.detail
@pytest.mark.asyncio
async def test_reject_subscription_success(
mock_session: AsyncMock, mock_repo: MagicMock
) -> None:
user_id = UUID("00000000-0000-0000-0000-000000000001")
item_id = uuid4()
inbox_message = MagicMock(spec=InboxMessage)
inbox_message.id = uuid4()
inbox_message.status = InboxMessageStatus.PENDING
service = ScheduleItemService(
repository=mock_repo,
session=mock_session,
current_user=CurrentUser(id=user_id),
inbox_repository=FakeInboxRepo(inbox_message),
)
result = await service.reject_subscription(item_id)
assert result == {"message": "Subscription rejected"}
mock_session.commit.assert_awaited_once()
@pytest.mark.asyncio
async def test_reject_subscription_not_found(
mock_session: AsyncMock, mock_repo: MagicMock
) -> None:
user_id = UUID("00000000-0000-0000-0000-000000000001")
item_id = uuid4()
service = ScheduleItemService(
repository=mock_repo,
session=mock_session,
current_user=CurrentUser(id=user_id),
inbox_repository=FakeInboxRepo(None),
)
with pytest.raises(HTTPException) as exc_info:
await service.reject_subscription(item_id)
assert exc_info.value.status_code == 404
assert "No pending invitation found" in exc_info.value.detail
@pytest.mark.asyncio
async def test_list_by_date_range_with_subscriptions(
mock_session: AsyncMock, mock_repo: MagicMock
) -> None:
user_id = UUID("00000000-0000-0000-0000-000000000001")
owner_id = UUID("00000000-0000-0000-0000-000000000002")
item_id = uuid4()
owned_item = _create_mock_schedule_item(item_id=item_id, owner_id=user_id)
subscribed_item = _create_mock_schedule_item(
item_id=uuid4(), owner_id=owner_id, title="Subscribed Event"
)
subscription = MagicMock(spec=ScheduleSubscription)
subscription.item_id = subscribed_item.id
subscription.permission = 1
subscription.subscriber_id = user_id
mock_repo.list_by_date_range = AsyncMock(return_value=[owned_item])
mock_repo.get_user_subscriptions = AsyncMock(return_value=[subscription])
mock_repo.get_by_id = AsyncMock(return_value=subscribed_item)
service = ScheduleItemService(
repository=mock_repo,
session=mock_session,
current_user=CurrentUser(id=user_id),
inbox_repository=FakeInboxRepo(),
)
from v1.schedule_items.schemas import ScheduleItemListRequest
request = ScheduleItemListRequest(
start_at=datetime(2026, 2, 1, 0, 0, 0, tzinfo=timezone.utc),
end_at=datetime(2026, 3, 1, 0, 0, 0, tzinfo=timezone.utc),
)
result = await service.list_by_date_range(request)
assert len(result) == 2
assert result[0].is_owner is True
assert result[1].is_owner is False
assert result[1].permission == 1