75 lines
2.8 KiB
Python
75 lines
2.8 KiB
Python
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from backend.app import pipeline
|
|
from backend.app.pipeline import process_files
|
|
|
|
|
|
WORKDIR = Path("/Users/icemilk/Workspace/zld_POC")
|
|
AI_FILE = WORKDIR / "【2026-04-09】端午 - 背标 - 天问.ai"
|
|
DOCX_FILE = WORKDIR / "天问礼品粽【260331】.docx"
|
|
OUTPUT_DIR = WORKDIR / ".tmp_test_output"
|
|
|
|
|
|
def test_process_files_builds_preview_and_mineru_field_results(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
def fake_parse_with_mineru(_preview_path: Path, _output_dir: Path):
|
|
return {
|
|
"pdf_info": [
|
|
{
|
|
"page_idx": 0,
|
|
"page_size": [2772, 1961],
|
|
"para_blocks": [
|
|
{
|
|
"bbox": [704, 134, 2106, 229],
|
|
"lines": [{"spans": [{"content": "食品名称:天问礼品粽"}]}],
|
|
},
|
|
{
|
|
"bbox": [10, 20, 40, 60],
|
|
"lines": [{"spans": [{"content": "Word中不存在的内容"}]}],
|
|
},
|
|
],
|
|
}
|
|
]
|
|
}
|
|
|
|
monkeypatch.setattr(pipeline, "_parse_preview_with_mineru", fake_parse_with_mineru)
|
|
|
|
result = process_files(AI_FILE, DOCX_FILE, OUTPUT_DIR, job_id="test-job")
|
|
|
|
assert result["preview"]["type"] == "pdf"
|
|
assert result["preview"]["url"] == "/api/files/test-job/preview.pdf"
|
|
assert result["preview"]["pageWidthPt"] == 2772
|
|
assert result["preview"]["pageHeightPt"] == 1961
|
|
assert result["fields"][0]["text"] == "食品名称:天问礼品粽"
|
|
assert result["fields"][0]["validation_status"] == "matched"
|
|
assert result["fields"][0]["x0_pt"] == 704.0
|
|
assert any(field["validation_status"] == "unmatched" for field in result["fields"])
|
|
assert (OUTPUT_DIR / "preview.pdf").exists()
|
|
|
|
|
|
def test_parse_preview_with_mineru_reads_key_from_env_file(monkeypatch: pytest.MonkeyPatch, tmp_path: Path) -> None:
|
|
captured: dict[str, str] = {}
|
|
|
|
class FakeMineruClient:
|
|
def __init__(self, api_key: str) -> None:
|
|
captured["api_key"] = api_key
|
|
|
|
def parse_pdf(self, preview_path: Path, output_dir: Path) -> dict:
|
|
return {"preview_path": str(preview_path), "output_dir": str(output_dir)}
|
|
|
|
env_file = tmp_path / ".env"
|
|
env_file.write_text("MINERU_API_KEY=from-env-file\n", encoding="utf-8")
|
|
|
|
monkeypatch.delenv("MINERU_API_KEY", raising=False)
|
|
monkeypatch.setattr(pipeline, "ENV_FILE_CANDIDATES", (env_file,))
|
|
monkeypatch.setattr(pipeline, "MineruClient", FakeMineruClient)
|
|
|
|
preview_path = tmp_path / "preview.pdf"
|
|
preview_path.write_bytes(b"%PDF-1.7")
|
|
|
|
result = pipeline._parse_preview_with_mineru(preview_path, tmp_path)
|
|
|
|
assert captured["api_key"] == "from-env-file"
|
|
assert result["preview_path"] == str(preview_path)
|