mirror of
https://github.com/beetbox/beets.git
synced 2025-12-06 16:42:42 +01:00
508d28f66b
External Python packages interfacing beets may want to use an in-memory beets library instance for testing beets-related code. The `TestHelper` class is very helpful for this purpose. Previously `TestHelper` was located in the `test/` directory. Now it is part of `beets` itself (`beets.test.helper.TestHelper`) and can be easily imported.
397 lines
12 KiB
Python
397 lines
12 KiB
Python
# This file is part of beets.
|
|
# Copyright 2016, Thomas Scholtes.
|
|
#
|
|
# Permission is hereby granted, free of charge, to any person obtaining
|
|
# a copy of this software and associated documentation files (the
|
|
# "Software"), to deal in the Software without restriction, including
|
|
# without limitation the rights to use, copy, modify, merge, publish,
|
|
# distribute, sublicense, and/or sell copies of the Software, and to
|
|
# permit persons to whom the Software is furnished to do so, subject to
|
|
# the following conditions:
|
|
#
|
|
# The above copyright notice and this permission notice shall be
|
|
# included in all copies or substantial portions of the Software.
|
|
|
|
|
|
import os
|
|
import shutil
|
|
import tempfile
|
|
import unittest
|
|
from shlex import quote
|
|
|
|
import beets
|
|
from beets.test import _common, helper
|
|
|
|
|
|
class PlaylistTestHelper(helper.TestHelper):
|
|
def setUp(self):
|
|
self.setup_beets()
|
|
self.lib = beets.library.Library(":memory:")
|
|
|
|
self.music_dir = os.path.expanduser(os.path.join("~", "Music"))
|
|
|
|
i1 = _common.item()
|
|
i1.path = beets.util.normpath(
|
|
os.path.join(
|
|
self.music_dir,
|
|
"a",
|
|
"b",
|
|
"c.mp3",
|
|
)
|
|
)
|
|
i1.title = "some item"
|
|
i1.album = "some album"
|
|
self.lib.add(i1)
|
|
self.lib.add_album([i1])
|
|
|
|
i2 = _common.item()
|
|
i2.path = beets.util.normpath(
|
|
os.path.join(
|
|
self.music_dir,
|
|
"d",
|
|
"e",
|
|
"f.mp3",
|
|
)
|
|
)
|
|
i2.title = "another item"
|
|
i2.album = "another album"
|
|
self.lib.add(i2)
|
|
self.lib.add_album([i2])
|
|
|
|
i3 = _common.item()
|
|
i3.path = beets.util.normpath(
|
|
os.path.join(
|
|
self.music_dir,
|
|
"x",
|
|
"y",
|
|
"z.mp3",
|
|
)
|
|
)
|
|
i3.title = "yet another item"
|
|
i3.album = "yet another album"
|
|
self.lib.add(i3)
|
|
self.lib.add_album([i3])
|
|
|
|
self.playlist_dir = tempfile.mkdtemp()
|
|
self.config["directory"] = self.music_dir
|
|
self.config["playlist"]["playlist_dir"] = self.playlist_dir
|
|
|
|
self.setup_test()
|
|
self.load_plugins("playlist")
|
|
|
|
def setup_test(self):
|
|
raise NotImplementedError
|
|
|
|
def tearDown(self):
|
|
self.unload_plugins()
|
|
shutil.rmtree(self.playlist_dir)
|
|
self.teardown_beets()
|
|
|
|
|
|
class PlaylistQueryTestHelper(PlaylistTestHelper):
|
|
def test_name_query_with_absolute_paths_in_playlist(self):
|
|
q = "playlist:absolute"
|
|
results = self.lib.items(q)
|
|
self.assertEqual(
|
|
{i.title for i in results},
|
|
{
|
|
"some item",
|
|
"another item",
|
|
},
|
|
)
|
|
|
|
def test_path_query_with_absolute_paths_in_playlist(self):
|
|
q = "playlist:{}".format(
|
|
quote(
|
|
os.path.join(
|
|
self.playlist_dir,
|
|
"absolute.m3u",
|
|
)
|
|
)
|
|
)
|
|
results = self.lib.items(q)
|
|
self.assertEqual(
|
|
{i.title for i in results},
|
|
{
|
|
"some item",
|
|
"another item",
|
|
},
|
|
)
|
|
|
|
def test_name_query_with_relative_paths_in_playlist(self):
|
|
q = "playlist:relative"
|
|
results = self.lib.items(q)
|
|
self.assertEqual(
|
|
{i.title for i in results},
|
|
{
|
|
"some item",
|
|
"another item",
|
|
},
|
|
)
|
|
|
|
def test_path_query_with_relative_paths_in_playlist(self):
|
|
q = "playlist:{}".format(
|
|
quote(
|
|
os.path.join(
|
|
self.playlist_dir,
|
|
"relative.m3u",
|
|
)
|
|
)
|
|
)
|
|
results = self.lib.items(q)
|
|
self.assertEqual(
|
|
{i.title for i in results},
|
|
{
|
|
"some item",
|
|
"another item",
|
|
},
|
|
)
|
|
|
|
def test_name_query_with_nonexisting_playlist(self):
|
|
q = "playlist:nonexisting"
|
|
results = self.lib.items(q)
|
|
self.assertEqual(set(results), set())
|
|
|
|
def test_path_query_with_nonexisting_playlist(self):
|
|
q = "playlist:{}".format(
|
|
quote(
|
|
os.path.join(
|
|
self.playlist_dir,
|
|
self.playlist_dir,
|
|
"nonexisting.m3u",
|
|
)
|
|
)
|
|
)
|
|
results = self.lib.items(q)
|
|
self.assertEqual(set(results), set())
|
|
|
|
|
|
class PlaylistTestRelativeToLib(PlaylistQueryTestHelper, unittest.TestCase):
|
|
def setup_test(self):
|
|
with open(os.path.join(self.playlist_dir, "absolute.m3u"), "w") as f:
|
|
f.write(
|
|
"{}\n".format(os.path.join(self.music_dir, "a", "b", "c.mp3"))
|
|
)
|
|
f.write(
|
|
"{}\n".format(os.path.join(self.music_dir, "d", "e", "f.mp3"))
|
|
)
|
|
f.write(
|
|
"{}\n".format(os.path.join(self.music_dir, "nonexisting.mp3"))
|
|
)
|
|
|
|
with open(os.path.join(self.playlist_dir, "relative.m3u"), "w") as f:
|
|
f.write("{}\n".format(os.path.join("a", "b", "c.mp3")))
|
|
f.write("{}\n".format(os.path.join("d", "e", "f.mp3")))
|
|
f.write("{}\n".format("nonexisting.mp3"))
|
|
|
|
self.config["playlist"]["relative_to"] = "library"
|
|
|
|
|
|
class PlaylistTestRelativeToDir(PlaylistQueryTestHelper, unittest.TestCase):
|
|
def setup_test(self):
|
|
with open(os.path.join(self.playlist_dir, "absolute.m3u"), "w") as f:
|
|
f.write(
|
|
"{}\n".format(os.path.join(self.music_dir, "a", "b", "c.mp3"))
|
|
)
|
|
f.write(
|
|
"{}\n".format(os.path.join(self.music_dir, "d", "e", "f.mp3"))
|
|
)
|
|
f.write(
|
|
"{}\n".format(os.path.join(self.music_dir, "nonexisting.mp3"))
|
|
)
|
|
|
|
with open(os.path.join(self.playlist_dir, "relative.m3u"), "w") as f:
|
|
f.write("{}\n".format(os.path.join("a", "b", "c.mp3")))
|
|
f.write("{}\n".format(os.path.join("d", "e", "f.mp3")))
|
|
f.write("{}\n".format("nonexisting.mp3"))
|
|
|
|
self.config["playlist"]["relative_to"] = self.music_dir
|
|
|
|
|
|
class PlaylistTestRelativeToPls(PlaylistQueryTestHelper, unittest.TestCase):
|
|
def setup_test(self):
|
|
with open(os.path.join(self.playlist_dir, "absolute.m3u"), "w") as f:
|
|
f.write(
|
|
"{}\n".format(os.path.join(self.music_dir, "a", "b", "c.mp3"))
|
|
)
|
|
f.write(
|
|
"{}\n".format(os.path.join(self.music_dir, "d", "e", "f.mp3"))
|
|
)
|
|
f.write(
|
|
"{}\n".format(os.path.join(self.music_dir, "nonexisting.mp3"))
|
|
)
|
|
|
|
with open(os.path.join(self.playlist_dir, "relative.m3u"), "w") as f:
|
|
f.write(
|
|
"{}\n".format(
|
|
os.path.relpath(
|
|
os.path.join(self.music_dir, "a", "b", "c.mp3"),
|
|
start=self.playlist_dir,
|
|
)
|
|
)
|
|
)
|
|
f.write(
|
|
"{}\n".format(
|
|
os.path.relpath(
|
|
os.path.join(self.music_dir, "d", "e", "f.mp3"),
|
|
start=self.playlist_dir,
|
|
)
|
|
)
|
|
)
|
|
f.write(
|
|
"{}\n".format(
|
|
os.path.relpath(
|
|
os.path.join(self.music_dir, "nonexisting.mp3"),
|
|
start=self.playlist_dir,
|
|
)
|
|
)
|
|
)
|
|
|
|
self.config["playlist"]["relative_to"] = "playlist"
|
|
self.config["playlist"]["playlist_dir"] = self.playlist_dir
|
|
|
|
|
|
class PlaylistUpdateTestHelper(PlaylistTestHelper):
|
|
def setup_test(self):
|
|
with open(os.path.join(self.playlist_dir, "absolute.m3u"), "w") as f:
|
|
f.write(
|
|
"{}\n".format(os.path.join(self.music_dir, "a", "b", "c.mp3"))
|
|
)
|
|
f.write(
|
|
"{}\n".format(os.path.join(self.music_dir, "d", "e", "f.mp3"))
|
|
)
|
|
f.write(
|
|
"{}\n".format(os.path.join(self.music_dir, "nonexisting.mp3"))
|
|
)
|
|
|
|
with open(os.path.join(self.playlist_dir, "relative.m3u"), "w") as f:
|
|
f.write("{}\n".format(os.path.join("a", "b", "c.mp3")))
|
|
f.write("{}\n".format(os.path.join("d", "e", "f.mp3")))
|
|
f.write("{}\n".format("nonexisting.mp3"))
|
|
|
|
self.config["playlist"]["auto"] = True
|
|
self.config["playlist"]["relative_to"] = "library"
|
|
|
|
|
|
class PlaylistTestItemMoved(PlaylistUpdateTestHelper, unittest.TestCase):
|
|
def test_item_moved(self):
|
|
# Emit item_moved event for an item that is in a playlist
|
|
results = self.lib.items(
|
|
"path:{}".format(
|
|
quote(os.path.join(self.music_dir, "d", "e", "f.mp3"))
|
|
)
|
|
)
|
|
item = results[0]
|
|
beets.plugins.send(
|
|
"item_moved",
|
|
item=item,
|
|
source=item.path,
|
|
destination=beets.util.bytestring_path(
|
|
os.path.join(self.music_dir, "g", "h", "i.mp3")
|
|
),
|
|
)
|
|
|
|
# Emit item_moved event for an item that is not in a playlist
|
|
results = self.lib.items(
|
|
"path:{}".format(
|
|
quote(os.path.join(self.music_dir, "x", "y", "z.mp3"))
|
|
)
|
|
)
|
|
item = results[0]
|
|
beets.plugins.send(
|
|
"item_moved",
|
|
item=item,
|
|
source=item.path,
|
|
destination=beets.util.bytestring_path(
|
|
os.path.join(self.music_dir, "u", "v", "w.mp3")
|
|
),
|
|
)
|
|
|
|
# Emit cli_exit event
|
|
beets.plugins.send("cli_exit", lib=self.lib)
|
|
|
|
# Check playlist with absolute paths
|
|
playlist_path = os.path.join(self.playlist_dir, "absolute.m3u")
|
|
with open(playlist_path) as f:
|
|
lines = [line.strip() for line in f.readlines()]
|
|
|
|
self.assertEqual(
|
|
lines,
|
|
[
|
|
os.path.join(self.music_dir, "a", "b", "c.mp3"),
|
|
os.path.join(self.music_dir, "g", "h", "i.mp3"),
|
|
os.path.join(self.music_dir, "nonexisting.mp3"),
|
|
],
|
|
)
|
|
|
|
# Check playlist with relative paths
|
|
playlist_path = os.path.join(self.playlist_dir, "relative.m3u")
|
|
with open(playlist_path) as f:
|
|
lines = [line.strip() for line in f.readlines()]
|
|
|
|
self.assertEqual(
|
|
lines,
|
|
[
|
|
os.path.join("a", "b", "c.mp3"),
|
|
os.path.join("g", "h", "i.mp3"),
|
|
"nonexisting.mp3",
|
|
],
|
|
)
|
|
|
|
|
|
class PlaylistTestItemRemoved(PlaylistUpdateTestHelper, unittest.TestCase):
|
|
def test_item_removed(self):
|
|
# Emit item_removed event for an item that is in a playlist
|
|
results = self.lib.items(
|
|
"path:{}".format(
|
|
quote(os.path.join(self.music_dir, "d", "e", "f.mp3"))
|
|
)
|
|
)
|
|
item = results[0]
|
|
beets.plugins.send("item_removed", item=item)
|
|
|
|
# Emit item_removed event for an item that is not in a playlist
|
|
results = self.lib.items(
|
|
"path:{}".format(
|
|
quote(os.path.join(self.music_dir, "x", "y", "z.mp3"))
|
|
)
|
|
)
|
|
item = results[0]
|
|
beets.plugins.send("item_removed", item=item)
|
|
|
|
# Emit cli_exit event
|
|
beets.plugins.send("cli_exit", lib=self.lib)
|
|
|
|
# Check playlist with absolute paths
|
|
playlist_path = os.path.join(self.playlist_dir, "absolute.m3u")
|
|
with open(playlist_path) as f:
|
|
lines = [line.strip() for line in f.readlines()]
|
|
|
|
self.assertEqual(
|
|
lines,
|
|
[
|
|
os.path.join(self.music_dir, "a", "b", "c.mp3"),
|
|
os.path.join(self.music_dir, "nonexisting.mp3"),
|
|
],
|
|
)
|
|
|
|
# Check playlist with relative paths
|
|
playlist_path = os.path.join(self.playlist_dir, "relative.m3u")
|
|
with open(playlist_path) as f:
|
|
lines = [line.strip() for line in f.readlines()]
|
|
|
|
self.assertEqual(
|
|
lines,
|
|
[
|
|
os.path.join("a", "b", "c.mp3"),
|
|
"nonexisting.mp3",
|
|
],
|
|
)
|
|
|
|
|
|
def suite():
|
|
return unittest.TestLoader().loadTestsFromName(__name__)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main(defaultTest="suite")
|