diff --git a/tests/test_sync_app.py b/tests/test_sync_app.py index 8e3ff89..d49adec 100644 --- a/tests/test_sync_app.py +++ b/tests/test_sync_app.py @@ -3,9 +3,9 @@ import os import sqlite3 import tempfile import unittest +from unittest.mock import MagicMock, Mock -from anki.consts import SYNC_VER - +from ankisyncd.sync import SYNC_VER from ankisyncd.sync_app import SyncCollectionHandler from ankisyncd.sync_app import SyncUserSession @@ -14,8 +14,13 @@ from collection_test_base import CollectionTestBase class SyncCollectionHandlerTest(CollectionTestBase): def setUp(self): - CollectionTestBase.setUp(self) - self.syncCollectionHandler = SyncCollectionHandler(self.collection) + super().setUp() + self.session = MagicMock() + self.session.name = 'test' + self.syncCollectionHandler = SyncCollectionHandler( + self.collection, + self.session + ) def tearDown(self): CollectionTestBase.tearDown(self) @@ -60,6 +65,7 @@ class SyncCollectionHandlerTest(CollectionTestBase): self.assertTrue((type(meta['ts']) == int) and meta['ts'] > 0) self.assertEqual(meta['mod'], self.collection.mod) self.assertEqual(meta['usn'], self.collection._usn) + self.assertEqual(meta['uname'], self.session.name) self.assertEqual(meta['musn'], self.collection.media.lastUsn()) self.assertEqual(meta['msg'], '') self.assertEqual(meta['cont'], True)