fixed the test_functions --- didnt update when function got removed

db/test_functions/create_data.py

@@ -60,9 +60,17 @@ def seed():
         language_code="en",
     )
     print("Seed data inserted.")
-    entries = db.list_entries()
-    for e in entries:
-        print(f"  {e['provider_id']} | {e['manga_title']} ch.{e['chapter_number']} | {e['last_updated']}")
+    mangas = db.list_mangas(order_by="updated_at", order_desc=True)
+    print("\n=== list_mangas ===")
+    for m in mangas:
+        print(f"  {m['provider_id']} | {m['manga_title']}")
+
+    print("\n=== list_chapters (first manga) ===")
+    if mangas:
+        first = mangas[0]
+        chapters = db.list_chapters(first["manga_title"], provider_id=first["provider_id"])
+        for c in chapters:
+            print(f"  {c['provider_id']} | {c['manga_title']} ch.{c['chapter_number']} | {c['updated_at']}")
 
 
 if __name__ == "__main__":

db/test_functions/delete_data.py

@@ -8,8 +8,7 @@ import db
 
 
 def clear_seed_data():
-    """Remove the manga chapters inserted by seed_data.py."""
-    # Match the data from seed_data.py
+    """Remove the manga chapters inserted by create_data.py."""
     to_delete = [
         (db.PROVIDER_LOCAL, "One Piece", 1.0),
         (db.PROVIDER_LOCAL, "One Piece", 2.0),
@@ -23,11 +22,17 @@ def clear_seed_data():
 
 def clear_all():
     """Delete all chapters, then all manga."""
-    entries = db.list_entries()
-    for e in entries:
-        db.delete_chapter_segments(e["provider_id"], e["manga_title"], e["chapter_number"])
-        print(f"Deleted {e['provider_id']} | {e['manga_title']} ch.{e['chapter_number']}")
-    print(f"Cleared {len(entries)} chapter(s).")
+    mangas = db.list_mangas()
+    deleted = 0
+    for m in mangas:
+        provider_id = m["provider_id"]
+        manga_title = m["manga_title"]
+        chapters = db.list_chapters(manga_title, provider_id=provider_id)
+        for ch in chapters:
+            db.delete_chapter_segments(provider_id, manga_title, ch["chapter_number"])
+            print(f"Deleted {provider_id} | {manga_title} ch.{ch['chapter_number']}")
+            deleted += 1
+    print(f"Cleared {deleted} chapter(s).")
     db.delete_all_manga()
     print("Cleared all manga.")
 

db/test_functions/query_data.py

@@ -8,27 +8,32 @@ import db
 
 
 def run():
-    print("=== list_entries (default: newest first) ===\n")
-    entries = db.list_entries()
-    for e in entries:
-        print(f"  {e['provider_id']} | {e['manga_title']} ch.{e['chapter_number']} | {e['last_updated']}")
-    print(f"\n  Total: {len(entries)} chapter(s)\n")
-
-    print("=== list_entries (order_by manga_title, asc) ===\n")
-    entries_by_title = db.list_entries(order_by="manga_title", order_desc=False)
-    for e in entries_by_title:
-        print(f"  {e['provider_id']} | {e['manga_title']} ch.{e['chapter_number']}")
-    print()
-
-    if not entries:
-        print("No data. Run:  python -m db.seed_data")
+    print("=== list_mangas (default: newest first) ===\n")
+    mangas = db.list_mangas()
+    for m in mangas:
+        print(f"  {m['provider_id']} | {m['manga_title']}")
+    print(f"\n  Total: {len(mangas)} manga(s)\n")
+
+    if not mangas:
+        print("No data. Run:  python -m db.test_functions.create_data")
+        return
+
+    first_manga = mangas[0]
+    provider_id = first_manga["provider_id"]
+    manga_title = first_manga["manga_title"]
+
+    chapters = db.list_chapters(manga_title, provider_id=provider_id)
+    if not chapters:
+        print(f"No chapters for {manga_title}. Run create_data to add pages.")
         return
+    chapter_number = chapters[0]["chapter_number"]
 
-    # Use first entry for get_segments / get_chapter_segments examples
-    first = entries[0]
-    provider_id = first["provider_id"]
-    manga_title = first["manga_title"]
-    chapter_number = first["chapter_number"]
+    print(f"=== list_chapters({manga_title!r}, provider_id={provider_id!r}) ===\n")
+    for c in chapters[:5]:
+        print(f"  ch.{c['chapter_number']} (id={c['id']})")
+    if len(chapters) > 5:
+        print(f"  ... and {len(chapters) - 5} more chapter(s)")
+    print(f"\n  Total: {len(chapters)} chapter(s)\n")
 
     print(f"=== get_chapter_segments({provider_id!r}, {manga_title!r}, {chapter_number}) ===\n")
     segments = db.get_chapter_segments(provider_id, manga_title, chapter_number)