ai-agent-book 精选快照（<2MB 代码与文档，来自 github.com/bojieli/ai-agent-book）

chapter4/perception-tools/test_pubchem_async.py

@@ -0,0 +1,52 @@
+"""Deterministic regressions for PubChem asynchronous ListKey handling."""
+
+import sys
+from pathlib import Path
+
+sys.path.insert(0, str(Path(__file__).parent / "src"))
+
+from pubchem_tools import PubChemClient
+
+
+class _Response:
+    def __init__(self, status_code, payload):
+        self.status_code = status_code
+        self._payload = payload
+        self.text = str(payload)
+
+    def json(self):
+        return self._payload
+
+
+def test_listkey_job_fault_resubmits_original_query(monkeypatch):
+    client = PubChemClient()
+    original = (
+        "https://pubchem.ncbi.nlm.nih.gov/rest/pug/compound/formula/C9H8O4/"
+        "property/Title,MolecularFormula/JSON"
+    )
+    calls = []
+    responses = iter([
+        _Response(202, {"Waiting": {"ListKey": "first"}}),
+        _Response(500, {"Fault": {"Code": "PUGREST.Unknown"}}),
+        _Response(202, {"Waiting": {"ListKey": "second"}}),
+        _Response(200, {"PropertyTable": {"Properties": [{"CID": 2244}]}}),
+    ])
+
+    def fake_get(url, params=None, timeout=None):
+        calls.append(url)
+        return next(responses)
+
+    monkeypatch.setattr(client.session, "get", fake_get)
+    monkeypatch.setattr(client, "_rate_limit", lambda: 0.0)
+    monkeypatch.setattr("pubchem_tools.time.sleep", lambda _seconds: None)
+
+    payload, latency = client.make_request(original)
+
+    assert payload["PropertyTable"]["Properties"][0]["CID"] == 2244
+    assert latency >= 0
+    assert calls == [
+        original,
+        original.replace("formula/C9H8O4", "listkey/first"),
+        original,
+        original.replace("formula/C9H8O4", "listkey/second"),
+    ]