Works on python 2/pypy

1 files changed, 96 insertions, 0 deletions

diff --git a/tests/watcher_test.py b/tests/watcher_test.py
new file mode 100644
index 0000000..5f92d36
--- /dev/null
+++ b/tests/watcher_test.py
@@ -0,0 +1,96 @@
+"""Tests for the watcher module."""
+from os import remove
+from os.path import dirname, join, realpath
+from shutil import copy, move, rmtree
+from tempfile import mkdtemp
+import unittest
+
+import mutagen
+import six
+
+from db.db_manager import DbManager
+from models.track import Track
+from watcher import LibraryWatcher
+
+
+if six.PY2:
+    def _u(string):
+        return unicode(string, encoding="utf_8")
+else:
+    def _u(string):
+        return string
+
+
+class WatcherTestCase(unittest.TestCase):
+    """Defines tests for the watcher module.
+
+    Extends:
+        unittest.Testcase
+
+    """
+
+    @classmethod
+    def setUpClass(cls):
+        """Set up fixtures for the tests."""
+        cls._tempdir = mkdtemp()
+        cls._tempdbdir = mkdtemp()
+        cls._db_path = join(cls._tempdbdir, "test.db")
+        copy(join(dirname(realpath(__file__)), "test.db"), cls._db_path)
+        cls._db = DbManager(cls._db_path)
+        cls._watcher = LibraryWatcher(cls._tempdir, cls._db_path)
+
+    @classmethod
+    def tearDownClass(cls):
+        """Remove test fixtures."""
+        cls._watcher.stop()
+        rmtree(cls._tempdbdir)
+        rmtree(cls._tempdir)
+
+    def test_watcher_actions(self):
+        """Test creating, moving, modifying and deleting a file."""
+        new_file = join(dirname(realpath(__file__)), "testnew.ogg")
+        copy(new_file, self._tempdir)
+
+        self._watcher.check_for_events()
+
+        found_track = Track.find_by_path(join(self._tempdir, "testnew.ogg"),
+                                         self._db)
+        assert found_track
+        assert "testnew.ogg" in found_track.filename
+        assert found_track.artists
+        found_artist = False
+        for artist in found_track.artists:
+            if _u("Art Ist") == artist.name:
+                found_artist = True
+                break
+
+        assert found_artist
+
+        original_file = join(self._tempdir, "testnew.ogg")
+        moved_file = join(self._tempdir, "testmoved.ogg")
+        move(original_file, moved_file)
+
+        self._watcher.check_for_events()
+
+        moved_track = Track.find_by_path(moved_file, self._db)
+
+        assert moved_track
+        assert "testmoved.ogg" in moved_track.filename
+
+        original_metadata = mutagen.File(moved_file, easy=True)
+
+        original_metadata["title"] = [_u("New title")]
+        original_metadata.save()
+
+        self._watcher.check_for_events()
+
+        modified_track = Track.find_by_path(moved_file, self._db)
+
+        assert modified_track
+        assert modified_track.name == "New title"
+
+        remove(moved_file)
+
+        self._watcher.check_for_events()
+
+        assert Track.find_by_path(moved_file, self._db) is None