38 lines
1.5 KiB
Python
38 lines
1.5 KiB
Python
# music-lab/tests/test_market.py
|
|
|
|
def test_ingest_and_report(tmp_db):
|
|
from app.db import init_db
|
|
from app.market import ingest_trends, get_suggestions
|
|
init_db()
|
|
|
|
trends = [
|
|
{"source": "youtube", "country": "BR", "genre": "lo-fi", "keyword": "lofi study", "score": 0.9, "rank": 1, "metadata": {}},
|
|
{"source": "youtube", "country": "ID", "genre": "pop", "keyword": "pop hits", "score": 0.7, "rank": 2, "metadata": {}},
|
|
{"source": "billboard", "country": "US", "genre": "pop", "keyword": "top 40", "score": 0.8, "rank": 1, "metadata": {}},
|
|
]
|
|
report = ingest_trends(trends, "2026-05-01")
|
|
assert report["report_date"] == "2026-05-01"
|
|
assert len(report["top_genres"]) >= 2
|
|
# pop이 lo-fi보다 score 높아야 함 (2건)
|
|
genres_by_score = [g["genre"] for g in report["top_genres"]]
|
|
assert genres_by_score[0] == "pop"
|
|
|
|
suggestions = get_suggestions(limit=3)
|
|
assert len(suggestions) >= 1
|
|
assert "suno_prompt" in suggestions[0]
|
|
|
|
|
|
def test_ingest_idempotent(tmp_db):
|
|
"""같은 날 두 번 ingest해도 report가 upsert 돼야 함."""
|
|
from app.db import init_db, get_trend_reports
|
|
from app.market import ingest_trends
|
|
init_db()
|
|
|
|
trends = [{"source": "youtube", "country": "BR", "genre": "lo-fi",
|
|
"keyword": "chill", "score": 0.8, "rank": 1, "metadata": {}}]
|
|
ingest_trends(trends, "2026-05-01")
|
|
ingest_trends(trends, "2026-05-01") # 두 번째
|
|
|
|
reports = get_trend_reports()
|
|
assert len([r for r in reports if r["report_date"] == "2026-05-01"]) == 1
|