diff --git a/music-lab/tests/test_pipeline_flow.py b/music-lab/tests/test_pipeline_flow.py new file mode 100644 index 0000000..4e6c910 --- /dev/null +++ b/music-lab/tests/test_pipeline_flow.py @@ -0,0 +1,113 @@ +"""풀 파이프라인 happy-path 통합 테스트 — 모든 외부 호출 mock.""" +import pytest +import sqlite3 +from unittest.mock import AsyncMock, patch +from fastapi.testclient import TestClient +from app.main import app +from app import db + + +@pytest.fixture +def client(monkeypatch, tmp_path): + monkeypatch.setattr(db, "DB_PATH", str(tmp_path / "music.db")) + db.init_db() + # 기본 트랙 1개 등록 (music_library 테이블) + conn = sqlite3.connect(db.DB_PATH) + cur = conn.cursor() + try: + cur.execute( + """INSERT INTO music_library + (id, title, genre, moods, instruments, duration_sec, bpm, key, scale, + prompt, audio_url, file_path, task_id, tags) + VALUES (1, 'Integration Test', 'lo-fi', '["chill"]', '["piano"]', + 120, 85, 'C', 'maj', 'p', + '/media/music/x.mp3', '/app/data/music/x.mp3', NULL, '[]')""", + ) + conn.commit() + except sqlite3.OperationalError as e: + # 스키마가 다르면 테스트 스킵 가이드 표시 + pytest.skip(f"music_library schema mismatch: {e}") + conn.close() + return TestClient(app) + + +# `new=...`로 patch한 항목은 mock 인자를 함수에 주입하지 않음. +# return_value로 patch한 두 개(cover/video는 new=AsyncMock 사용 → 미주입, +# thumb/video는 return_value → 주입, youtube.upload_video도 return_value → 주입). +# 데코레이터는 아래에서 위 순서로 인자에 주입됨: +# 1) cover (new=AsyncMock) → 미주입 +# 2) video (return_value) → 주입 → mock_video +# 3) thumb (return_value) → 주입 → mock_thumb +# 4) metadata (new=AsyncMock) → 미주입 +# 5) review (new=AsyncMock) → 미주입 +# 6) youtube.upload_video (return_value) → 주입 → mock_yt +@patch("app.pipeline.youtube.upload_video", return_value={"video_id": "VID999"}) +@patch("app.pipeline.review.run_4_axis", new=AsyncMock(return_value={ + "metadata_quality": {"score": 80, "notes": "ok"}, + "policy_compliance": {"score": 90, "issues": []}, + "viewer_experience": {"score": 80, "notes": "ok"}, + "trend_alignment": {"score": 70, "matched_keywords": ["lofi"]}, + "weighted_total": 80.0, "verdict": "pass", "summary": "good", "used_fallback": False, +})) +@patch("app.pipeline.metadata.generate", new=AsyncMock(return_value={ + "title": "Integration Test", "description": "Test desc", + "tags": ["lofi"], "category_id": 10, "used_fallback": False, "error": None, +})) +@patch("app.pipeline.thumb.generate", return_value={ + "url": "/media/videos/1/thumbnail.jpg", "used_fallback": False, +}) +@patch("app.pipeline.video.generate", return_value={ + "url": "/media/videos/1/video.mp4", "used_fallback": False, "duration_sec": 120, +}) +@patch("app.pipeline.cover.generate", new=AsyncMock(return_value={ + "url": "/media/videos/1/cover.jpg", "used_fallback": False, "error": None, +})) +def test_full_pipeline_happy_path(mock_video, mock_thumb, mock_yt, client): + # 1. 파이프라인 생성 + pid = client.post("/api/music/pipeline", json={"track_id": 1}).json()["id"] + assert db.get_pipeline(pid)["state"] == "created" + + # 2. 시작 → cover 자동 생성 → cover_pending + r = client.post(f"/api/music/pipeline/{pid}/start") + assert r.status_code == 202 + assert db.get_pipeline(pid)["state"] == "cover_pending" + + # 3. 4단계 사용자 승인 (cover, video, thumb, meta) + for step in ["cover", "video", "thumb", "meta"]: + r = client.post(f"/api/music/pipeline/{pid}/feedback", + json={"step": step, "intent": "approve"}) + assert r.status_code == 202 + + # 4. ai_review 자동 진행 후 publish_pending + p = db.get_pipeline(pid) + assert p["state"] == "publish_pending" + assert p["review"]["verdict"] == "pass" + + # 5. 발행 트리거 + r = client.post(f"/api/music/pipeline/{pid}/publish") + assert r.status_code == 202 + + # 6. 최종 published + p = db.get_pipeline(pid) + assert p["state"] == "published" + assert p["youtube_video_id"] == "VID999" + + +# cover.generate는 new=AsyncMock이므로 함수 인자에 주입되지 않음. +@patch("app.pipeline.cover.generate", new=AsyncMock(return_value={ + "url": "/media/videos/2/cover.jpg", "used_fallback": False, "error": None, +})) +def test_pipeline_reject_and_regenerate(client): + pid = client.post("/api/music/pipeline", json={"track_id": 1}).json()["id"] + client.post(f"/api/music/pipeline/{pid}/start") + assert db.get_pipeline(pid)["state"] == "cover_pending" + + # 반려 + 피드백 → 같은 단계 재진입 + r = client.post(f"/api/music/pipeline/{pid}/feedback", + json={"step": "cover", "intent": "reject", "feedback_text": "더 어둡게"}) + assert r.status_code == 202 + p = db.get_pipeline(pid) + assert p["state"] == "cover_pending" # 같은 단계 유지 + assert p["feedback_count_per_step"]["cover"] == 1 + history = db.get_feedback_history(pid) + assert history[0]["feedback_text"] == "더 어둡게"