cottongin/NtR-soudcloud-fetcher
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add database query methods for tracks, shows, and show_tracks

Browse Source 
Made-with: Cursor
This commit is contained in:
cottongin
2026-03-12 01:33:17 -04:00
parent 2b38fda25a
commit d2ab4e94c0
2 changed files with 463 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

207
tests/test_db.py
View File

@@ -1,8 +1,10 @@
import sqlite3
from datetime import datetime, timezone
import pytest
from ntr_fetcher.db import Database
from ntr_fetcher.models import Track
@pytest.fixture
@@ -28,3 +30,208 @@ def test_tables_created(db):
def test_initialize_idempotent(db):
"""Calling initialize twice doesn't raise."""
db.initialize()
def _make_track(id: int, liked_at: str, title: str = "Test", artist: str = "Artist") -> Track:
return Track(
id=id,
title=title,
artist=artist,
permalink_url=f"https://soundcloud.com/test/track-{id}",
artwork_url=None,
duration_ms=180000,
license="cc-by",
liked_at=datetime.fromisoformat(liked_at),
raw_json="{}",
)
def test_upsert_track(db):
track = _make_track(100, "2026-03-10T12:00:00+00:00")
db.upsert_track(track)
result = db.get_track(100)
assert result is not None
assert result.title == "Test"
def test_upsert_track_updates_existing(db):
track1 = _make_track(100, "2026-03-10T12:00:00+00:00", title="Original")
db.upsert_track(track1)
track2 = _make_track(100, "2026-03-10T12:00:00+00:00", title="Updated")
db.upsert_track(track2)
result = db.get_track(100)
assert result.title == "Updated"
def test_get_or_create_show(db):
week_start = datetime(2026, 3, 13, 2, 0, 0, tzinfo=timezone.utc)
week_end = datetime(2026, 3, 20, 2, 0, 0, tzinfo=timezone.utc)
show = db.get_or_create_show(week_start, week_end)
assert show.id is not None
assert show.week_start == week_start
show2 = db.get_or_create_show(week_start, week_end)
assert show2.id == show.id
def test_set_show_tracks(db):
week_start = datetime(2026, 3, 13, 2, 0, 0, tzinfo=timezone.utc)
week_end = datetime(2026, 3, 20, 2, 0, 0, tzinfo=timezone.utc)
show = db.get_or_create_show(week_start, week_end)
t1 = _make_track(1, "2026-03-14T01:00:00+00:00", title="First")
t2 = _make_track(2, "2026-03-14T02:00:00+00:00", title="Second")
db.upsert_track(t1)
db.upsert_track(t2)
db.set_show_tracks(show.id, [t1.id, t2.id])
tracks = db.get_show_tracks(show.id)
assert len(tracks) == 2
assert tracks[0]["position"] == 1
assert tracks[0]["title"] == "First"
assert tracks[1]["position"] == 2
def test_set_show_tracks_preserves_existing_positions(db):
week_start = datetime(2026, 3, 13, 2, 0, 0, tzinfo=timezone.utc)
week_end = datetime(2026, 3, 20, 2, 0, 0, tzinfo=timezone.utc)
show = db.get_or_create_show(week_start, week_end)
t1 = _make_track(1, "2026-03-14T01:00:00+00:00")
db.upsert_track(t1)
db.set_show_tracks(show.id, [t1.id])
t2 = _make_track(2, "2026-03-14T02:00:00+00:00")
db.upsert_track(t2)
db.set_show_tracks(show.id, [t1.id, t2.id])
tracks = db.get_show_tracks(show.id)
assert tracks[0]["track_id"] == 1
assert tracks[0]["position"] == 1
assert tracks[1]["track_id"] == 2
assert tracks[1]["position"] == 2
def test_set_show_tracks_removes_unliked(db):
"""Tracks no longer in the likes list are removed and positions re-compact."""
week_start = datetime(2026, 3, 13, 2, 0, 0, tzinfo=timezone.utc)
week_end = datetime(2026, 3, 20, 2, 0, 0, tzinfo=timezone.utc)
show = db.get_or_create_show(week_start, week_end)
t1 = _make_track(1, "2026-03-14T01:00:00+00:00", title="First")
t2 = _make_track(2, "2026-03-14T02:00:00+00:00", title="Second")
t3 = _make_track(3, "2026-03-14T03:00:00+00:00", title="Third")
db.upsert_track(t1)
db.upsert_track(t2)
db.upsert_track(t3)
db.set_show_tracks(show.id, [t1.id, t2.id, t3.id])
db.set_show_tracks(show.id, [t1.id, t3.id])
tracks = db.get_show_tracks(show.id)
assert len(tracks) == 2
assert tracks[0]["track_id"] == 1
assert tracks[0]["position"] == 1
assert tracks[1]["track_id"] == 3
assert tracks[1]["position"] == 2
def test_get_show_track_by_position(db):
week_start = datetime(2026, 3, 13, 2, 0, 0, tzinfo=timezone.utc)
week_end = datetime(2026, 3, 20, 2, 0, 0, tzinfo=timezone.utc)
show = db.get_or_create_show(week_start, week_end)
t1 = _make_track(1, "2026-03-14T01:00:00+00:00", title="First")
db.upsert_track(t1)
db.set_show_tracks(show.id, [t1.id])
result = db.get_show_track_by_position(show.id, 1)
assert result is not None
assert result["title"] == "First"
result_missing = db.get_show_track_by_position(show.id, 99)
assert result_missing is None
def test_list_shows(db):
s1 = db.get_or_create_show(
datetime(2026, 3, 6, 3, 0, 0, tzinfo=timezone.utc),
datetime(2026, 3, 13, 2, 0, 0, tzinfo=timezone.utc),
)
s2 = db.get_or_create_show(
datetime(2026, 3, 13, 2, 0, 0, tzinfo=timezone.utc),
datetime(2026, 3, 20, 2, 0, 0, tzinfo=timezone.utc),
)
shows = db.list_shows(limit=10, offset=0)
assert len(shows) == 2
assert shows[0].id == s2.id
def test_max_position_for_show(db):
week_start = datetime(2026, 3, 13, 2, 0, 0, tzinfo=timezone.utc)
week_end = datetime(2026, 3, 20, 2, 0, 0, tzinfo=timezone.utc)
show = db.get_or_create_show(week_start, week_end)
assert db.get_max_position(show.id) == 0
t1 = _make_track(1, "2026-03-14T01:00:00+00:00")
db.upsert_track(t1)
db.set_show_tracks(show.id, [t1.id])
assert db.get_max_position(show.id) == 1
def test_remove_show_track(db):
week_start = datetime(2026, 3, 13, 2, 0, 0, tzinfo=timezone.utc)
week_end = datetime(2026, 3, 20, 2, 0, 0, tzinfo=timezone.utc)
show = db.get_or_create_show(week_start, week_end)
t1 = _make_track(1, "2026-03-14T01:00:00+00:00")
t2 = _make_track(2, "2026-03-14T02:00:00+00:00")
t3 = _make_track(3, "2026-03-14T03:00:00+00:00")
db.upsert_track(t1)
db.upsert_track(t2)
db.upsert_track(t3)
db.set_show_tracks(show.id, [t1.id, t2.id, t3.id])
db.remove_show_track(show.id, 2)
tracks = db.get_show_tracks(show.id)
assert len(tracks) == 2
assert tracks[0]["position"] == 1
assert tracks[0]["track_id"] == 1
assert tracks[1]["position"] == 2
assert tracks[1]["track_id"] == 3
def test_move_show_track(db):
week_start = datetime(2026, 3, 13, 2, 0, 0, tzinfo=timezone.utc)
week_end = datetime(2026, 3, 20, 2, 0, 0, tzinfo=timezone.utc)
show = db.get_or_create_show(week_start, week_end)
t1 = _make_track(1, "2026-03-14T01:00:00+00:00")
t2 = _make_track(2, "2026-03-14T02:00:00+00:00")
t3 = _make_track(3, "2026-03-14T03:00:00+00:00")
db.upsert_track(t1)
db.upsert_track(t2)
db.upsert_track(t3)
db.set_show_tracks(show.id, [t1.id, t2.id, t3.id])
db.move_show_track(show.id, track_id=3, new_position=1)
tracks = db.get_show_tracks(show.id)
assert tracks[0]["track_id"] == 3
assert tracks[0]["position"] == 1
assert tracks[1]["track_id"] == 1
assert tracks[1]["position"] == 2
assert tracks[2]["track_id"] == 2
assert tracks[2]["position"] == 3
def test_add_track_to_show_at_position(db):
week_start = datetime(2026, 3, 13, 2, 0, 0, tzinfo=timezone.utc)
week_end = datetime(2026, 3, 20, 2, 0, 0, tzinfo=timezone.utc)
show = db.get_or_create_show(week_start, week_end)
t1 = _make_track(1, "2026-03-14T01:00:00+00:00")
t2 = _make_track(2, "2026-03-14T02:00:00+00:00")
t3 = _make_track(3, "2026-03-14T03:00:00+00:00")
db.upsert_track(t1)
db.upsert_track(t2)
db.upsert_track(t3)
db.set_show_tracks(show.id, [t1.id, t2.id])
db.add_track_to_show(show.id, track_id=3, position=2)
tracks = db.get_show_tracks(show.id)
assert len(tracks) == 3
assert tracks[0]["track_id"] == 1
assert tracks[1]["track_id"] == 3
assert tracks[2]["track_id"] == 2
def test_has_track_in_show(db):
week_start = datetime(2026, 3, 13, 2, 0, 0, tzinfo=timezone.utc)
week_end = datetime(2026, 3, 20, 2, 0, 0, tzinfo=timezone.utc)
show = db.get_or_create_show(week_start, week_end)
t1 = _make_track(1, "2026-03-14T01:00:00+00:00")
db.upsert_track(t1)
db.set_show_tracks(show.id, [t1.id])
assert db.has_track_in_show(show.id, 1) is True
assert db.has_track_in_show(show.id, 999) is False