Updated media test to work with latest changes

tests/test_media.py

@@ -1,5 +1,6 @@
 import os.path
 import unittest
+from unittest.mock import MagicMock
 
 import ankisyncd.media
 import helpers.collection_utils
@@ -44,19 +45,26 @@ class ServerMediaManagerTest(unittest.TestCase):
             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,)])
+        self.assertEqual(
+            list(sm.db.execute("SELECT usn FROM media")),
+            [(161,), (161,)]
+        )
 
     def test_mediaChanges_lastUsn_order(self):
         col = self.colutils.create_empty_col()
         col.media = ankisyncd.media.ServerMediaManager(col)
-        mh = ankisyncd.sync_app.SyncMediaHandler(col)
-        mh.col.media.db.execute("""
-            INSERT INTO media (fname, usn, csum)
-            VALUES
+        session = MagicMock()
+        session.name = 'test'
+        mh = ankisyncd.sync_app.SyncMediaHandler(col, session)
+        mh.col.media.addMedia(
+            (
                 ('fileA', 101, '53059abba1a72c7aff34a3eaf7fef10ed65541ce'),
-                ('fileB', 100, 'a5ae546046d09559399c80fa7076fb10f1ce4bcd')
-        """)
-
+                ('fileB', 100, 'a5ae546046d09559399c80fa7076fb10f1ce4bcd'),
+            )
+        )
         # anki assumes mh.col.media.lastUsn() == mh.mediaChanges()['data'][-1][1]
         # ref: anki/sync.py:720 (commit cca3fcb2418880d0430a5c5c2e6b81ba260065b7)
-        self.assertEqual(mh.mediaChanges(lastUsn=99)['data'][-1][1], mh.col.media.lastUsn())
+        self.assertEqual(
+            mh.mediaChanges(lastUsn=99)['data'][-1][1],
+            mh.col.media.lastUsn()
+        )