67 lines
2.6 KiB
Python
67 lines
2.6 KiB
Python
import os
|
|
import pytest
|
|
from unittest.mock import patch, MagicMock
|
|
from app.pipeline import video, thumb, storage
|
|
|
|
|
|
@pytest.fixture
|
|
def tmp_storage(monkeypatch, tmp_path):
|
|
monkeypatch.setattr(storage, "VIDEO_DATA_DIR", str(tmp_path))
|
|
# 더미 입력 파일들
|
|
audio = tmp_path / "audio.mp3"
|
|
audio.write_bytes(b"\x00" * 100)
|
|
cover_dir = tmp_path / "50"
|
|
cover_dir.mkdir()
|
|
cover = cover_dir / "cover.jpg"
|
|
cover.write_bytes(b"\x00" * 100)
|
|
return tmp_path
|
|
|
|
|
|
@patch("subprocess.run")
|
|
def test_generate_video_calls_ffmpeg(mock_run, tmp_storage):
|
|
mock_run.return_value = MagicMock(returncode=0, stderr="")
|
|
out = video.generate(pipeline_id=50, audio_path=str(tmp_storage / "audio.mp3"),
|
|
cover_path=str(tmp_storage / "50" / "cover.jpg"),
|
|
genre="lo-fi", duration_sec=120, resolution="1920x1080",
|
|
style="visualizer")
|
|
assert out["url"].endswith("/50/video.mp4")
|
|
assert out["used_fallback"] is False
|
|
args = mock_run.call_args[0][0]
|
|
assert args[0] == "ffmpeg"
|
|
assert "-i" in args
|
|
assert "showwaves" in " ".join(args)
|
|
|
|
|
|
@patch("subprocess.run")
|
|
def test_generate_video_failure_marks_failed(mock_run, tmp_storage):
|
|
mock_run.return_value = MagicMock(returncode=1, stderr="bad codec")
|
|
with pytest.raises(video.VideoGenerationError):
|
|
video.generate(pipeline_id=51, audio_path=str(tmp_storage / "audio.mp3"),
|
|
cover_path=str(tmp_storage / "50" / "cover.jpg"),
|
|
genre="lo-fi", duration_sec=120, resolution="1920x1080",
|
|
style="visualizer")
|
|
|
|
|
|
@patch("subprocess.run")
|
|
def test_thumb_extracts_frame(mock_run, tmp_storage):
|
|
mock_run.return_value = MagicMock(returncode=0, stderr="")
|
|
video_path = tmp_storage / "60" / "video.mp4"
|
|
video_path.parent.mkdir()
|
|
video_path.write_bytes(b"\x00" * 100)
|
|
out = thumb.generate(pipeline_id=60, video_path=str(video_path),
|
|
track_title="Midnight Drive", overlay_text=False)
|
|
assert out["url"].endswith("/60/thumbnail.jpg")
|
|
args = mock_run.call_args[0][0]
|
|
assert args[0] == "ffmpeg"
|
|
|
|
|
|
@patch("subprocess.run")
|
|
def test_thumb_failure_raises(mock_run, tmp_storage):
|
|
mock_run.return_value = MagicMock(returncode=1, stderr="bad input")
|
|
video_path = tmp_storage / "61" / "video.mp4"
|
|
video_path.parent.mkdir()
|
|
video_path.write_bytes(b"\x00" * 100)
|
|
with pytest.raises(thumb.ThumbGenerationError):
|
|
thumb.generate(pipeline_id=61, video_path=str(video_path),
|
|
track_title="X", overlay_text=False)
|