mirror of
https://github.com/hansjone/oclaw.git
synced 2026-10-09 04:40:45 +08:00
oliver
31e3962c74
This persists referenced media safely, adds explicit attachment ACL/backfill flows, and fixes the Weixin sidecar so official login state can bridge reliably into oclaw without duplicate replays or dropped replies. Made-with: Cursor
64 lines
2.7 KiB
Python
64 lines
2.7 KiB
Python
from __future__ import annotations
|
|
|
|
import base64
|
|
import os
|
|
|
|
from oclaw.runtime.chat.media_redact import ingest_embedded_image_blobs_as_refs, redact_embedded_image_blobs
|
|
|
|
|
|
def test_redact_nested_mcp_image_block() -> None:
|
|
big = "/9j/" + "a" * 800
|
|
obj = {
|
|
"ok": True,
|
|
"result": {"content": [{"type": "image", "mime": "image/jpeg", "data": big}]},
|
|
}
|
|
out = redact_embedded_image_blobs(obj)
|
|
assert out["result"]["content"][0].get("_image_payload_redacted") is True
|
|
assert "data" not in out["result"]["content"][0]
|
|
assert isinstance(out["result"]["content"][0].get("_redacted_payload_chars"), int)
|
|
|
|
|
|
def test_redact_keeps_small_data_field() -> None:
|
|
obj = {"type": "image", "mime": "image/png", "data": "abc"}
|
|
assert redact_embedded_image_blobs(obj) == obj
|
|
|
|
|
|
def test_ingest_embedded_image_blob_as_ref(tmp_path) -> None:
|
|
raw = base64.b64encode(b"png-bytes").decode("ascii")
|
|
obj = {"result": {"content": [{"type": "image", "mime": "image/png", "data": raw, "name": "x.png"}]}}
|
|
out, refs = ingest_embedded_image_blobs_as_refs(obj, root_dir=str(tmp_path), filename_prefix="unit")
|
|
block = out["result"]["content"][0]
|
|
assert block["type"] == "image_ref"
|
|
assert str(block.get("attachment_id") or "")
|
|
assert "data" not in block
|
|
assert refs and refs[0]["attachment_id"] == block["attachment_id"]
|
|
|
|
|
|
def test_ingest_embedded_binary_blob_as_binary_ref(tmp_path) -> None:
|
|
raw = base64.b64encode(b"%PDF-1.4-fake").decode("ascii")
|
|
obj = {"result": {"content": [{"type": "file", "mime": "application/pdf", "base64": raw, "name": "a.pdf"}]}}
|
|
out, refs = ingest_embedded_image_blobs_as_refs(obj, root_dir=str(tmp_path), filename_prefix="unit")
|
|
block = out["result"]["content"][0]
|
|
assert block["type"] == "binary_ref"
|
|
assert str(block.get("attachment_id") or "")
|
|
assert "base64" not in block
|
|
assert refs and refs[0]["attachment_id"] == block["attachment_id"]
|
|
|
|
|
|
def test_ingest_rejects_oversized_blob_by_env_limit(tmp_path) -> None:
|
|
prev = os.environ.get("AIA_MAX_ATTACHMENT_BYTES")
|
|
os.environ["AIA_MAX_ATTACHMENT_BYTES"] = "4"
|
|
try:
|
|
raw = base64.b64encode(b"12345").decode("ascii")
|
|
obj = {"result": {"content": [{"type": "file", "mime": "application/octet-stream", "base64": raw, "name": "x.bin"}]}}
|
|
out, refs = ingest_embedded_image_blobs_as_refs(obj, root_dir=str(tmp_path), filename_prefix="unit")
|
|
finally:
|
|
if prev is None:
|
|
os.environ.pop("AIA_MAX_ATTACHMENT_BYTES", None)
|
|
else:
|
|
os.environ["AIA_MAX_ATTACHMENT_BYTES"] = prev
|
|
block = out["result"]["content"][0]
|
|
assert block["type"] == "binary_ref"
|
|
assert block.get("error") == "attachment_too_large"
|
|
assert int(block.get("actual_bytes") or 0) >= 5
|
|
assert refs == []
|