feat(music-lab): orchestrator _resolve_input — track/compile_job 통합 입력

music-lab/tests/test_orchestrator_resolve.py

@@ -0,0 +1,61 @@
+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)