61 lines
1.9 KiB
Python
61 lines
1.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import os
|
|
import unittest
|
|
import configparser
|
|
|
|
from ankisyncd.collection import CollectionWrapper
|
|
from ankisyncd.collection import get_collection_wrapper
|
|
|
|
import helpers.server_utils
|
|
|
|
|
|
class FakeCollectionWrapper(CollectionWrapper):
|
|
def __init__(self, config, path, setup_new_collection=None):
|
|
self._CollectionWrapper__col = None
|
|
pass
|
|
|
|
|
|
class BadCollectionWrapper:
|
|
pass
|
|
|
|
|
|
class CollectionWrapperFactoryTest(unittest.TestCase):
|
|
def test_get_collection_wrapper(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)
|
|
path = os.path.realpath("fake/collection.anki2")
|
|
|
|
# Use custom files and dirs in settings. Should be CollectionWrapper
|
|
config["sync_app"].update(server_paths)
|
|
self.assertTrue(
|
|
type(get_collection_wrapper(config["sync_app"], path) == CollectionWrapper)
|
|
)
|
|
|
|
# A conf-specified CollectionWrapper is loaded
|
|
config.set(
|
|
"sync_app",
|
|
"collection_wrapper",
|
|
"test_collection_wrappers.FakeCollectionWrapper",
|
|
)
|
|
self.assertTrue(
|
|
type(get_collection_wrapper(config["sync_app"], path))
|
|
== FakeCollectionWrapper
|
|
)
|
|
|
|
# Should fail at load time if the class doesn't inherit from CollectionWrapper
|
|
config.set(
|
|
"sync_app",
|
|
"collection_wrapper",
|
|
"test_collection_wrappers.BadCollectionWrapper",
|
|
)
|
|
with self.assertRaises(TypeError):
|
|
pm = get_collection_wrapper(config["sync_app"], path)
|