mirror of
https://github.com/hansjone/oclaw.git
synced 2026-10-09 05:50:44 +08:00
oliver
420abac9f1
- Rename platform/ to svc/ to avoid shadowing stdlib platform. - Replace from oclaw.* with from svc/runtime/interfaces; update -m CLI paths. - tests/conftest: prepend repo root to sys.path (no parent-folder package name). - CI: paths and offline_eval script under repo root. - Ops scripts: PYTHONPATH must be repo root for python -m runtime.* (fixes gateway/WhatsApp sidecar startup). - Fix default oclaw.json path in tabular/file attachment limits; stabilize attachment test config. Co-authored-by: Cursor <cursoragent@cursor.com>
64 lines
2.7 KiB
Python
64 lines
2.7 KiB
Python
from __future__ import annotations
|
|
|
|
import base64
|
|
import os
|
|
|
|
from 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 == []
|