mirror of
https://github.com/snowy-yang/qqbotsdk.git
synced 2026-10-21 13:04:49 +00:00
372 lines
12 KiB
Python
372 lines
12 KiB
Python
import pytest
|
||
from aiohttp import ClientSession, ClientTimeout, web
|
||
from aiohttp.test_utils import TestServer
|
||
|
||
from qqbotsdk.api import ApiError, BotApi, button, keyboard
|
||
from qqbotsdk.config import Config
|
||
from qqbotsdk.token import AccessToken
|
||
|
||
|
||
def make_api() -> tuple[BotApi, dict]:
|
||
"""替换掉 request 记录入参，只测业务方法拼参。"""
|
||
captured: dict = {}
|
||
|
||
class Recording(BotApi):
|
||
async def request(self, method, path, **kwargs): # type: ignore[override]
|
||
captured["call"] = (method, path, kwargs)
|
||
return {}
|
||
|
||
return Recording.__new__(Recording), captured
|
||
|
||
|
||
async def test_post_group_message_passive():
|
||
api, cap = make_api()
|
||
await api.post_group_message("G1", content="hi", msg_id="m1", msg_seq=2)
|
||
method, path, kwargs = cap["call"]
|
||
assert (method, path) == ("POST", "/v2/groups/G1/messages")
|
||
assert kwargs["json"] == {
|
||
"msg_type": 0,
|
||
"content": "hi",
|
||
"msg_id": "m1",
|
||
"msg_seq": 2,
|
||
}
|
||
|
||
|
||
async def test_post_c2c_message_active_no_msg_seq():
|
||
api, cap = make_api()
|
||
await api.post_c2c_message("U1", content="yo")
|
||
_, path, kwargs = cap["call"]
|
||
assert path == "/v2/users/U1/messages"
|
||
assert kwargs["json"] == {"msg_type": 0, "content": "yo"}
|
||
|
||
|
||
async def test_post_message_with_media_sets_msg_type():
|
||
api, cap = make_api()
|
||
media = {"file_info": "f1"}
|
||
await api.post_group_message("G1", content="看图", msg_id="m1", media=media)
|
||
body = cap["call"][2]["json"]
|
||
assert body["msg_type"] == 7
|
||
assert body["media"] == media
|
||
|
||
|
||
async def test_upload_group_file():
|
||
api, cap = make_api()
|
||
await api.upload_group_file("G1", file_type=1, url="http://x/img.png")
|
||
method, path, kwargs = cap["call"]
|
||
assert (method, path) == ("POST", "/v2/groups/G1/files")
|
||
assert kwargs["json"] == {
|
||
"file_type": 1,
|
||
"url": "http://x/img.png",
|
||
"srv_send_msg": False,
|
||
}
|
||
|
||
|
||
async def test_channel_message_and_reaction():
|
||
api, cap = make_api()
|
||
await api.post_channel_message("C1", content="hi", msg_id="m1", image="http://i")
|
||
method, path, kwargs = cap["call"]
|
||
assert (method, path) == ("POST", "/channels/C1/messages")
|
||
assert kwargs["json"] == {
|
||
"msg_seq": 1,
|
||
"content": "hi",
|
||
"msg_id": "m1",
|
||
"image": "http://i",
|
||
}
|
||
|
||
await api.put_channel_reaction("C1", "m1", emoji_type=1, emoji_id="4")
|
||
assert cap["call"][:2] == ("PUT", "/channels/C1/messages/m1/reactions/1/4")
|
||
|
||
await api.delete_channel_reaction("C1", "m1", emoji_type=1, emoji_id="4")
|
||
assert cap["call"][:2] == ("DELETE", "/channels/C1/messages/m1/reactions/1/4")
|
||
|
||
await api.withdraw_channel_message("C1", "m1", hide_tip=True)
|
||
method, path, kwargs = cap["call"]
|
||
assert (method, path) == ("DELETE", "/channels/C1/messages/m1")
|
||
assert kwargs["params"] == {"hidetip": "true"}
|
||
|
||
|
||
async def test_withdraw_v2_messages():
|
||
api, cap = make_api()
|
||
await api.withdraw_group_message("G1", "ROBOT1.0_x")
|
||
assert cap["call"][:2] == ("DELETE", "/v2/groups/G1/messages/ROBOT1.0_x")
|
||
await api.withdraw_c2c_message("U1", "ROBOT1.0_y")
|
||
assert cap["call"][:2] == ("DELETE", "/v2/users/U1/messages/ROBOT1.0_y")
|
||
|
||
|
||
async def test_event_id_passive_reply():
|
||
api, cap = make_api()
|
||
await api.post_group_message("G1", content="hi", event_id="e1")
|
||
body = cap["call"][2]["json"]
|
||
assert body["event_id"] == "e1"
|
||
assert "msg_id" not in body
|
||
assert body["msg_seq"] == 1
|
||
|
||
|
||
async def test_mute_and_schedule_official_field_names():
|
||
api, cap = make_api()
|
||
await api.mute_guild_member("GID", "UID", mute_seconds=3600)
|
||
assert cap["call"][2]["json"] == {"mute_seconds": "3600"}
|
||
|
||
await api.create_channel_schedule(
|
||
"C1", "活动", "1642076453000", "1642083653000", "C2"
|
||
)
|
||
method, path, kwargs = cap["call"]
|
||
assert (method, path) == ("POST", "/channels/C1/schedules")
|
||
info = kwargs["json"]["schedule"]
|
||
assert info["remind_type"] == "0"
|
||
assert "schedule_info" not in kwargs["json"]
|
||
|
||
await api.mute_group_member("G1", "2026-01-01T12:00:00+08:00")
|
||
method, path, kwargs = cap["call"]
|
||
assert (method, path) == ("POST", "/v2/groups/G1/restrict_chat_setting")
|
||
assert kwargs["json"] == {"mute_expire_at": "2026-01-01T12:00:00+08:00"}
|
||
|
||
|
||
async def test_me():
|
||
api, cap = make_api()
|
||
await api.me()
|
||
assert cap["call"][:2] == ("GET", "/users/@me")
|
||
|
||
|
||
async def test_post_message_markdown_and_keyboard():
|
||
api, cap = make_api()
|
||
kb = {"content": {"rows": []}}
|
||
await api.post_group_message("G1", markdown="# 标题", keyboard=kb, msg_id="m1")
|
||
method, path, kwargs = cap["call"]
|
||
assert (method, path) == ("POST", "/v2/groups/G1/messages")
|
||
body = kwargs["json"]
|
||
assert body["msg_type"] == 2
|
||
assert body["markdown"] == {"content": "# 标题"}
|
||
assert "content" not in body # markdown 与 content 互斥
|
||
assert body["keyboard"] == kb
|
||
assert body["msg_id"] == "m1"
|
||
|
||
|
||
async def test_post_message_markdown_dict_and_event_id():
|
||
api, cap = make_api()
|
||
await api.post_c2c_message(
|
||
"U1", markdown={"content": "# t"}, event_id="e1", msg_seq=2
|
||
)
|
||
body = cap["call"][2]["json"]
|
||
assert body["msg_type"] == 2
|
||
assert body["markdown"] == {"content": "# t"}
|
||
assert body["event_id"] == "e1"
|
||
assert body["msg_seq"] == 2
|
||
|
||
|
||
async def test_media_takes_precedence_over_markdown():
|
||
api, cap = make_api()
|
||
await api.post_group_message(
|
||
"G1", content="看图", msg_id="m1", media={"file_info": "f"}, markdown="# t"
|
||
)
|
||
body = cap["call"][2]["json"]
|
||
assert body["msg_type"] == 7
|
||
assert "markdown" not in body
|
||
|
||
|
||
async def test_respond_interaction():
|
||
api, cap = make_api()
|
||
await api.respond_interaction("I1")
|
||
method, path, kwargs = cap["call"]
|
||
assert (method, path) == ("PUT", "/interactions/I1")
|
||
assert kwargs["json"] == {"code": 0}
|
||
|
||
|
||
def test_button_defaults_and_overrides():
|
||
btn = button("点我", data="d1")
|
||
assert btn["render_data"] == {"label": "点我", "style": 1}
|
||
assert btn["action"] == {"type": 1, "permission": {"type": 2}, "data": "d1"}
|
||
assert len(btn["id"]) == 8 # 未指定 id 时自动生成短 id
|
||
|
||
btn2 = button("跳转", data="https://x", type=0, permission=1, style=0, id="b2")
|
||
assert btn2["id"] == "b2"
|
||
assert btn2["action"]["type"] == 0
|
||
assert btn2["action"]["permission"] == {"type": 1}
|
||
assert btn2["render_data"]["style"] == 0
|
||
|
||
|
||
def test_keyboard_rows():
|
||
b1, b2, b3 = button("a", data="1"), button("b", data="2"), button("c", data="3")
|
||
kb = keyboard(b1, [b2, b3])
|
||
rows = kb["content"]["rows"]
|
||
assert rows[0]["buttons"] == [b1] # 单按钮独占一行
|
||
assert rows[1]["buttons"] == [b2, b3] # 列表同行并排
|
||
assert len(rows) == 2
|
||
|
||
|
||
async def test_request_tolerates_non_json_content_type():
|
||
"""回应互动等接口 200 响应的 Content-Type 是 text/plain（body 为 JSON），
|
||
request 不应抛 ContentTypeError（真实 HTTP 栈回归）。"""
|
||
|
||
async def handler(request):
|
||
return web.Response(text="{}", content_type="text/plain")
|
||
|
||
app = web.Application()
|
||
app.router.add_route("PUT", "/interactions/I1", handler)
|
||
server = TestServer(app)
|
||
await server.start_server()
|
||
try:
|
||
config = Config(
|
||
app_id="id",
|
||
app_secret="s",
|
||
base_url=f"http://{server.host}:{server.port}",
|
||
timeout=ClientTimeout(total=5),
|
||
connecter="websocket",
|
||
intents_file="intents.toml",
|
||
webhook_host="0.0.0.0",
|
||
webhook_port=8080,
|
||
webhook_path="/qqbot/webhook",
|
||
)
|
||
|
||
class FixedToken(AccessToken):
|
||
async def get_access_token(self) -> str:
|
||
return "test-token"
|
||
|
||
http = ClientSession()
|
||
try:
|
||
api = BotApi(config, http, FixedToken(config, http))
|
||
assert await api.respond_interaction("I1") == {}
|
||
finally:
|
||
await http.close()
|
||
finally:
|
||
await server.close()
|
||
|
||
|
||
# ---------- 请求内核：429 限流重试 / 401 刷新 token 重试 ----------
|
||
|
||
|
||
class ScriptedResponse:
|
||
def __init__(self, status: int, payload, headers: dict | None = None) -> None:
|
||
self.status = status
|
||
self.headers = headers or {}
|
||
self._payload = payload
|
||
|
||
async def json(self, **kwargs):
|
||
return self._payload
|
||
|
||
async def __aenter__(self):
|
||
return self
|
||
|
||
async def __aexit__(self, *exc):
|
||
return False
|
||
|
||
|
||
class ScriptedSession:
|
||
"""按序弹出响应，记录 request 调用。"""
|
||
|
||
def __init__(self, *responses: ScriptedResponse) -> None:
|
||
self._responses = list(responses)
|
||
self.calls: list[tuple[str, str]] = []
|
||
|
||
def request(self, method: str, url: str, **kwargs):
|
||
self.calls.append((method, url))
|
||
return self._responses.pop(0)
|
||
|
||
|
||
class StubToken:
|
||
def __init__(self) -> None:
|
||
self.fetches = 0
|
||
self.invalidated = 0
|
||
|
||
async def get_access_token(self) -> str:
|
||
self.fetches += 1
|
||
return "t1"
|
||
|
||
def invalidate(self) -> None:
|
||
self.invalidated += 1
|
||
|
||
|
||
def make_core_api(http: ScriptedSession, token: StubToken | None = None) -> BotApi:
|
||
config = Config(
|
||
app_id="id",
|
||
app_secret="s",
|
||
base_url="https://x",
|
||
timeout=ClientTimeout(total=5),
|
||
connecter="websocket",
|
||
intents_file="intents.toml",
|
||
webhook_host="0.0.0.0",
|
||
webhook_port=8080,
|
||
webhook_path="/qqbot/webhook",
|
||
)
|
||
return BotApi(config, http, token or StubToken()) # type: ignore[arg-type]
|
||
|
||
|
||
async def test_request_retries_429_honoring_retry_after(monkeypatch):
|
||
delays: list[float] = []
|
||
|
||
async def fake_sleep(delay):
|
||
delays.append(delay)
|
||
|
||
monkeypatch.setattr("qqbotsdk.api.core.asyncio.sleep", fake_sleep)
|
||
http = ScriptedSession(
|
||
ScriptedResponse(
|
||
429, {"code": 11244, "message": "limit"}, {"Retry-After": "2"}
|
||
),
|
||
ScriptedResponse(200, {"ok": 1}),
|
||
)
|
||
api = make_core_api(http)
|
||
assert await api.request("POST", "/v1/x") == {"ok": 1}
|
||
assert len(http.calls) == 2
|
||
assert delays == [2.0]
|
||
|
||
|
||
async def test_request_429_dirty_header_falls_back_to_backoff(monkeypatch):
|
||
delays: list[float] = []
|
||
|
||
async def fake_sleep(delay):
|
||
delays.append(delay)
|
||
|
||
monkeypatch.setattr("qqbotsdk.api.core.asyncio.sleep", fake_sleep)
|
||
http = ScriptedSession(
|
||
# 线上见过的脏值 "30,120"：取首段 30
|
||
ScriptedResponse(429, {}, {"Retry-After": "30,120"}),
|
||
# 无 Retry-After：指数退避 1 * 2**1 = 2
|
||
ScriptedResponse(429, {}),
|
||
ScriptedResponse(200, {}),
|
||
)
|
||
api = make_core_api(http)
|
||
assert await api.get("/v1/x") == {}
|
||
assert delays == [30.0, 2.0]
|
||
|
||
|
||
async def test_request_429_retry_after_capped(monkeypatch):
|
||
delays: list[float] = []
|
||
|
||
async def fake_sleep(delay):
|
||
delays.append(delay)
|
||
|
||
monkeypatch.setattr("qqbotsdk.api.core.asyncio.sleep", fake_sleep)
|
||
http = ScriptedSession(
|
||
ScriptedResponse(429, {}, {"Retry-After": "3600"}),
|
||
ScriptedResponse(200, {}),
|
||
)
|
||
api = make_core_api(http)
|
||
await api.get("/v1/x")
|
||
assert delays == [30.0] # 封顶，不真睡一小时
|
||
|
||
|
||
async def test_request_429_exhausted_raises_api_error(monkeypatch):
|
||
async def fake_sleep(delay):
|
||
pass
|
||
|
||
monkeypatch.setattr("qqbotsdk.api.core.asyncio.sleep", fake_sleep)
|
||
http = ScriptedSession(*(ScriptedResponse(429, {}) for _ in range(4)))
|
||
api = make_core_api(http)
|
||
with pytest.raises(ApiError) as exc_info:
|
||
await api.get("/v1/x")
|
||
assert exc_info.value.status == 429
|
||
assert len(http.calls) == 4 # 首次 + 3 次重试
|
||
|
||
|
||
async def test_request_401_invalidates_token_and_retries_once():
|
||
http = ScriptedSession(
|
||
ScriptedResponse(401, {"code": -1, "message": "unauthorized"}),
|
||
ScriptedResponse(200, {"ok": 1}),
|
||
)
|
||
token = StubToken()
|
||
api = make_core_api(http, token)
|
||
assert await api.get("/v1/x") == {"ok": 1}
|
||
assert token.invalidated == 1
|
||
assert token.fetches == 2
|
||
assert len(http.calls) == 2
|