Jonathan
30 lines
1.2 KiB
Python
30 lines
1.2 KiB
Python
from lyra_worker.adapters.qobuz import QobuzAdapter
|
|
from lyra_worker.claim import claim_next
|
|
from lyra_worker.pipeline import run_pipeline
|
|
from lyra_worker.registry import build_resolver
|
|
from lyra_worker.types import MBTarget
|
|
from tests.conftest import insert_request
|
|
from tests.test_qobuz_adapter import FakeQobuzClient
|
|
|
|
|
|
def test_build_resolver_returns_a_resolver():
|
|
r = build_resolver()
|
|
assert hasattr(r, "resolve")
|
|
|
|
|
|
def test_enriched_target_flows_end_to_end(conn):
|
|
class FakeResolver:
|
|
def resolve(self, artist, album):
|
|
return MBTarget(artist="John Mayer", album="Continuum", track_count=12, year=2006)
|
|
|
|
# FakeQobuzClient defaults to a 12-track match and 12-track download -> imports cleanly.
|
|
job_id = insert_request(conn, artist="j. mayer", album="continuum (deluxe)")
|
|
claim_next(conn)
|
|
run_pipeline(conn, job_id, [QobuzAdapter(FakeQobuzClient())], resolver=FakeResolver(), dest_root="/tmp/lib")
|
|
|
|
with conn.cursor() as cur:
|
|
cur.execute('SELECT state FROM "Job" WHERE id = %s', (job_id,))
|
|
assert cur.fetchone()[0] == "imported"
|
|
cur.execute('SELECT artist, album FROM "LibraryItem"')
|
|
assert cur.fetchone() == ("John Mayer", "Continuum") # canonicalized, not the messy request
|