78 lines
3.2 KiB
Python
78 lines
3.2 KiB
Python
import pytest
|
|
from unittest.mock import patch, MagicMock
|
|
from app.pipeline.orchestrator import _resolve_input
|
|
|
|
|
|
def test_resolve_input_track():
|
|
pipeline = {"id": 1, "track_id": 13, "compile_job_id": None}
|
|
track = {
|
|
"id": 13, "title": "Lo-Fi Drive", "genre": "lo-fi",
|
|
"moods": ["chill"], "duration_sec": 176,
|
|
"file_path": "/app/data/x.mp3", "audio_url": "/media/music/x.mp3",
|
|
}
|
|
with patch("app.pipeline.orchestrator.db.get_track_by_id", return_value=track):
|
|
result = _resolve_input(pipeline)
|
|
assert result["audio_path"] == "/app/data/x.mp3"
|
|
assert result["duration_sec"] == 176
|
|
assert len(result["tracks"]) == 1
|
|
assert result["tracks"][0]["start_offset_sec"] == 0
|
|
assert result["title"] == "Lo-Fi Drive"
|
|
assert result["genre"] == "lo-fi"
|
|
|
|
|
|
def test_resolve_input_compile_job():
|
|
pipeline = {"id": 2, "track_id": None, "compile_job_id": 5}
|
|
job = {
|
|
"id": 5, "status": "succeeded", "title": "Chill Mix",
|
|
"audio_path": "/app/data/compiles/5.mp3",
|
|
"track_ids": [13, 14, 15],
|
|
"crossfade_sec": 3,
|
|
}
|
|
tracks = {
|
|
13: {"id": 13, "title": "T1", "duration_sec": 180},
|
|
14: {"id": 14, "title": "T2", "duration_sec": 200},
|
|
15: {"id": 15, "title": "T3", "duration_sec": 150},
|
|
}
|
|
with patch("app.pipeline.orchestrator.db.get_compile_job", return_value=job), \
|
|
patch("app.pipeline.orchestrator.db.get_track_by_id", side_effect=lambda i: tracks[i]):
|
|
result = _resolve_input(pipeline)
|
|
assert result["audio_path"] == "/app/data/compiles/5.mp3"
|
|
# 누적 = 180+200+150 - 2*3(crossfade pair gaps) = 524
|
|
assert result["duration_sec"] == 524
|
|
assert len(result["tracks"]) == 3
|
|
assert result["tracks"][0]["start_offset_sec"] == 0
|
|
assert result["tracks"][1]["start_offset_sec"] == 177 # 180 - 3
|
|
assert result["tracks"][2]["start_offset_sec"] == 374 # 177 + 200 - 3
|
|
assert result["title"] == "Chill Mix"
|
|
assert result["genre"] == "mix"
|
|
|
|
|
|
def test_resolve_input_compile_not_ready():
|
|
pipeline = {"id": 3, "track_id": None, "compile_job_id": 6}
|
|
job = {"id": 6, "status": "rendering"}
|
|
with patch("app.pipeline.orchestrator.db.get_compile_job", return_value=job):
|
|
with pytest.raises(ValueError, match="not ready"):
|
|
_resolve_input(pipeline)
|
|
|
|
|
|
def test_resolve_input_neither():
|
|
pipeline = {"id": 4, "track_id": None, "compile_job_id": None}
|
|
with pytest.raises(ValueError):
|
|
_resolve_input(pipeline)
|
|
|
|
|
|
def test_resolve_input_compile_job_done_status():
|
|
"""compile job status='done'도 accept (production convention)."""
|
|
pipeline = {"id": 5, "track_id": None, "compile_job_id": 7}
|
|
job = {
|
|
"id": 7, "status": "done", "title": "Done Mix",
|
|
"audio_path": "/app/data/compiles/7.mp3",
|
|
"track_ids": [1], "crossfade_sec": 0,
|
|
}
|
|
track = {"id": 1, "title": "T1", "duration_sec": 100}
|
|
with patch("app.pipeline.orchestrator.db.get_compile_job", return_value=job), \
|
|
patch("app.pipeline.orchestrator.db.get_track_by_id", return_value=track):
|
|
result = _resolve_input(pipeline)
|
|
assert result["audio_path"] == "/app/data/compiles/7.mp3"
|
|
assert result["title"] == "Done Mix"
|