Migrate old media DB if present

2019-12-31 16:56:17 +01:00
parent e8f274ee84
commit e8850f2a5a
2 changed files with 56 additions and 1 deletions
--- a/tests/test_media.py
+++ b/tests/test_media.py
@@ -0,0 +1,44 @@
+import os.path
+import unittest
+
+import ankisyncd.media
+import helpers.collection_utils
+
+
+class ServerMediaManagerTest(unittest.TestCase):
+    @classmethod
+    def setUpClass(cls):
+        cls.colutils = helpers.collection_utils.CollectionUtils()
+
+    @classmethod
+    def tearDownClass(cls):
+        cls.colutils.clean_up()
+        cls.colutils = None
+
+    def test_upgrade(self):
+        col = self.colutils.create_empty_col()
+        cm = col.media
+
+        fpath = os.path.join(cm.dir(), "file")
+        with open(fpath + "A", "w") as f:
+            f.write("some contents")
+        with open(fpath + "B", "w") as f:
+            f.write("other contents")
+        cm._logChanges()
+
+        self.assertEqual(
+            set(cm.db.execute("SELECT fname, csum FROM media")),
+            {
+                ("fileA", "53059abba1a72c7aff34a3eaf7fef10ed65541ce"),
+                ("fileB", "a5ae546046d09559399c80fa7076fb10f1ce4bcd"),
+            },
+        )
+        cm.setLastUsn(161)
+
+        sm = ankisyncd.media.ServerMediaManager(col)
+        self.assertEqual(
+            list(sm.db.execute("SELECT fname, csum FROM media")),
+            list(cm.db.execute("SELECT fname, csum FROM media")),
+        )
+        self.assertEqual(cm.lastUsn(), sm.lastUsn())
+        self.assertEqual(list(sm.db.execute("SELECT usn FROM media")), [(161,), (161,)])