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

tests/test_ch10_message_bus_non_json_payload.py

@@ -0,0 +1,50 @@
+import sys
+import os
+
+sys.path.insert(0, os.path.abspath("chapter10/parallel-web-research"))
+
+from message_bus import Envelope, MessageBus
+
+
+def test_envelope_short_handles_non_json_payload():
+    """Contract: Envelope.short does not raise TypeError when payload contains non-JSON serializable objects."""
+    class CustomObject:
+        def __str__(self):
+            return "<CustomObject>"
+
+    payload = {
+        "tags": {"python", "asyncio"},
+        "object": CustomObject(),
+        "bytes": b"raw_data",
+    }
+
+    env = Envelope(
+        sender_id="agent_1",
+        target="agent_2",
+        type="data_sync",
+        payload=payload,
+    )
+
+    short_str = env.short()
+    assert isinstance(short_str, str)
+    assert "agent_1" in short_str
+    assert "data_sync" in short_str
+    assert "agent_2" in short_str
+
+
+def test_message_bus_publish_verbose_non_json_payload(capsys):
+    """Contract: MessageBus.publish in verbose mode logs without crashing on non-JSON payload."""
+    bus = MessageBus(verbose=True)
+    env = Envelope(
+        sender_id="sender",
+        target="*",
+        type="broadcast_event",
+        payload={"set_val": {1, 2, 3}},
+    )
+
+    import asyncio
+    asyncio.run(bus.publish(env))
+
+    captured = capsys.readouterr()
+    assert "BUS" in captured.out
+    assert "broadcast_event" in captured.out