gahusb
77b8d05ad7
- batch_generator.py: 장르별 N트랙 순차 Suno 생성 → 자동 compile → 자동 video pipeline
- main.py: POST/GET /api/music/generate-batch, GET /api/music/generate-batch/{id} 추가
- tests: 10개 endpoint 테스트 (검증·필터·404)
90 lines
3.0 KiB
Python
90 lines
3.0 KiB
Python
import pytest
|
|
from unittest.mock import AsyncMock, patch
|
|
from fastapi.testclient import TestClient
|
|
import app.main as main_module
|
|
from app import db
|
|
|
|
|
|
@pytest.fixture
|
|
def client(monkeypatch, tmp_path):
|
|
monkeypatch.setattr(db, "DB_PATH", str(tmp_path / "music.db"))
|
|
db.init_db()
|
|
monkeypatch.setenv("SUNO_API_KEY", "test")
|
|
# main.py의 SUNO_API_KEY 모듈 변수도 갱신 필요할 수 있음
|
|
monkeypatch.setattr(main_module, "SUNO_API_KEY", "test", raising=False)
|
|
return TestClient(main_module.app)
|
|
|
|
|
|
def test_create_batch_201(client):
|
|
with patch.object(main_module, "_run_batch", new=AsyncMock()):
|
|
r = client.post("/api/music/generate-batch",
|
|
json={"genre": "lo-fi", "count": 3})
|
|
assert r.status_code == 201, r.text
|
|
body = r.json()
|
|
assert body["genre"] == "lo-fi"
|
|
assert body["count"] == 3
|
|
assert body["status"] == "queued"
|
|
|
|
|
|
def test_create_batch_rejects_count_too_high(client):
|
|
r = client.post("/api/music/generate-batch",
|
|
json={"genre": "lo-fi", "count": 11})
|
|
assert r.status_code == 400
|
|
|
|
|
|
def test_create_batch_rejects_count_zero(client):
|
|
r = client.post("/api/music/generate-batch",
|
|
json={"genre": "lo-fi", "count": 0})
|
|
assert r.status_code == 400
|
|
|
|
|
|
def test_create_batch_rejects_no_genre(client):
|
|
r = client.post("/api/music/generate-batch", json={"count": 3})
|
|
assert r.status_code in (400, 422)
|
|
|
|
|
|
def test_create_batch_rejects_invalid_duration(client):
|
|
r = client.post("/api/music/generate-batch",
|
|
json={"genre": "lo-fi", "count": 3, "target_duration_sec": 30})
|
|
assert r.status_code == 400
|
|
|
|
|
|
def test_create_batch_rejects_no_suno_key(client, monkeypatch):
|
|
monkeypatch.setattr(main_module, "SUNO_API_KEY", "", raising=False)
|
|
r = client.post("/api/music/generate-batch",
|
|
json={"genre": "lo-fi", "count": 3})
|
|
assert r.status_code == 400
|
|
|
|
|
|
def test_get_batch_returns_tracks(client):
|
|
bid = db.create_batch_job(genre="lo-fi", count=2)
|
|
db.append_batch_track(bid, 999) # phantom track
|
|
r = client.get(f"/api/music/generate-batch/{bid}")
|
|
assert r.status_code == 200
|
|
body = r.json()
|
|
assert body["track_ids"] == [999]
|
|
assert body["tracks"] == [] # 999 not in music_library
|
|
|
|
|
|
def test_get_batch_404(client):
|
|
r = client.get("/api/music/generate-batch/99999")
|
|
assert r.status_code == 404
|
|
|
|
|
|
def test_list_batches(client):
|
|
db.create_batch_job(genre="lo-fi", count=1)
|
|
db.create_batch_job(genre="phonk", count=2)
|
|
r = client.get("/api/music/generate-batch")
|
|
assert r.status_code == 200
|
|
assert len(r.json()["batches"]) == 2
|
|
|
|
|
|
def test_list_batches_active_filter(client):
|
|
b1 = db.create_batch_job(genre="lo-fi", count=1)
|
|
b2 = db.create_batch_job(genre="phonk", count=2)
|
|
db.update_batch_job(b1, status="failed")
|
|
r = client.get("/api/music/generate-batch?status=active")
|
|
ids = [j["id"] for j in r.json()["batches"]]
|
|
assert b2 in ids
|
|
assert b1 not in ids
|