refactor: 重构 Agent 模块为 AgentScope,删除旧版 CrewAI/LiteLLM 实现
This commit is contained in:
@@ -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(
|
||||
|
||||
Reference in New Issue
Block a user