chengkai3
72 lines
3.3 KiB
Python
72 lines
3.3 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from app.core.config import get_settings
|
|
from app.services import legacy_atp_adapter
|
|
|
|
|
|
def test_resolve_legacy_atp_job_rejects_paths_outside_allowed_root(monkeypatch, tmp_path: Path) -> None:
|
|
settings = get_settings()
|
|
monkeypatch.setattr(settings, "wine_allowed_root", str(tmp_path / "wine-root"))
|
|
monkeypatch.setattr(settings, "atp_legacy_root", str(tmp_path / "outside" / "ATP"))
|
|
monkeypatch.setattr(settings, "atp_template_root", str(tmp_path / "outside" / "ATP" / "templates"))
|
|
monkeypatch.setattr(settings, "atp_run_root", str(tmp_path / "wine-root" / "runs"))
|
|
|
|
with pytest.raises(RuntimeError, match="legacy_root必须位于"):
|
|
legacy_atp_adapter.resolve_legacy_atp_job(adapter_config_json={}, execution_options={})
|
|
|
|
|
|
def test_resolve_legacy_atp_job_reports_missing_egm_directory(monkeypatch, tmp_path: Path) -> None:
|
|
allowed_root = tmp_path / "wine-root"
|
|
template_root = allowed_root / "ATP" / "templates"
|
|
template_dir = template_root / "raoji1"
|
|
template_dir.mkdir(parents=True)
|
|
(template_dir / "sample.atp").write_text("0123456789" * 200, encoding="utf-8")
|
|
(allowed_root / "ATP" / "rjtzl.exe").write_text("binary", encoding="utf-8")
|
|
(allowed_root / "ATP" / "tpbig.exe").write_text("binary", encoding="utf-8")
|
|
|
|
settings = get_settings()
|
|
monkeypatch.setattr(settings, "wine_allowed_root", str(allowed_root))
|
|
monkeypatch.setattr(settings, "atp_legacy_root", str(allowed_root / "ATP"))
|
|
monkeypatch.setattr(settings, "atp_template_root", str(template_root))
|
|
monkeypatch.setattr(settings, "atp_run_root", str(allowed_root / "runs"))
|
|
monkeypatch.setattr(settings, "atp_rjtzl_executable", "ATP/rjtzl.exe")
|
|
monkeypatch.setattr(settings, "atp_tpbig_executable", "ATP/tpbig.exe")
|
|
|
|
with pytest.raises(RuntimeError, match="EGM子目录不存在"):
|
|
legacy_atp_adapter.resolve_legacy_atp_job(
|
|
adapter_config_json={
|
|
"calculation_mode": "raoji1",
|
|
"template_subdir": "raoji1",
|
|
"use_egm": True,
|
|
},
|
|
execution_options={},
|
|
)
|
|
|
|
|
|
def test_build_legacy_atp_status_checks_marks_missing_binaries(monkeypatch, tmp_path: Path) -> None:
|
|
allowed_root = tmp_path / "wine-root"
|
|
template_root = allowed_root / "ATP" / "templates"
|
|
template_root.mkdir(parents=True)
|
|
(template_root / "EGM").mkdir(parents=True)
|
|
|
|
settings = get_settings()
|
|
monkeypatch.setattr(settings, "wine_allowed_root", str(allowed_root))
|
|
monkeypatch.setattr(settings, "wine_binary_path", "wine")
|
|
monkeypatch.setattr(settings, "atp_legacy_root", str(allowed_root / "ATP"))
|
|
monkeypatch.setattr(settings, "atp_template_root", str(template_root))
|
|
monkeypatch.setattr(settings, "atp_run_root", str(allowed_root / "runs"))
|
|
monkeypatch.setattr(settings, "atp_tpbig_executable", "ATP/tpbig.exe")
|
|
monkeypatch.setattr(settings, "atp_rjtzl_executable", "ATP/rjtzl.exe")
|
|
monkeypatch.setattr(legacy_atp_adapter, "_resolve_binary", lambda _raw: None)
|
|
|
|
checks = legacy_atp_adapter.build_legacy_atp_status_checks()
|
|
|
|
assert checks["wine"]["available"] is False
|
|
assert checks["tpbig_executable"]["available"] is False
|
|
assert checks["rjtzl_executable"]["available"] is False
|
|
assert checks["egm_subdir"]["available"] is True
|