91 lines
2.7 KiB
Python
91 lines
2.7 KiB
Python
#!/usr/bin/env python3
|
|
"""BtToxin_Digger 容器最小可行测试
|
|
|
|
使用 tests/test_data 下的 97-27.fna 与 C15.fna 作为输入，验证 nucl 类型分析，
|
|
检查是否生成关键输出文件。
|
|
"""
|
|
from pathlib import Path
|
|
import sys
|
|
import logging
|
|
|
|
sys.path.append(str(Path(__file__).resolve().parents[1] / "backend"))
|
|
from app.utils.docker_client import DockerContainerManager # noqa: E402
|
|
|
|
|
|
logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def ensure_test_layout() -> tuple[Path, Path, Path]:
|
|
base = Path(__file__).resolve().parents[1] / "tests"
|
|
input_dir = base / "test_data"
|
|
output_dir = base / "output"
|
|
log_dir = base / "logs"
|
|
for d in (input_dir, output_dir, log_dir):
|
|
d.mkdir(parents=True, exist_ok=True)
|
|
return input_dir, output_dir, log_dir
|
|
|
|
|
|
def check_input_files(input_dir: Path) -> bool:
|
|
files = ["97-27.fna", "C15.fna"]
|
|
missing = [f for f in files if not (input_dir / f).exists()]
|
|
if missing:
|
|
logger.error(f"缺少测试输入文件: {missing}")
|
|
logger.info("请将 97-27.fna 与 C15.fna 放入 tests/test_data 目录")
|
|
return False
|
|
logger.info(f"发现测试文件: {files}")
|
|
return True
|
|
|
|
|
|
def run_test() -> int:
|
|
input_dir, output_dir, log_dir = ensure_test_layout()
|
|
if not check_input_files(input_dir):
|
|
return 1
|
|
|
|
mgr = DockerContainerManager(
|
|
image="quay.io/biocontainers/bttoxin_digger:1.0.10--hdfd78af_0",
|
|
platform="linux/amd64",
|
|
)
|
|
|
|
logger.info("开始 BtToxin_Digger 分析 (nucl)")
|
|
result = mgr.run_bttoxin_digger(
|
|
input_dir=input_dir,
|
|
output_dir=output_dir,
|
|
log_dir=log_dir,
|
|
sequence_type="nucl",
|
|
scaf_suffix=".fna",
|
|
threads=4,
|
|
)
|
|
|
|
if not result.get("success"):
|
|
logger.error(f"分析失败: {result.get('error')} (exit={result.get('exit_code')})")
|
|
logs = result.get("logs")
|
|
if logs:
|
|
logger.error(f"容器日志(节选):\n{logs[:1000]}")
|
|
return 2
|
|
|
|
toxins_dir = output_dir / "Results" / "Toxins"
|
|
expected = [
|
|
toxins_dir / "97-27.list",
|
|
toxins_dir / "97-27.gbk",
|
|
toxins_dir / "C15.list",
|
|
toxins_dir / "C15.gbk",
|
|
toxins_dir / "Bt_all_genes.table",
|
|
toxins_dir / "All_Toxins.txt",
|
|
]
|
|
missing = [p.name for p in expected if not p.exists()]
|
|
if missing:
|
|
logger.error(f"缺少预期输出文件: {missing}")
|
|
return 3
|
|
|
|
logger.info("✓ 测试通过，关键输出文件存在")
|
|
logger.info(f"输出目录: {toxins_dir}")
|
|
logger.info(f"日志文件: {log_dir / 'digger_execution.log'}")
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(run_test())
|
|
|
|
|