From 94da88a3b2b3c20ea997d440b80eec2206b840ff Mon Sep 17 00:00:00 2001 From: Karsten Lehmann Date: Fri, 28 Aug 2020 20:03:26 +0200 Subject: [PATCH] Updated media test to work with latest changes --- tests/test_media.py | 26 +++++++++++++++++--------- 1 file changed, 17 insertions(+), 9 deletions(-) diff --git a/tests/test_media.py b/tests/test_media.py index 5e9f34f..a7aacd9 100644 --- a/tests/test_media.py +++ b/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() + )