fix: 重试临时外部 API 错误

2026-07-06 16:09:08 +08:00
parent 918dd62e21
commit 44d459fc50
2 changed files with 81 additions and 5 deletions
@@ -53,6 +53,61 @@ def test_tikhub_client_raises_structured_error_after_retries(monkeypatch):
    assert sleeps == [1, 2, 4]


+def test_tikhub_client_retries_transient_api_error(monkeypatch):
+    sleeps = []
+    monkeypatch.setattr("app.platforms.base.time.sleep", sleeps.append)
+    transport = SequenceTransport(
+        [
+            httpx.Response(400, json={"message_zh": "临时请求失败"}),
+            httpx.Response(200, json={"ok": True}),
+        ]
+    )
+    http_client = httpx.Client(transport=httpx.MockTransport(transport))
+    client = TikHubClient(base_url="https://api.test", api_key="secret-token", http_client=http_client)
+
+    result = client.get("/demo")
+
+    assert result == {"ok": True}
+    assert sleeps == [1]
+    assert len(transport.requests) == 2
+
+
+def test_tikhub_client_includes_response_summary_after_api_error_retries(monkeypatch):
+    sleeps = []
+    monkeypatch.setattr("app.platforms.base.time.sleep", sleeps.append)
+    transport = SequenceTransport(
+        [httpx.Response(400, json={"message_zh": "笔记评论暂不可用"}) for _ in range(4)]
+    )
+    http_client = httpx.Client(transport=httpx.MockTransport(transport))
+    client = TikHubClient(base_url="https://api.test", api_key="secret-token", http_client=http_client)
+
+    with pytest.raises(PlatformAPIError) as exc_info:
+        client.get("/demo")
+
+    assert exc_info.value.error_type == "api_error"
+    assert exc_info.value.status_code == 400
+    assert "External API returned HTTP 400" in str(exc_info.value)
+    assert "笔记评论暂不可用" in str(exc_info.value)
+    assert "secret-token" not in str(exc_info.value)
+    assert sleeps == [1, 2, 4]
+
+
+def test_tikhub_client_includes_text_response_summary_after_api_error_retries(monkeypatch):
+    sleeps = []
+    monkeypatch.setattr("app.platforms.base.time.sleep", sleeps.append)
+    transport = SequenceTransport([httpx.Response(502, text="upstream temporary failure") for _ in range(4)])
+    http_client = httpx.Client(transport=httpx.MockTransport(transport))
+    client = TikHubClient(base_url="https://api.test", api_key="secret-token", http_client=http_client)
+
+    with pytest.raises(PlatformAPIError) as exc_info:
+        client.get("/demo")
+
+    assert exc_info.value.error_type == "api_error"
+    assert exc_info.value.status_code == 502
+    assert str(exc_info.value) == "External API returned HTTP 502: upstream temporary failure"
+    assert sleeps == [1, 2, 4]
+
+
 def test_tikhub_client_reports_401_as_auth_error_without_leaking_token():
    transport = SequenceTransport([httpx.Response(401, json={"message": "Unauthorized"})])
    http_client = httpx.Client(transport=httpx.MockTransport(transport))