feat: 日历分享改为按手机号+好友关系校验
This commit is contained in:
@@ -261,21 +261,12 @@ async def test_calendar_share_executes_with_valid_invitee(
|
||||
monkeypatch.setattr(
|
||||
calendar_module, "create_schedule_service", lambda *_: fake_service
|
||||
)
|
||||
target_user_id = str(uuid4())
|
||||
monkeypatch.setattr(
|
||||
calendar_module,
|
||||
"resolve_share_target_phone_map",
|
||||
lambda user_ids: {target_user_id: "+8613900001234"}
|
||||
if target_user_id in user_ids
|
||||
else {},
|
||||
)
|
||||
|
||||
event_id = str(uuid4())
|
||||
result = await calendar_module.calendar_share(
|
||||
event_id=event_id,
|
||||
invitees=[
|
||||
calendar_module.CalendarShareInvitee(
|
||||
userId=target_user_id,
|
||||
phone="13900001234",
|
||||
permissionView=True,
|
||||
permissionEdit=False,
|
||||
permissionInvite=False,
|
||||
@@ -287,9 +278,37 @@ async def test_calendar_share_executes_with_valid_invitee(
|
||||
payload = _decode_tool_response(result)
|
||||
|
||||
assert payload["status"] == "success"
|
||||
assert payload["result"].startswith("status=success invited_count=1")
|
||||
assert payload["result"].startswith("status=success success=1 failed=0")
|
||||
assert "+8613900001234" in payload["result"]
|
||||
assert len(fake_service.share_calls) == 1
|
||||
share_call = fake_service.share_calls[0]
|
||||
assert share_call["item_id"] == event_id
|
||||
assert share_call["request"].phone == "+8613900001234"
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_calendar_share_rejects_invalid_phone(
|
||||
monkeypatch: pytest.MonkeyPatch,
|
||||
) -> None:
|
||||
fake_service = _FakeService()
|
||||
monkeypatch.setattr(
|
||||
calendar_module, "create_schedule_service", lambda *_: fake_service
|
||||
)
|
||||
|
||||
result = await calendar_module.calendar_share(
|
||||
event_id=str(uuid4()),
|
||||
invitees=[
|
||||
calendar_module.CalendarShareInvitee(
|
||||
phone="12345",
|
||||
permissionView=True,
|
||||
permissionEdit=False,
|
||||
permissionInvite=False,
|
||||
)
|
||||
],
|
||||
session=SimpleNamespace(),
|
||||
owner_id=uuid4(),
|
||||
)
|
||||
payload = _decode_tool_response(result)
|
||||
|
||||
assert payload["status"] == "failure"
|
||||
assert payload["error"]["code"] == "INVALID_ARGUMENT"
|
||||
|
||||
@@ -0,0 +1,63 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
from types import SimpleNamespace
|
||||
from typing import Any
|
||||
from uuid import uuid4
|
||||
|
||||
import pytest
|
||||
from agentscope.tool import ToolResponse
|
||||
from core.agentscope.tools.custom.user_lookup import user_lookup
|
||||
import core.agentscope.tools.custom.user_lookup as user_lookup_module
|
||||
|
||||
|
||||
def _decode_tool_response(response: ToolResponse) -> dict[str, Any]:
|
||||
assert response.content
|
||||
first = response.content[0]
|
||||
if isinstance(first, dict):
|
||||
text = str(first.get("text", ""))
|
||||
else:
|
||||
text = str(getattr(first, "text", ""))
|
||||
return json.loads(text)
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_user_lookup_requires_runtime_context() -> None:
|
||||
result = await user_lookup()
|
||||
payload = _decode_tool_response(result)
|
||||
|
||||
assert payload["status"] == "failure"
|
||||
assert payload["error"]["code"] == "MISSING_RUNTIME_ARGS"
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_user_lookup_returns_friend_contacts(
|
||||
monkeypatch: pytest.MonkeyPatch,
|
||||
) -> None:
|
||||
async def _fake_list_friend_contacts(**_: Any) -> list[dict[str, str]]:
|
||||
return [
|
||||
{
|
||||
"userId": "00000000-0000-0000-0000-000000000101",
|
||||
"username": "alice",
|
||||
"phone": "+8613900000001",
|
||||
},
|
||||
{
|
||||
"userId": "00000000-0000-0000-0000-000000000102",
|
||||
"username": "bob",
|
||||
"phone": "+8613900000002",
|
||||
},
|
||||
]
|
||||
|
||||
monkeypatch.setattr(
|
||||
user_lookup_module,
|
||||
"_list_friend_contacts",
|
||||
_fake_list_friend_contacts,
|
||||
)
|
||||
|
||||
result = await user_lookup(session=SimpleNamespace(), owner_id=uuid4())
|
||||
payload = _decode_tool_response(result)
|
||||
|
||||
assert payload["status"] == "success"
|
||||
assert "friends_count=2" in payload["result"]
|
||||
assert "username=alice" in payload["result"]
|
||||
assert "+8613900000001" in payload["result"]
|
||||
Reference in New Issue
Block a user