Move the upload/download sqlite3 file logic to a manager

Also add a factory method so the manager can be controlled via config
This commit is contained in:
Anton Melser
2019-01-28 21:28:30 +08:00
parent 50cc6a12d9
commit 9ee9697582
3 changed files with 109 additions and 28 deletions

44
tests/test_full_sync.py Normal file
View File

@@ -0,0 +1,44 @@
# -*- coding: utf-8 -*-
import os
import unittest
import configparser
from ankisyncd.full_sync import FullSyncManager, get_full_sync_manager
import helpers.server_utils
class FakeFullSyncManager(FullSyncManager):
def __init__(self, config):
pass
class BadFullSyncManager:
pass
class FullSyncManagerFactoryTest(unittest.TestCase):
def test_get_full_sync_manager(self):
# Get absolute path to development ini file.
script_dir = os.path.dirname(os.path.realpath(__file__))
ini_file_path = os.path.join(script_dir,
"assets",
"test.conf")
# Create temporary files and dirs the server will use.
server_paths = helpers.server_utils.create_server_paths()
config = configparser.ConfigParser()
config.read(ini_file_path)
# Use custom files and dirs in settings. Should be PersistenceManager
config['sync_app'].update(server_paths)
self.assertTrue(type(get_full_sync_manager(config['sync_app']) == FullSyncManager))
# A conf-specified FullSyncManager is loaded
config.set("sync_app", "full_sync_manager", 'test_full_sync.FakeFullSyncManager')
self.assertTrue(type(get_full_sync_manager(config['sync_app'])) == FakeFullSyncManager)
# Should fail at load time if the class doesn't inherit from FullSyncManager
config.set("sync_app", "full_sync_manager", 'test_full_sync.BadFullSyncManager')
with self.assertRaises(TypeError):
pm = get_full_sync_manager(config['sync_app'])