94 lines
2.5 KiB
Python
94 lines
2.5 KiB
Python
import runpy
|
|
from pathlib import Path
|
|
|
|
from wov_node_ass.__main__ import parse_srt, write_ass, invoke
|
|
from wov_sdk.models import InvokeRequest
|
|
|
|
|
|
SAMPLE_SRT = """
|
|
1
|
|
00:00:01,000 --> 00:00:03,000
|
|
第一行
|
|
第二行
|
|
|
|
2
|
|
00:00:04,000 --> 00:00:06,000
|
|
第三行
|
|
"""
|
|
|
|
|
|
def test_parse_and_write(tmp_path) -> None:
|
|
entries = parse_srt(SAMPLE_SRT)
|
|
assert len(entries) == 2
|
|
assert entries[0][2] == r"第一行\N第二行"
|
|
|
|
output = tmp_path / "out.ass"
|
|
write_ass(entries, output, "3840x1920")
|
|
content = output.read_text(encoding="utf-8")
|
|
assert "PlayResX: 3840" in content
|
|
assert "PlayResY: 1920" in content
|
|
assert "LeftEye" in content
|
|
assert "RightEye" in content
|
|
assert r"第一行\N第二行" in content
|
|
|
|
|
|
def test_parse_malformed(tmp_path) -> None:
|
|
source = tmp_path / "bad.srt"
|
|
source.write_text("1\nnot a time line\n", encoding="utf-8")
|
|
assert parse_srt(source.read_text(encoding="utf-8")) == []
|
|
|
|
source.write_text("1", encoding="utf-8")
|
|
assert parse_srt(source.read_text(encoding="utf-8")) == []
|
|
|
|
|
|
def test_invoke_success(tmp_path) -> None:
|
|
source = tmp_path / "in.srt"
|
|
source.write_text(SAMPLE_SRT, encoding="utf-8")
|
|
response = invoke(
|
|
InvokeRequest(
|
|
run_id="run_1",
|
|
node_instance_id="ni_1",
|
|
inputs={"cn_srt_uri": str(source)},
|
|
params={"resolution": "1920x1080"},
|
|
output_dir=str(tmp_path / "out"),
|
|
)
|
|
)
|
|
assert response.status == "completed"
|
|
assert "PlayResX: 1920" in Path(response.outputs["ass_uri"]).read_text(encoding="utf-8")
|
|
|
|
|
|
def test_invoke_missing_input(tmp_path) -> None:
|
|
response = invoke(
|
|
InvokeRequest(
|
|
run_id="run_2",
|
|
node_instance_id="ni_2",
|
|
inputs={},
|
|
output_dir=str(tmp_path),
|
|
)
|
|
)
|
|
assert response.status == "failed"
|
|
|
|
|
|
def test_invoke_missing_file(tmp_path) -> None:
|
|
response = invoke(
|
|
InvokeRequest(
|
|
run_id="run_3",
|
|
node_instance_id="ni_3",
|
|
inputs={"cn_srt_uri": str(tmp_path / "missing.srt")},
|
|
output_dir=str(tmp_path),
|
|
)
|
|
)
|
|
assert response.status == "failed"
|
|
|
|
|
|
def test_entrypoint(monkeypatch) -> None:
|
|
module_path = Path(__file__).resolve().parent.parent / "wov_node_ass" / "__main__.py"
|
|
captured = {}
|
|
|
|
def fake_run_node(manifest, handler) -> None:
|
|
captured["id"] = manifest.id
|
|
|
|
monkeypatch.setattr("wov_sdk.server.run_node", fake_run_node)
|
|
runpy.run_path(str(module_path), run_name="__main__")
|
|
assert captured["id"] == "srt-to-dual-eye-ass"
|