Add database migration from 3.1.0 -> 3.1.2

tests/migration/test_v3_1_0.py

@@ -0,0 +1,65 @@
+import sqlite3
+from pathlib import Path
+import pytest
+
+import sqlite3
+from pathlib import Path
+import pytest
+
+from routes.migrations.v3_1_0 import MigrationV3_1_0
+
+# Override the autouse credentials fixture from conftest for this module
+@pytest.fixture(scope="session", autouse=True)
+def setup_credentials_for_tests():
+    # No-op to avoid external API calls
+    yield
+
+
+def _create_310_watch_artists_db(db_path: Path) -> None:
+    db_path.parent.mkdir(parents=True, exist_ok=True)
+    with sqlite3.connect(str(db_path)) as conn:
+        conn.executescript(
+            """
+            CREATE TABLE watched_artists (
+                spotify_id TEXT PRIMARY KEY,
+                name TEXT
+            );
+            CREATE TABLE "artist_a1b2c3" (
+                album_spotify_id TEXT PRIMARY KEY,
+                artist_spotify_id TEXT,
+                name TEXT,
+                album_type TEXT,
+                release_date TEXT,
+                total_tracks INTEGER,
+                link TEXT,
+                image_url TEXT,
+                added_to_db INTEGER,
+                last_seen_on_spotify INTEGER
+            );
+            """
+        )
+        conn.execute("INSERT INTO watched_artists (spotify_id) VALUES (?)", ('a1b2c3',))
+
+
+def test_watch_artists_migration(tmp_path):
+    # 1. Setup mock v3.1.0 database
+    db_path = tmp_path / "artists.db"
+    _create_310_watch_artists_db(db_path)
+
+    # 2. Run the migration
+    migration = MigrationV3_1_0()
+    with sqlite3.connect(db_path) as conn:
+        # Sanity check before migration
+        cur = conn.execute('PRAGMA table_info("artist_a1b2c3")')
+        columns_before = {row[1] for row in cur.fetchall()}
+        assert 'download_status' not in columns_before
+
+        # Apply migration
+        migration.update_watch_artists(conn)
+
+        # 3. Assert migration was successful
+        cur = conn.execute('PRAGMA table_info("artist_a1b2c3")')
+        columns_after = {row[1] for row in cur.fetchall()}
+
+    expected_columns = migration.ARTIST_ALBUMS_EXPECTED_COLUMNS.keys()
+    assert set(expected_columns).issubset(columns_after)