diff --git a/tests/base/__init__.py b/tests/base/__init__.py index 612c620..261ddd3 100644 --- a/tests/base/__init__.py +++ b/tests/base/__init__.py @@ -12,12 +12,14 @@ import unittest from .test_db import DbTestCase from .test_scanner import ScannerTestCase +from .test_watcher import suite as watcher_suite def suite(): suite = unittest.TestSuite() suite.addTest(unittest.makeSuite(DbTestCase)) suite.addTest(unittest.makeSuite(ScannerTestCase)) + suite.addTest(watcher_suite()) return suite diff --git a/tests/base/test_watcher.py b/tests/base/test_watcher.py new file mode 100644 index 0000000..1cdf7fc --- /dev/null +++ b/tests/base/test_watcher.py @@ -0,0 +1,144 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- +# vim:fenc=utf-8 +# +# This file is part of Supysonic. +# Supysonic is a Python implementation of the Subsonic server API. +# +# Copyright (C) 2017 Alban 'spl0k' FĂ©ron +# +# Distributed under terms of the GNU AGPLv3 license. + +import io +import os +import shutil +import tempfile +import time +import unittest + +from contextlib import contextmanager +from threading import Thread + +from supysonic.db import get_store, Track, Artist +from supysonic.managers.folder import FolderManager +from supysonic.watcher import SupysonicWatcher + +from ..testbase import TestConfig + +class WatcherTestConfig(TestConfig): + DAEMON = { + 'wait_delay': 0.5, + 'log_file': None, + 'log_level': 'DEBUG' + } + + def __init__(self, db_uri): + super(WatcherTestConfig, self).__init__(False, False) + self.BASE['database_uri'] = db_uri + +class WatcherTestBase(unittest.TestCase): + @contextmanager + def _get_store(self): + store = None + try: + store = get_store('sqlite:///' + self.__dbfile) + yield store + store.commit() + store.close() + except: + store.rollback() + store.close() + raise + + def setUp(self): + self.__dbfile = tempfile.mkstemp()[1] + conf = WatcherTestConfig('sqlite:///' + self.__dbfile) + self.__sleep_time = conf.DAEMON['wait_delay'] + 1 + + with self._get_store() as store: + with io.open('schema/sqlite.sql', 'r') as sql: + schema = sql.read() + for statement in schema.split(';'): + store.execute(statement) + + self.__watcher = SupysonicWatcher(conf) + self.__thread = Thread(target = self.__watcher.run) + + def tearDown(self): + os.unlink(self.__dbfile) + + def _start(self): + self.__thread.start() + time.sleep(0.2) + + def _stop(self): + self.__watcher.stop() + self.__thread.join() + + def _is_alive(self): + return self.__thread.is_alive() + + def _sleep(self): + time.sleep(self.__sleep_time) + +class NothingToWatchTestCase(WatcherTestBase): + def test_spawn_useless_watcher(self): + self._start() + time.sleep(0.2) + self.assertFalse(self._is_alive()) + self._stop() + +class WatcherTestCase(WatcherTestBase): + def setUp(self): + super(WatcherTestCase, self).setUp() + self.__dir = tempfile.mkdtemp() + with self._get_store() as store: + FolderManager.add(store, 'Folder', self.__dir) + self._start() + + def tearDown(self): + self._stop() + shutil.rmtree(self.__dir) + super(WatcherTestCase, self).tearDown() + + @staticmethod + def _tempname(): + with tempfile.NamedTemporaryFile() as f: + return os.path.basename(f.name) + + def assertTrackCountEqual(self, expected): + with self._get_store() as store: + self.assertEqual(store.find(Track).count(), expected) + + def test_add(self): + shutil.copyfile('tests/assets/folder/silence.mp3', os.path.join(self.__dir, self._tempname() + '.mp3')) + self.assertTrackCountEqual(0) + self._sleep() + self.assertTrackCountEqual(1) + + def test_add_nowait_stop(self): + shutil.copyfile('tests/assets/folder/silence.mp3', os.path.join(self.__dir, self._tempname() + '.mp3')) + self._stop() + self.assertTrackCountEqual(1) + + def test_add_multiple(self): + shutil.copyfile('tests/assets/folder/silence.mp3', os.path.join(self.__dir, self._tempname() + '.mp3')) + shutil.copyfile('tests/assets/folder/silence.mp3', os.path.join(self.__dir, self._tempname() + '.mp3')) + shutil.copyfile('tests/assets/folder/silence.mp3', os.path.join(self.__dir, self._tempname() + '.mp3')) + self.assertTrackCountEqual(0) + self._sleep() + with self._get_store() as store: + self.assertEqual(store.find(Track).count(), 3) + self.assertEqual(store.find(Artist).count(), 1) + +def suite(): + suite = unittest.TestSuite() + + suite.addTest(unittest.makeSuite(NothingToWatchTestCase)) + suite.addTest(unittest.makeSuite(WatcherTestCase)) + + return suite + +if __name__ == '__main__': + unittest.main() + diff --git a/tests/testbase.py b/tests/testbase.py index 1258edb..13bd98b 100644 --- a/tests/testbase.py +++ b/tests/testbase.py @@ -8,6 +8,7 @@ # # Distributed under terms of the GNU AGPLv3 license. +import inspect import io import shutil import sys @@ -33,14 +34,15 @@ class TestConfig(DefaultConfig): def __init__(self, with_webui, with_api): super(TestConfig, self).__init__() - for attr, value in self.__class__.__dict__.iteritems(): - if attr.startswith('_') or attr != attr.upper(): - continue + for cls in reversed(inspect.getmro(self.__class__)): + for attr, value in cls.__dict__.iteritems(): + if attr.startswith('_') or attr != attr.upper(): + continue - if isinstance(value, dict): - setattr(self, attr, value.copy()) - else: - setattr(self, attr, value) + if isinstance(value, dict): + setattr(self, attr, value.copy()) + else: + setattr(self, attr, value) self.WEBAPP.update({ 'mount_webui': with_webui,