fix tests

dylanljones · Dec 14, 2023 · fbda9d1 · fbda9d1
1 parent 1e1be38
commit fbda9d1
Show file tree

Hide file tree

Showing 3 changed files with 44 additions and 4 deletions.
diff --git a/pyrekordbox/db6/database.py b/pyrekordbox/db6/database.py
@@ -1991,7 +1991,7 @@ def to_dict(self, verbose=False):
             a list of dicts.
         """
         data = dict()
-        for table_name in tables.__all__:
+        for table_name in tables.TABLES:
             if table_name.startswith("Stats") or table_name == "Base":
                 continue
             if verbose:

diff --git a/pyrekordbox/db6/tables.py b/pyrekordbox/db6/tables.py
@@ -17,6 +17,7 @@
 from .registry import RekordboxAgentRegistry
 
 __all__ = [
+    "TABLES",
     "PlaylistType",
     "StatsTime",
     "StatsFull",
@@ -60,6 +61,47 @@
 ]
 
 
+TABLES = [
+    "AgentRegistry",
+    "CloudAgentRegistry",
+    "ContentActiveCensor",
+    "ContentCue",
+    "ContentFile",
+    "DjmdActiveCensor",
+    "DjmdAlbum",
+    "DjmdArtist",
+    "DjmdCategory",
+    "DjmdColor",
+    "DjmdContent",
+    "DjmdCue",
+    "DjmdDevice",
+    "DjmdGenre",
+    "DjmdHistory",
+    "DjmdHotCueBanklist",
+    "DjmdKey",
+    "DjmdLabel",
+    "DjmdMenuItems",
+    "DjmdMixerParam",
+    "DjmdMyTag",
+    "DjmdPlaylist",
+    "DjmdProperty",
+    "DjmdRelatedTracks",
+    "DjmdSampler",
+    "DjmdSongHistory",
+    "DjmdSongHotCueBanklist",
+    "DjmdSongMyTag",
+    "DjmdSongPlaylist",
+    "DjmdSongRelatedTracks",
+    "DjmdSongSampler",
+    "DjmdSongTagList",
+    "DjmdSort",
+    "HotCueBanklistCue",
+    "ImageFile",
+    "SettingFile",
+    "UuidIDMap",
+]
+
+
 class DateTime(TypeDecorator):
     """Custom datetime column with timezone support.
 

diff --git a/tests/test_db6.py b/tests/test_db6.py
@@ -1101,9 +1101,7 @@ def test_copy_unlocked():
 
     db2 = Rekordbox6Database(UNLOCKED_OUT, unlock=False)
     # Check everything got copied
-    for name in tables.__all__:
-        if name.startswith("Stats") or name == "Base":
-            continue
+    for name in tables.TABLES:
         table = getattr(tables, name)
         for row in db.query(table):
             data = row.to_dict()